A reaper/go_test.db

M reaper/reaper.go → reaper/reaper.go

@@ -93,13 +93,11 @@ log.Printf("[err] reaper: could not set feed fetch error '%s'\n", err)
 	}
 }
 
-// Have checks whether a given url is represented
+// HasFeed checks whether a given url is represented
 // in the reaper cache.
 func (r *Reaper) HasFeed(url string) bool {
-	for _, f := range r.feeds {
-		if f.UpdateURL == url {
-			return true
-		}
+	if _, ok := r.feeds[url]; ok {
+		return true
 	}
 	return false
 }

A reaper/reaper_test.go

@@ -0,0 +1,26 @@ 
+package reaper
+
+import (
+	"testing"
+
+	"git.j3s.sh/vore/rss"
+	"git.j3s.sh/vore/sqlite"
+)
+
+func TestHasFeed(t *testing.T) {
+	db := sqlite.New("go_test.db")
+	r := New(db)
+	f1 := rss.Feed{UpdateURL: "something"}
+	f2 := rss.Feed{UpdateURL: "strange"}
+	r.addFeed(&f1)
+	r.addFeed(&f2)
+	if r.HasFeed("banana") == true {
+		t.Fatal("reaper should not have a banana")
+	}
+	if r.HasFeed("something") == false {
+		t.Fatal("reaper should have something")
+	}
+	if r.HasFeed("strange") == false {
+		t.Fatal("reaper should have strange")
+	}
+}