diff --git a/persistence/user_queries.go b/persistence/user_queries.go index 2147fd8..e5025e7 100644 --- a/persistence/user_queries.go +++ b/persistence/user_queries.go @@ -245,7 +245,10 @@ func (p Profile) IsFollowing(user scraper.User) bool { var ret bool err := row.Scan(&ret) if err != nil { - panic(err) + if err == sql.ErrNoRows { + return false + } + panic(err) // A real error } return ret } diff --git a/persistence/user_queries_test.go b/persistence/user_queries_test.go index c5d36ff..89a8618 100644 --- a/persistence/user_queries_test.go +++ b/persistence/user_queries_test.go @@ -226,6 +226,8 @@ func TestIsFollowingUser(t *testing.T) { // Create the user user := create_dummy_user() assert.False(user.IsFollowed) + assert.False(profile.IsFollowing(user)) + err := profile.SaveUser(&user) assert.NoError(err)