ResolveSpec/tests/test_helpers.go

package test

import (
	"bytes"
	"encoding/json"
	"fmt"
	"io"
	"net/http"
	"net/http/httptest"
	"os"
	"testing"

	"github.com/Warky-Devs/ResolveSpec/pkg/logger"
	"github.com/Warky-Devs/ResolveSpec/pkg/models"
	"github.com/Warky-Devs/ResolveSpec/pkg/resolvespec"
	"github.com/Warky-Devs/ResolveSpec/pkg/testmodels"
	"github.com/glebarez/sqlite"
	"github.com/gorilla/mux"
	"github.com/stretchr/testify/assert"

	"gorm.io/gorm"
)

var (
	testDB        *gorm.DB
	testServer    *httptest.Server
	testServerURL string
)

// makeRequest is a helper function to make HTTP requests in tests
func makeRequest(t *testing.T, path string, payload interface{}) *http.Response {
	jsonData, err := json.Marshal(payload)
	assert.NoError(t, err, "Failed to marshal request payload")

	logger.Debug("Making request to %s with payload: %s", path, string(jsonData))

	req, err := http.NewRequest("POST", testServerURL+path, bytes.NewBuffer(jsonData))
	assert.NoError(t, err, "Failed to create request")

	req.Header.Set("Content-Type", "application/json")

	client := &http.Client{}
	resp, err := client.Do(req)
	assert.NoError(t, err, "Failed to execute request")

	if resp.StatusCode != http.StatusOK {
		body, _ := io.ReadAll(resp.Body)
		logger.Error("Request failed with status %d: %s", resp.StatusCode, string(body))
	} else {
		logger.Debug("Request successful with status %d", resp.StatusCode)
	}

	return resp
}

// verifyResponse is a helper function to verify response status and decode body
func verifyResponse(t *testing.T, resp *http.Response) map[string]interface{} {
	assert.Equal(t, http.StatusOK, resp.StatusCode, "Unexpected response status")

	var result map[string]interface{}
	err := json.NewDecoder(resp.Body).Decode(&result)
	assert.NoError(t, err, "Failed to decode response")
	assert.True(t, result["success"].(bool), "Response indicates failure")

	return result
}

// TestSetup initializes the test environment
func TestSetup(m *testing.M) int {
	logger.Init(true)

	logger.Info("Setting up test environment")

	// Create test database
	db, err := setupTestDB()
	if err != nil {
		logger.Error("Failed to setup test database: %v", err)
		return 1
	}
	testDB = db

	// Setup test server
	router := setupTestRouter(testDB)
	testServer = httptest.NewServer(router)

	fmt.Printf("ResolveSpec test server starting on  %s\n", testServer.URL)
	testServerURL = testServer.URL

	defer testServer.Close()

	// Run tests
	code := m.Run()

	// Cleanup
	logger.Info("Cleaning up test environment")
	cleanup()
	return code
}

// setupTestDB creates and initializes the test database
func setupTestDB() (*gorm.DB, error) {
	db, err := gorm.Open(sqlite.Open("file::memory:?cache=shared"), &gorm.Config{})
	if err != nil {
		return nil, fmt.Errorf("failed to open database: %v", err)
	}

	// Init Models
	testmodels.RegisterTestModels()

	// Auto migrate all test models
	err = autoMigrateModels(db)
	if err != nil {
		return nil, fmt.Errorf("failed to migrate models: %v", err)
	}

	return db, nil
}

// setupTestRouter creates and configures the test router
func setupTestRouter(db *gorm.DB) http.Handler {
	r := mux.NewRouter()
	handler := resolvespec.NewAPIHandler(db)

	r.HandleFunc("/{schema}/{entity}", func(w http.ResponseWriter, r *http.Request) {
		vars := mux.Vars(r)
		handler.Handle(w, r, vars)
	}).Methods("POST")

	r.HandleFunc("/{schema}/{entity}/{id}", func(w http.ResponseWriter, r *http.Request) {
		vars := mux.Vars(r)
		handler.Handle(w, r, vars)
	}).Methods("POST")

	return r
}

// cleanup performs test cleanup
func cleanup() {
	if testDB != nil {
		db, err := testDB.DB()
		if err == nil {
			db.Close()
		}
	}
	os.Remove("test.db")
}

// autoMigrateModels performs automigration for all test models
func autoMigrateModels(db *gorm.DB) error {
	modelList := models.GetModels()
	return db.AutoMigrate(modelList...)
}