package handlers_test import ( "database/sql" "encoding/json" "net/http" "net/http/httptest" "regexp" "testing" "time" "github.com/DATA-DOG/go-sqlmock" "github.com/rede5/gohorsejobs/backend/internal/api/handlers" "github.com/rede5/gohorsejobs/backend/internal/services" ) // createTestAdminHandlers creates handlers with mocks and optional DB func createTestAdminHandlers(t *testing.T, db *sql.DB) *handlers.AdminHandlers { t.Helper() var adminSvc *services.AdminService var jobSvc *services.JobService var auditSvc *services.AuditService if db != nil { adminSvc = services.NewAdminService(db) jobSvc = services.NewJobService(db) auditSvc = services.NewAuditService(db) } return handlers.NewAdminHandlers( adminSvc, auditSvc, jobSvc, nil, // Cloudflare service not needed for basic tests yet ) } func TestNewAdminHandlers(t *testing.T) { h := handlers.NewAdminHandlers(nil, nil, nil, nil) if h == nil { t.Error("NewAdminHandlers should not return nil") } } func TestAdminHandlers_ListCompanies(t *testing.T) { db, mock, err := sqlmock.New() if err != nil { t.Fatalf("an error '%s' was not expected when opening a stub database connection", err) } defer db.Close() handlers := createTestAdminHandlers(t, db) // Mock Count Query mock.ExpectQuery(regexp.QuoteMeta(`SELECT COUNT(*) FROM companies`)). WillReturnRows(sqlmock.NewRows([]string{"count"}).AddRow(1)) // Mock List Query // Mock List Query mock.ExpectQuery(regexp.QuoteMeta(`SELECT id, name, slug, type, document, address, region_id, city_id, phone, email, website, logo_url, description, active, verified, created_at, updated_at FROM companies ORDER BY created_at DESC LIMIT $1 OFFSET $2`)). WillReturnRows(sqlmock.NewRows([]string{"id", "name", "slug", "type", "document", "address", "region_id", "city_id", "phone", "email", "website", "logo_url", "description", "active", "verified", "created_at", "updated_at"}). AddRow(1, "Acme Corp", "acme-corp", "company", "1234567890", "Address", 1, 1, "123123123", "contact@acme.com", "https://acme.com", "", "Desc", true, true, time.Now(), time.Now())) req := httptest.NewRequest(http.MethodGet, "/api/v1/admin/companies", nil) rec := httptest.NewRecorder() handlers.ListCompanies(rec, req) if rec.Code != http.StatusOK { t.Errorf("Expected status %d, got %d. Body: %s", http.StatusOK, rec.Code, rec.Body.String()) } } func TestAdminHandlers_DuplicateJob(t *testing.T) { db, mock, err := sqlmock.New() if err != nil { t.Fatalf("an error '%s' was not expected when opening a stub database connection", err) } defer db.Close() handlers := createTestAdminHandlers(t, db) // 1. Mock GetJobByID (for duplication source) // Updated query based on actual implementation rows := sqlmock.NewRows([]string{"company_id", "created_by", "title", "description", "salary_min", "salary_max", "salary_type", "employment_type", "work_mode", "working_hours", "location", "region_id", "city_id", "requirements", "benefits", "visa_support", "language_level"}). AddRow(1, 1, "Original Job", "Desc", 1000, 2000, "monthly", "full-time", "remote", "8h", "Remote", 1, 1, "[]", "[]", false, "native") mock.ExpectQuery(regexp.QuoteMeta(`SELECT company_id, created_by, title, description, salary_min, salary_max, salary_type, employment_type, work_mode, working_hours, location, region_id, city_id, requirements, benefits, visa_support, language_level FROM jobs WHERE id = $1`)). WithArgs(sqlmock.AnyArg()). WillReturnRows(rows) // 2. Mock INSERT // Note: The implementation might be returning more or fewer columns, blindly matching logic usually safer but here we try to match. // Implementation typically inserts and returns ID. mock.ExpectQuery(regexp.QuoteMeta(`INSERT INTO jobs`)). WithArgs("1", "1", "Original Job", "Desc", 1000.0, 2000.0, "monthly", "full-time", "remote", "8h", "Remote", 1, 1, nil, nil, false, "native", "draft", false, sqlmock.AnyArg(), sqlmock.AnyArg()). WillReturnRows(sqlmock.NewRows([]string{"id"}).AddRow(101)) // Request req := httptest.NewRequest(http.MethodPost, "/api/v1/admin/jobs/100/duplicate", nil) req.SetPathValue("id", "100") rec := httptest.NewRecorder() handlers.DuplicateJob(rec, req) if rec.Code != http.StatusCreated { t.Errorf("Expected status %d, got %d. Body: %s", http.StatusCreated, rec.Code, rec.Body.String()) } } func TestAdminHandlers_ListAccessRoles(t *testing.T) { handlers := handlers.NewAdminHandlers(nil, nil, nil, nil) req := httptest.NewRequest(http.MethodGet, "/api/v1/admin/roles", nil) rec := httptest.NewRecorder() handlers.ListAccessRoles(rec, req) if rec.Code != http.StatusOK { t.Errorf("Expected status %d, got %d", http.StatusOK, rec.Code) } var roles []struct { Role string `json:"role"` Description string `json:"description"` Actions []string `json:"actions"` } if err := json.NewDecoder(rec.Body).Decode(&roles); err != nil { t.Fatalf("Failed to decode response: %v", err) } if len(roles) == 0 { t.Error("Expected roles, got empty list") } } func TestAdminHandlers_ListTags(t *testing.T) { db, mock, err := sqlmock.New() if err != nil { t.Fatalf("an error '%s' was not expected when opening a stub database connection", err) } defer db.Close() handlers := createTestAdminHandlers(t, db) // Mock ListTags Query rows := sqlmock.NewRows([]string{"id", "name", "category", "active", "created_at", "updated_at"}). AddRow(1, "Java", "skill", true, time.Now(), time.Now()). AddRow(2, "Remote", "benefit", true, time.Now(), time.Now()) mock.ExpectQuery(regexp.QuoteMeta(`SELECT id, name, category, active, created_at, updated_at FROM job_tags`)). WillReturnRows(rows) req := httptest.NewRequest(http.MethodGet, "/api/v1/admin/tags", nil) rec := httptest.NewRecorder() handlers.ListTags(rec, req) if rec.Code != http.StatusOK { t.Errorf("Expected status %d, got %d", http.StatusOK, rec.Code) } }