test: Update the HandleRoutes tests

Dobefu · Dec 29, 2024 · 5de4d27 · 5de4d27
1 parent e1579bc
commit 5de4d27
Showing 1 changed file with 48 additions and 0 deletions.
diff --git a/cmd/server/handle-routes_test.go b/cmd/server/handle-routes_test.go
@@ -356,6 +356,54 @@ func TestHandleRoutes(t *testing.T) {
 	assert.NotEqual(t, nil, body["error"])
 
 	os.Setenv("DB_CONN", oldDb)
+	err = database.Connect()
+	assert.Equal(t, nil, err)
+
+	body, err = request(
+		"GET",
+		fmt.Sprintf("%s/%s", server.URL, "translations"),
+		true,
+	)
+	assert.Equal(t, nil, err)
+	assert.NotEqual(t, nil, body["error"])
+
+	body, err = request(
+		"GET",
+		fmt.Sprintf("%s/%s", server.URL, "translations?locale=en"),
+		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, "translations?locale=en"),
+		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, "translations?locale=en"),
+		true,
+	)
+	assert.Equal(t, nil, err)
+	assert.NotEqual(t, nil, body["error"])
+
+	os.Setenv("DB_CONN", oldDb)
+	err = database.Connect()
+	assert.Equal(t, nil, err)
 }
 
 func request(method string, path string, withAuthToken bool) (body map[string]interface{}, err error) {