upaas/internal/handlers/api_test.go

package handlers_test

import (
	"encoding/json"
	"net/http"
	"net/http/httptest"
	"strings"
	"testing"

	"github.com/go-chi/chi/v5"
	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"

	"sneak.berlin/go/upaas/internal/service/app"
)

// apiRouter builds a chi router with the API routes using session auth middleware.
func apiRouter(tc *testContext) http.Handler {
	r := chi.NewRouter()

	r.Route("/api/v1", func(apiR chi.Router) {
		apiR.Post("/login", tc.handlers.HandleAPILoginPOST())

		apiR.Group(func(apiR chi.Router) {
			apiR.Use(tc.middleware.APISessionAuth())
			apiR.Get("/whoami", tc.handlers.HandleAPIWhoAmI())
			apiR.Get("/apps", tc.handlers.HandleAPIListApps())
			apiR.Get("/apps/{id}", tc.handlers.HandleAPIGetApp())
			apiR.Get("/apps/{id}/deployments", tc.handlers.HandleAPIListDeployments())

			// Backup/Restore API
			apiR.Get("/apps/{id}/export", tc.handlers.HandleAPIExportApp())
			apiR.Get("/backup/export", tc.handlers.HandleAPIExportAllApps())
			apiR.Post("/backup/import", tc.handlers.HandleAPIImportApps())
		})
	})

	return r
}

// setupAPITest creates a test context with a user and returns session cookies.
func setupAPITest(t *testing.T) (*testContext, []*http.Cookie) {
	t.Helper()

	tc := setupTestHandlers(t)

	// Create a user.
	_, err := tc.authSvc.CreateUser(t.Context(), "admin", "password123")
	require.NoError(t, err)

	// Login via the API to get session cookies.
	r := apiRouter(tc)

	loginBody := `{"username":"admin","password":"password123"}`
	req := httptest.NewRequest(http.MethodPost, "/api/v1/login", strings.NewReader(loginBody))
	req.Header.Set("Content-Type", "application/json")

	rr := httptest.NewRecorder()
	r.ServeHTTP(rr, req)

	require.Equal(t, http.StatusOK, rr.Code)

	cookies := rr.Result().Cookies()
	require.NotEmpty(t, cookies, "login should return session cookies")

	return tc, cookies
}

// apiGet makes an authenticated GET request using session cookies.
func apiGet(
	t *testing.T,
	tc *testContext,
	cookies []*http.Cookie,
	path string,
) *httptest.ResponseRecorder {
	t.Helper()

	req := httptest.NewRequest(http.MethodGet, path, nil)

	for _, c := range cookies {
		req.AddCookie(c)
	}

	rr := httptest.NewRecorder()

	r := apiRouter(tc)
	r.ServeHTTP(rr, req)

	return rr
}

func TestAPILoginSuccess(t *testing.T) {
	t.Parallel()

	tc := setupTestHandlers(t)

	_, err := tc.authSvc.CreateUser(t.Context(), "admin", "password123")
	require.NoError(t, err)

	r := apiRouter(tc)

	body := `{"username":"admin","password":"password123"}`
	req := httptest.NewRequest(http.MethodPost, "/api/v1/login", strings.NewReader(body))
	req.Header.Set("Content-Type", "application/json")

	rr := httptest.NewRecorder()
	r.ServeHTTP(rr, req)

	assert.Equal(t, http.StatusOK, rr.Code)

	var resp map[string]any
	require.NoError(t, json.Unmarshal(rr.Body.Bytes(), &resp))
	assert.Equal(t, "admin", resp["username"])

	// Should have a Set-Cookie header.
	assert.NotEmpty(t, rr.Result().Cookies())
}

func TestAPILoginInvalidCredentials(t *testing.T) {
	t.Parallel()

	tc := setupTestHandlers(t)

	_, err := tc.authSvc.CreateUser(t.Context(), "admin", "password123")
	require.NoError(t, err)

	r := apiRouter(tc)

	body := `{"username":"admin","password":"wrong"}`
	req := httptest.NewRequest(http.MethodPost, "/api/v1/login", strings.NewReader(body))
	req.Header.Set("Content-Type", "application/json")

	rr := httptest.NewRecorder()
	r.ServeHTTP(rr, req)

	assert.Equal(t, http.StatusUnauthorized, rr.Code)
}

func TestAPILoginMissingFields(t *testing.T) {
	t.Parallel()

	tc := setupTestHandlers(t)

	r := apiRouter(tc)

	body := `{"username":"","password":""}`
	req := httptest.NewRequest(http.MethodPost, "/api/v1/login", strings.NewReader(body))
	req.Header.Set("Content-Type", "application/json")

	rr := httptest.NewRecorder()
	r.ServeHTTP(rr, req)

	assert.Equal(t, http.StatusBadRequest, rr.Code)
}

func TestAPIRejectsUnauthenticated(t *testing.T) {
	t.Parallel()

	tc := setupTestHandlers(t)

	r := apiRouter(tc)

	req := httptest.NewRequest(http.MethodGet, "/api/v1/apps", nil)
	rr := httptest.NewRecorder()
	r.ServeHTTP(rr, req)

	assert.Equal(t, http.StatusUnauthorized, rr.Code)
}

func TestAPIWhoAmI(t *testing.T) {
	t.Parallel()

	tc, cookies := setupAPITest(t)

	rr := apiGet(t, tc, cookies, "/api/v1/whoami")
	assert.Equal(t, http.StatusOK, rr.Code)

	var resp map[string]any
	require.NoError(t, json.Unmarshal(rr.Body.Bytes(), &resp))
	assert.Equal(t, "admin", resp["username"])
}

func TestAPIListAppsEmpty(t *testing.T) {
	t.Parallel()

	tc, cookies := setupAPITest(t)

	rr := apiGet(t, tc, cookies, "/api/v1/apps")
	assert.Equal(t, http.StatusOK, rr.Code)

	var apps []any
	require.NoError(t, json.Unmarshal(rr.Body.Bytes(), &apps))
	assert.Empty(t, apps)
}

func TestAPIGetApp(t *testing.T) {
	t.Parallel()

	tc, cookies := setupAPITest(t)

	created, err := tc.appSvc.CreateApp(t.Context(), app.CreateAppInput{
		Name:    "my-app",
		RepoURL: "https://github.com/example/repo",
	})
	require.NoError(t, err)

	rr := apiGet(t, tc, cookies, "/api/v1/apps/"+created.ID)
	assert.Equal(t, http.StatusOK, rr.Code)

	var resp map[string]any
	require.NoError(t, json.Unmarshal(rr.Body.Bytes(), &resp))
	assert.Equal(t, "my-app", resp["name"])
}

func TestAPIGetAppNotFound(t *testing.T) {
	t.Parallel()

	tc, cookies := setupAPITest(t)

	rr := apiGet(t, tc, cookies, "/api/v1/apps/nonexistent")
	assert.Equal(t, http.StatusNotFound, rr.Code)
}

func TestAPIListDeployments(t *testing.T) {
	t.Parallel()

	tc, cookies := setupAPITest(t)

	created, err := tc.appSvc.CreateApp(t.Context(), app.CreateAppInput{
		Name:    "deploy-app",
		RepoURL: "https://github.com/example/repo",
	})
	require.NoError(t, err)

	rr := apiGet(t, tc, cookies, "/api/v1/apps/"+created.ID+"/deployments")
	assert.Equal(t, http.StatusOK, rr.Code)

	var deployments []any
	require.NoError(t, json.Unmarshal(rr.Body.Bytes(), &deployments))
	assert.Empty(t, deployments)
}