summary refs log tree commit diff stats
diff options
context:
space:
mode:
-rw-r--r--svc/handlers_test.go23
-rw-r--r--svc/init_test.go1
2 files changed, 22 insertions, 2 deletions
diff --git a/svc/handlers_test.go b/svc/handlers_test.go
index 82a8d86..41a8851 100644
--- a/svc/handlers_test.go
+++ b/svc/handlers_test.go
@@ -154,17 +154,36 @@ func Benchmark_apiTagsBaseHandler(b *testing.B) {
 }
 func Test_apiTagsHandler(t *testing.T) {
 	initTestConf()
+	mockRegistry()
 	t.Run("apiTagsHandler", func(t *testing.T) {
 		w := httptest.NewRecorder()
-		req := httptest.NewRequest("GET", "localhost"+testport+"/api/plain/tags/tag", nil)
+		req := httptest.NewRequest("GET", "http://localhost"+testport+"/api/plain/tags/programming", nil)
 		apiTagsHandler(w, req)
 		resp := w.Result()
+		data, err := ioutil.ReadAll(resp.Body)
+		if err != nil {
+			t.Errorf("%v\n", err)
+		}
 		if resp.StatusCode != http.StatusOK {
 			t.Errorf(fmt.Sprintf("%v", resp.StatusCode))
 		}
+		if len(data) == 0 {
+			t.Errorf("Got no data: %v\n", data)
+		}
 	})
 }
 
+func Benchmark_apiTagsHandler(b *testing.B) {
+	initTestConf()
+	mockRegistry()
+	w := httptest.NewRecorder()
+	r := httptest.NewRequest("GET", "http://localhost"+testport+"/api/plain/tags/programming", nil)
+	b.ResetTimer()
+	for i := 0; i < b.N; i++ {
+		apiTagsHandler(w, r)
+	}
+}
+
 func Test_cssHandler(t *testing.T) {
 	initTestConf()
 
@@ -175,7 +194,7 @@ func Test_cssHandler(t *testing.T) {
 	}
 
 	w := httptest.NewRecorder()
-	req := httptest.NewRequest("GET", "localhost"+testport+"/css", nil)
+	req := httptest.NewRequest("GET", "http://localhost"+testport+"/css", nil)
 
 	t.Run(name, func(t *testing.T) {
 		cssHandler(w, req)
diff --git a/svc/init_test.go b/svc/init_test.go
index b2fcd9a..51f72e2 100644
--- a/svc/init_test.go
+++ b/svc/init_test.go
@@ -93,6 +93,7 @@ func testConfig() {
 }
 
 func mockRegistry() {
+	twtxtCache = registry.NewIndex()
 	statuses, _, _ := registry.GetTwtxt("https://gbmor.dev/twtxt.txt")
 	parsed, _ := registry.ParseUserTwtxt(statuses, "gbmor", "https://gbmor.dev/twtxt.txt")
 	_ = twtxtCache.AddUser("gbmor", "https://gbmor.dev/twtxt.txt", "1", net.ParseIP("127.0.0.1"), parsed)