From 36e537b1cd412437bddf3015d59d6ddee3fdc6d5 Mon Sep 17 00:00:00 2001
From: Alessio <me@playfulpachyderm.com>
Date: Sat, 7 May 2022 14:35:53 -0700
Subject: [PATCH] Handle IsFollowing calls for users that aren't saved yet

---
 persistence/user_queries.go      | 5 ++++-
 persistence/user_queries_test.go | 2 ++
 2 files changed, 6 insertions(+), 1 deletion(-)

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)