test: Add some tests to the GetSitemapData function

Dobefu · Dec 26, 2024 · 10bb240 · 10bb240
1 parent c3cf107
commit 10bb240
Showing 1 changed file with 36 additions and 0 deletions.
diff --git a/cmd/server/handle-routes_test.go b/cmd/server/handle-routes_test.go
@@ -320,6 +320,42 @@ func TestHandleRoutes(t *testing.T) {
 	assert.NotEqual(t, nil, body["error"])
 
 	os.Setenv("CS_API_KEY", oldApiKey)
+
+	body, err = request(
+		"GET",
+		fmt.Sprintf("%s/%s", server.URL, "sitemap-data"),
+		true,
+	)
+	assert.Equal(t, nil, err)
+	assert.Equal(t, nil, body["error"])
+
+	oldApiKey = os.Getenv("CS_API_KEY")
+	os.Setenv("CS_API_KEY", "bogus")
+
+	body, err = request(
+		"GET",
+		fmt.Sprintf("%s/%s", server.URL, "sitemap-data"),
+		true,
+	)
+	assert.Equal(t, nil, err)
+	assert.Equal(t, nil, body["error"])
+
+	os.Setenv("CS_API_KEY", oldApiKey)
+
+	oldDb := os.Getenv("DB_CONN")
+	os.Setenv("DB_CONN", "file:/")
+	err = database.Connect()
+	assert.Equal(t, nil, err)
+
+	body, err = request(
+		"GET",
+		fmt.Sprintf("%s/%s", server.URL, "sitemap-data"),
+		true,
+	)
+	assert.Equal(t, nil, err)
+	assert.NotEqual(t, nil, body["error"])
+
+	os.Setenv("DB_CONN", oldDb)
 }
 
 func request(method string, path string, withAuthToken bool) (body map[string]interface{}, err error) {