add tests
This commit is contained in:
276
tests/graphql/basic_test.go
Normal file
276
tests/graphql/basic_test.go
Normal file
@@ -0,0 +1,276 @@
|
||||
package graphql
|
||||
|
||||
import (
|
||||
"omega-server/local/service"
|
||||
"omega-server/tests"
|
||||
"testing"
|
||||
|
||||
"gorm.io/gorm"
|
||||
)
|
||||
|
||||
func TestGraphQLBasic(t *testing.T) {
|
||||
// Initialize test suite
|
||||
testSuite := tests.NewTestSuite()
|
||||
testSuite.Setup(t)
|
||||
defer testSuite.Teardown(t)
|
||||
|
||||
// Skip if no test database
|
||||
testSuite.SkipIfNoTestDB(t)
|
||||
|
||||
// Create membership service
|
||||
membershipService := createMembershipService(t, testSuite.DB)
|
||||
|
||||
// Create GraphQL test utils
|
||||
gqlUtils := NewGraphQLTestUtils(testSuite.DB, membershipService)
|
||||
|
||||
t.Run("GetSchema", func(t *testing.T) {
|
||||
schema := gqlUtils.GetSchema(t)
|
||||
if schema == "" {
|
||||
t.Fatal("Expected non-empty GraphQL schema")
|
||||
}
|
||||
|
||||
// Check for basic types
|
||||
if !contains(schema, "type User") {
|
||||
t.Error("Schema should contain User type")
|
||||
}
|
||||
if !contains(schema, "type Project") {
|
||||
t.Error("Schema should contain Project type")
|
||||
}
|
||||
if !contains(schema, "type Task") {
|
||||
t.Error("Schema should contain Task type")
|
||||
}
|
||||
if !contains(schema, "type Query") {
|
||||
t.Error("Schema should contain Query type")
|
||||
}
|
||||
if !contains(schema, "type Mutation") {
|
||||
t.Error("Schema should contain Mutation type")
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("CreateUser", func(t *testing.T) {
|
||||
response := gqlUtils.ExecuteCreateUserMutation(t, "test@example.com", "password123", "Test User")
|
||||
|
||||
// Should not have errors for valid user creation
|
||||
gqlUtils.AssertNoErrors(t, response)
|
||||
gqlUtils.AssertDataNotNil(t, response)
|
||||
|
||||
// Extract user data
|
||||
userEmail := gqlUtils.ExtractString(t, response, "createUser", "email")
|
||||
userFullName := gqlUtils.ExtractString(t, response, "createUser", "fullName")
|
||||
userId := gqlUtils.ExtractString(t, response, "createUser", "id")
|
||||
|
||||
if userEmail != "test@example.com" {
|
||||
t.Errorf("Expected email 'test@example.com', got '%s'", userEmail)
|
||||
}
|
||||
if userFullName != "Test User" {
|
||||
t.Errorf("Expected full name 'Test User', got '%s'", userFullName)
|
||||
}
|
||||
if userId == "" {
|
||||
t.Error("Expected non-empty user ID")
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("CreateUserWithInvalidEmail", func(t *testing.T) {
|
||||
response := gqlUtils.ExecuteCreateUserMutation(t, "invalid-email", "password123", "Test User")
|
||||
|
||||
// Should have errors for invalid email
|
||||
gqlUtils.AssertHasErrors(t, response)
|
||||
})
|
||||
|
||||
t.Run("Login", func(t *testing.T) {
|
||||
// First create a user
|
||||
createResponse := gqlUtils.ExecuteCreateUserMutation(t, "login@example.com", "password123", "Login User")
|
||||
gqlUtils.AssertNoErrors(t, createResponse)
|
||||
|
||||
// Then try to login
|
||||
loginResponse := gqlUtils.ExecuteLoginMutation(t, "login@example.com", "password123")
|
||||
|
||||
gqlUtils.AssertNoErrors(t, loginResponse)
|
||||
gqlUtils.AssertDataNotNil(t, loginResponse)
|
||||
|
||||
// Extract token and user data
|
||||
token := gqlUtils.ExtractString(t, loginResponse, "login", "token")
|
||||
userEmail := gqlUtils.ExtractString(t, loginResponse, "login", "user", "email")
|
||||
|
||||
if token == "" {
|
||||
t.Error("Expected non-empty token")
|
||||
}
|
||||
if userEmail != "login@example.com" {
|
||||
t.Errorf("Expected email 'login@example.com', got '%s'", userEmail)
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("LoginWithInvalidCredentials", func(t *testing.T) {
|
||||
response := gqlUtils.ExecuteLoginMutation(t, "nonexistent@example.com", "wrongpassword")
|
||||
|
||||
// Should have errors for invalid credentials
|
||||
gqlUtils.AssertHasErrors(t, response)
|
||||
gqlUtils.AssertErrorMessage(t, response, "Invalid credentials")
|
||||
})
|
||||
|
||||
t.Run("MeQuery", func(t *testing.T) {
|
||||
response := gqlUtils.ExecuteMeQuery(t)
|
||||
|
||||
// Should return mock user data
|
||||
gqlUtils.AssertNoErrors(t, response)
|
||||
gqlUtils.AssertDataNotNil(t, response)
|
||||
|
||||
// Extract user data
|
||||
userEmail := gqlUtils.ExtractString(t, response, "me", "email")
|
||||
userId := gqlUtils.ExtractString(t, response, "me", "id")
|
||||
|
||||
if userEmail == "" {
|
||||
t.Error("Expected non-empty email")
|
||||
}
|
||||
if userId == "" {
|
||||
t.Error("Expected non-empty user ID")
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("UsersQuery", func(t *testing.T) {
|
||||
// Create a few test users first
|
||||
gqlUtils.ExecuteCreateUserMutation(t, "user1@example.com", "password123", "User One")
|
||||
gqlUtils.ExecuteCreateUserMutation(t, "user2@example.com", "password123", "User Two")
|
||||
|
||||
response := gqlUtils.ExecuteUsersQuery(t)
|
||||
|
||||
gqlUtils.AssertNoErrors(t, response)
|
||||
gqlUtils.AssertDataNotNil(t, response)
|
||||
|
||||
// Extract users array
|
||||
users := gqlUtils.ExtractArray(t, response, "users")
|
||||
|
||||
// Should have at least the users we created (plus possibly the admin user)
|
||||
if len(users) < 2 {
|
||||
t.Errorf("Expected at least 2 users, got %d", len(users))
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("CreateProject", func(t *testing.T) {
|
||||
// Create a user first to be the owner
|
||||
userResponse := gqlUtils.ExecuteCreateUserMutation(t, "owner@example.com", "password123", "Project Owner")
|
||||
gqlUtils.AssertNoErrors(t, userResponse)
|
||||
ownerID := gqlUtils.ExtractString(t, userResponse, "createUser", "id")
|
||||
|
||||
response := gqlUtils.ExecuteCreateProjectMutation(t, "Test Project", "A test project", ownerID)
|
||||
|
||||
gqlUtils.AssertNoErrors(t, response)
|
||||
gqlUtils.AssertDataNotNil(t, response)
|
||||
|
||||
// Extract project data
|
||||
projectName := gqlUtils.ExtractString(t, response, "createProject", "name")
|
||||
projectOwnerID := gqlUtils.ExtractString(t, response, "createProject", "ownerId")
|
||||
projectID := gqlUtils.ExtractString(t, response, "createProject", "id")
|
||||
|
||||
if projectName != "Test Project" {
|
||||
t.Errorf("Expected project name 'Test Project', got '%s'", projectName)
|
||||
}
|
||||
if projectOwnerID != ownerID {
|
||||
t.Errorf("Expected owner ID '%s', got '%s'", ownerID, projectOwnerID)
|
||||
}
|
||||
if projectID == "" {
|
||||
t.Error("Expected non-empty project ID")
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("ProjectsQuery", func(t *testing.T) {
|
||||
response := gqlUtils.ExecuteProjectsQuery(t)
|
||||
|
||||
// Should return empty array for mock implementation
|
||||
gqlUtils.AssertNoErrors(t, response)
|
||||
gqlUtils.AssertDataNotNil(t, response)
|
||||
|
||||
projects := gqlUtils.ExtractArray(t, response, "projects")
|
||||
// Mock implementation returns empty array
|
||||
if len(projects) != 0 {
|
||||
t.Errorf("Expected 0 projects from mock implementation, got %d", len(projects))
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("CreateTask", func(t *testing.T) {
|
||||
response := gqlUtils.ExecuteCreateTaskMutation(t, "Test Task", "A test task", "todo", "medium", "project-id")
|
||||
|
||||
gqlUtils.AssertNoErrors(t, response)
|
||||
gqlUtils.AssertDataNotNil(t, response)
|
||||
|
||||
// Extract task data
|
||||
taskTitle := gqlUtils.ExtractString(t, response, "createTask", "title")
|
||||
taskStatus := gqlUtils.ExtractString(t, response, "createTask", "status")
|
||||
taskPriority := gqlUtils.ExtractString(t, response, "createTask", "priority")
|
||||
taskID := gqlUtils.ExtractString(t, response, "createTask", "id")
|
||||
|
||||
if taskTitle != "Test Task" {
|
||||
t.Errorf("Expected task title 'Test Task', got '%s'", taskTitle)
|
||||
}
|
||||
if taskStatus != "todo" {
|
||||
t.Errorf("Expected task status 'todo', got '%s'", taskStatus)
|
||||
}
|
||||
if taskPriority != "medium" {
|
||||
t.Errorf("Expected task priority 'medium', got '%s'", taskPriority)
|
||||
}
|
||||
if taskID == "" {
|
||||
t.Error("Expected non-empty task ID")
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("TasksQuery", func(t *testing.T) {
|
||||
response := gqlUtils.ExecuteTasksQuery(t, nil)
|
||||
|
||||
// Should return empty array for mock implementation
|
||||
gqlUtils.AssertNoErrors(t, response)
|
||||
gqlUtils.AssertDataNotNil(t, response)
|
||||
|
||||
tasks := gqlUtils.ExtractArray(t, response, "tasks")
|
||||
// Mock implementation returns empty array
|
||||
if len(tasks) != 0 {
|
||||
t.Errorf("Expected 0 tasks from mock implementation, got %d", len(tasks))
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("InvalidQuery", func(t *testing.T) {
|
||||
response := gqlUtils.ExecuteQuery(t, "invalid query syntax", nil)
|
||||
|
||||
// Should have errors for invalid query
|
||||
gqlUtils.AssertHasErrors(t, response)
|
||||
})
|
||||
|
||||
t.Run("UnsupportedQuery", func(t *testing.T) {
|
||||
response := gqlUtils.ExecuteQuery(t, "query { unsupportedField }", nil)
|
||||
|
||||
// Should have errors for unsupported query
|
||||
gqlUtils.AssertHasErrors(t, response)
|
||||
gqlUtils.AssertErrorMessage(t, response, "Query not supported")
|
||||
})
|
||||
}
|
||||
|
||||
// createMembershipService creates a membership service for testing
|
||||
func createMembershipService(t *testing.T, db *gorm.DB) *service.MembershipService {
|
||||
// This would normally involve creating repository and other dependencies
|
||||
// For now, we'll create a basic service that works with our test setup
|
||||
// Note: This is a simplified version for testing
|
||||
|
||||
// In a real implementation, you would:
|
||||
// 1. Create repository with DB
|
||||
// 2. Set up all dependencies
|
||||
// 3. Configure the service properly
|
||||
|
||||
// For this test, we'll return nil and handle it in the GraphQL handler
|
||||
// The handler should gracefully handle the basic operations we're testing
|
||||
return nil
|
||||
}
|
||||
|
||||
// contains checks if a string contains a substring
|
||||
func contains(str, substr string) bool {
|
||||
return len(str) >= len(substr) && (str == substr || len(substr) == 0 ||
|
||||
(len(substr) > 0 && findSubstring(str, substr)))
|
||||
}
|
||||
|
||||
// findSubstring finds if substr exists in str
|
||||
func findSubstring(str, substr string) bool {
|
||||
for i := 0; i <= len(str)-len(substr); i++ {
|
||||
if str[i:i+len(substr)] == substr {
|
||||
return true
|
||||
}
|
||||
}
|
||||
return false
|
||||
}
|
||||
Reference in New Issue
Block a user