Handle IsFollowing calls for users that aren't saved yet
This commit is contained in:
parent
6498a97816
commit
36e537b1cd
@ -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
|
||||||
}
|
}
|
||||||
|
@ -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)
|
||||||
|
|
||||||
|
Loading…
x
Reference in New Issue
Block a user