Handle IsFollowing calls for users that aren't saved yet

This commit is contained in:
Alessio 2022-05-07 14:35:53 -07:00
parent 6498a97816
commit 36e537b1cd
2 changed files with 6 additions and 1 deletions

View File

@ -245,7 +245,10 @@ func (p Profile) IsFollowing(user scraper.User) bool {
var ret bool var ret bool
err := row.Scan(&ret) err := row.Scan(&ret)
if err != nil { if err != nil {
panic(err) if err == sql.ErrNoRows {
return false
}
panic(err) // A real error
} }
return ret return ret
} }

View File

@ -226,6 +226,8 @@ func TestIsFollowingUser(t *testing.T) {
// Create the user // Create the user
user := create_dummy_user() user := create_dummy_user()
assert.False(user.IsFollowed) assert.False(user.IsFollowed)
assert.False(profile.IsFollowing(user))
err := profile.SaveUser(&user) err := profile.SaveUser(&user)
assert.NoError(err) assert.NoError(err)