diff --git a/cmd/server/handle-routes_test.go b/cmd/server/handle-routes_test.go index 119f903..aa7b96b 100644 --- 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) {