8bd13d5ed7## Summary - Applied `gofmt -s -w` across the entire codebase to ensure consistent Go formatting - This will trigger a deployment with minimal code changes ## Test plan - [x] Pre-commit hooks pass - [x] No functional changes, formatting only 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-authored-by: Claude <noreply@anthropic.com>
969 lines
25 KiB
Go
969 lines
25 KiB
Go
package database_test
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/jackc/pgx/v5"
|
|
"github.com/modelcontextprotocol/registry/internal/database"
|
|
apiv0 "github.com/modelcontextprotocol/registry/pkg/api/v0"
|
|
"github.com/modelcontextprotocol/registry/pkg/model"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestPostgreSQL_CreateServer(t *testing.T) {
|
|
db := database.NewTestDB(t)
|
|
ctx := context.Background()
|
|
|
|
tests := []struct {
|
|
name string
|
|
serverJSON *apiv0.ServerJSON
|
|
officialMeta *apiv0.RegistryExtensions
|
|
expectError bool
|
|
errorType error
|
|
}{
|
|
{
|
|
name: "successful server creation",
|
|
serverJSON: &apiv0.ServerJSON{
|
|
Name: "com.example/test-server",
|
|
Description: "A test server",
|
|
Version: "1.0.0",
|
|
Remotes: []model.Transport{
|
|
{Type: "http", URL: "https://api.example.com/mcp"},
|
|
},
|
|
},
|
|
officialMeta: &apiv0.RegistryExtensions{
|
|
Status: model.StatusActive,
|
|
PublishedAt: time.Now(),
|
|
UpdatedAt: time.Now(),
|
|
IsLatest: true,
|
|
},
|
|
expectError: false,
|
|
},
|
|
{
|
|
name: "duplicate server version should fail",
|
|
serverJSON: &apiv0.ServerJSON{
|
|
Name: "com.example/duplicate-server",
|
|
Description: "A duplicate test server",
|
|
Version: "1.0.0",
|
|
},
|
|
officialMeta: &apiv0.RegistryExtensions{
|
|
Status: model.StatusActive,
|
|
PublishedAt: time.Now(),
|
|
UpdatedAt: time.Now(),
|
|
IsLatest: true,
|
|
},
|
|
expectError: true,
|
|
// Note: Expecting generic database error for constraint violation
|
|
},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
// Create the first server to test duplicates
|
|
if tt.name == "duplicate server version should fail" {
|
|
_, err := db.CreateServer(ctx, nil, tt.serverJSON, tt.officialMeta)
|
|
require.NoError(t, err, "First creation should succeed")
|
|
}
|
|
|
|
result, err := db.CreateServer(ctx, nil, tt.serverJSON, tt.officialMeta)
|
|
|
|
if tt.expectError {
|
|
assert.Error(t, err)
|
|
if tt.errorType != nil {
|
|
assert.ErrorIs(t, err, tt.errorType)
|
|
}
|
|
assert.Nil(t, result)
|
|
} else {
|
|
assert.NoError(t, err)
|
|
assert.NotNil(t, result)
|
|
assert.Equal(t, tt.serverJSON.Name, result.Server.Name)
|
|
assert.Equal(t, tt.serverJSON.Version, result.Server.Version)
|
|
assert.Equal(t, tt.serverJSON.Description, result.Server.Description)
|
|
assert.NotNil(t, result.Meta.Official)
|
|
assert.Equal(t, tt.officialMeta.Status, result.Meta.Official.Status)
|
|
assert.Equal(t, tt.officialMeta.IsLatest, result.Meta.Official.IsLatest)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestPostgreSQL_GetServerByName(t *testing.T) {
|
|
db := database.NewTestDB(t)
|
|
ctx := context.Background()
|
|
|
|
// Setup test data
|
|
serverJSON := &apiv0.ServerJSON{
|
|
Name: "com.example/get-test-server",
|
|
Description: "A server for get testing",
|
|
Version: "1.0.0",
|
|
}
|
|
officialMeta := &apiv0.RegistryExtensions{
|
|
Status: model.StatusActive,
|
|
PublishedAt: time.Now(),
|
|
UpdatedAt: time.Now(),
|
|
IsLatest: true,
|
|
}
|
|
|
|
// Create the server
|
|
_, err := db.CreateServer(ctx, nil, serverJSON, officialMeta)
|
|
require.NoError(t, err)
|
|
|
|
tests := []struct {
|
|
name string
|
|
serverName string
|
|
expectError bool
|
|
errorType error
|
|
}{
|
|
{
|
|
name: "get existing server",
|
|
serverName: "com.example/get-test-server",
|
|
},
|
|
{
|
|
name: "get non-existent server",
|
|
serverName: "com.example/non-existent",
|
|
expectError: true,
|
|
errorType: database.ErrNotFound,
|
|
},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
result, err := db.GetServerByName(ctx, nil, tt.serverName)
|
|
|
|
if tt.expectError {
|
|
assert.Error(t, err)
|
|
if tt.errorType != nil {
|
|
assert.ErrorIs(t, err, tt.errorType)
|
|
}
|
|
assert.Nil(t, result)
|
|
} else {
|
|
assert.NoError(t, err)
|
|
assert.NotNil(t, result)
|
|
assert.Equal(t, tt.serverName, result.Server.Name)
|
|
assert.NotNil(t, result.Meta.Official)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestPostgreSQL_GetServerByNameAndVersion(t *testing.T) {
|
|
db := database.NewTestDB(t)
|
|
ctx := context.Background()
|
|
|
|
// Setup test data with multiple versions
|
|
serverName := "com.example/version-test-server"
|
|
versions := []string{"1.0.0", "1.1.0", "2.0.0"}
|
|
|
|
for i, version := range versions {
|
|
serverJSON := &apiv0.ServerJSON{
|
|
Name: serverName,
|
|
Description: "A server for version testing",
|
|
Version: version,
|
|
}
|
|
officialMeta := &apiv0.RegistryExtensions{
|
|
Status: model.StatusActive,
|
|
PublishedAt: time.Now(),
|
|
UpdatedAt: time.Now(),
|
|
IsLatest: i == len(versions)-1, // Only last version is latest
|
|
}
|
|
|
|
_, err := db.CreateServer(ctx, nil, serverJSON, officialMeta)
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
tests := []struct {
|
|
name string
|
|
serverName string
|
|
version string
|
|
expectError bool
|
|
errorType error
|
|
}{
|
|
{
|
|
name: "get existing server version",
|
|
serverName: serverName,
|
|
version: "1.1.0",
|
|
},
|
|
{
|
|
name: "get non-existent version",
|
|
serverName: serverName,
|
|
version: "3.0.0",
|
|
expectError: true,
|
|
errorType: database.ErrNotFound,
|
|
},
|
|
{
|
|
name: "get non-existent server",
|
|
serverName: "com.example/non-existent",
|
|
version: "1.0.0",
|
|
expectError: true,
|
|
errorType: database.ErrNotFound,
|
|
},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
result, err := db.GetServerByNameAndVersion(ctx, nil, tt.serverName, tt.version)
|
|
|
|
if tt.expectError {
|
|
assert.Error(t, err)
|
|
if tt.errorType != nil {
|
|
assert.ErrorIs(t, err, tt.errorType)
|
|
}
|
|
assert.Nil(t, result)
|
|
} else {
|
|
assert.NoError(t, err)
|
|
assert.NotNil(t, result)
|
|
assert.Equal(t, tt.serverName, result.Server.Name)
|
|
assert.Equal(t, tt.version, result.Server.Version)
|
|
assert.NotNil(t, result.Meta.Official)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestPostgreSQL_ListServers(t *testing.T) {
|
|
db := database.NewTestDB(t)
|
|
ctx := context.Background()
|
|
|
|
// Setup test data
|
|
testServers := []struct {
|
|
name string
|
|
version string
|
|
status model.Status
|
|
remoteURL string
|
|
isLatest bool
|
|
publishedAt time.Time
|
|
}{
|
|
{
|
|
name: "com.example/server-a",
|
|
version: "1.0.0",
|
|
status: model.StatusActive,
|
|
remoteURL: "https://api-a.example.com/mcp",
|
|
isLatest: true,
|
|
publishedAt: time.Now().Add(-2 * time.Hour),
|
|
},
|
|
{
|
|
name: "com.example/server-b",
|
|
version: "2.0.0",
|
|
status: model.StatusActive,
|
|
remoteURL: "https://api-b.example.com/mcp",
|
|
isLatest: true,
|
|
publishedAt: time.Now().Add(-1 * time.Hour),
|
|
},
|
|
{
|
|
name: "com.example/server-c",
|
|
version: "1.0.0",
|
|
status: model.StatusDeprecated,
|
|
remoteURL: "https://api-c.example.com/mcp",
|
|
isLatest: true,
|
|
publishedAt: time.Now().Add(-30 * time.Minute),
|
|
},
|
|
}
|
|
|
|
// Create test servers
|
|
for _, server := range testServers {
|
|
serverJSON := &apiv0.ServerJSON{
|
|
Name: server.name,
|
|
Description: "Test server for listing",
|
|
Version: server.version,
|
|
Remotes: []model.Transport{
|
|
{Type: "http", URL: server.remoteURL},
|
|
},
|
|
}
|
|
officialMeta := &apiv0.RegistryExtensions{
|
|
Status: server.status,
|
|
PublishedAt: server.publishedAt,
|
|
UpdatedAt: server.publishedAt,
|
|
IsLatest: server.isLatest,
|
|
}
|
|
|
|
_, err := db.CreateServer(ctx, nil, serverJSON, officialMeta)
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
tests := []struct {
|
|
name string
|
|
filter *database.ServerFilter
|
|
cursor string
|
|
limit int
|
|
expectedCount int
|
|
expectedNames []string
|
|
expectError bool
|
|
}{
|
|
{
|
|
name: "list all servers",
|
|
filter: nil,
|
|
limit: 10,
|
|
expectedCount: 3,
|
|
expectedNames: []string{"com.example/server-a", "com.example/server-b", "com.example/server-c"},
|
|
},
|
|
{
|
|
name: "filter by name",
|
|
filter: &database.ServerFilter{
|
|
Name: stringPtr("com.example/server-a"),
|
|
},
|
|
limit: 10,
|
|
expectedCount: 1,
|
|
expectedNames: []string{"com.example/server-a"},
|
|
},
|
|
{
|
|
name: "filter by remote URL",
|
|
filter: &database.ServerFilter{
|
|
RemoteURL: stringPtr("https://api-b.example.com/mcp"),
|
|
},
|
|
limit: 10,
|
|
expectedCount: 1,
|
|
expectedNames: []string{"com.example/server-b"},
|
|
},
|
|
{
|
|
name: "filter by substring name",
|
|
filter: &database.ServerFilter{
|
|
SubstringName: stringPtr("server-"),
|
|
},
|
|
limit: 10,
|
|
expectedCount: 3,
|
|
},
|
|
{
|
|
name: "filter by version",
|
|
filter: &database.ServerFilter{
|
|
Version: stringPtr("1.0.0"),
|
|
},
|
|
limit: 10,
|
|
expectedCount: 2,
|
|
},
|
|
{
|
|
name: "filter by isLatest",
|
|
filter: &database.ServerFilter{
|
|
IsLatest: boolPtr(true),
|
|
},
|
|
limit: 10,
|
|
expectedCount: 3,
|
|
},
|
|
{
|
|
name: "filter by updatedSince",
|
|
filter: &database.ServerFilter{
|
|
UpdatedSince: timePtr(time.Now().Add(-45 * time.Minute)),
|
|
},
|
|
limit: 10,
|
|
expectedCount: 1, // Only server-c was updated in the last 45 minutes
|
|
},
|
|
{
|
|
name: "test pagination with limit",
|
|
filter: nil,
|
|
limit: 2,
|
|
expectedCount: 2,
|
|
},
|
|
{
|
|
name: "test cursor pagination",
|
|
filter: nil,
|
|
cursor: "com.example/server-a",
|
|
limit: 10,
|
|
// Should return servers after 'server-a' alphabetically
|
|
expectedCount: 2,
|
|
expectedNames: []string{"com.example/server-b", "com.example/server-c"},
|
|
},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
results, nextCursor, err := db.ListServers(ctx, nil, tt.filter, tt.cursor, tt.limit)
|
|
|
|
if tt.expectError {
|
|
assert.Error(t, err)
|
|
return
|
|
}
|
|
|
|
assert.NoError(t, err)
|
|
assert.Len(t, results, tt.expectedCount)
|
|
|
|
if len(tt.expectedNames) > 0 {
|
|
actualNames := make([]string, len(results))
|
|
for i, result := range results {
|
|
actualNames[i] = result.Server.Name
|
|
}
|
|
assert.Subset(t, tt.expectedNames, actualNames)
|
|
}
|
|
|
|
// Test cursor behavior
|
|
if tt.limit < len(testServers) && len(results) == tt.limit {
|
|
assert.NotEmpty(t, nextCursor, "Should return next cursor when results are limited")
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestPostgreSQL_UpdateServer(t *testing.T) {
|
|
db := database.NewTestDB(t)
|
|
ctx := context.Background()
|
|
|
|
// Setup test data
|
|
serverName := "com.example/update-test-server"
|
|
version := "1.0.0"
|
|
serverJSON := &apiv0.ServerJSON{
|
|
Name: serverName,
|
|
Description: "Original description",
|
|
Version: version,
|
|
}
|
|
officialMeta := &apiv0.RegistryExtensions{
|
|
Status: model.StatusActive,
|
|
PublishedAt: time.Now(),
|
|
UpdatedAt: time.Now(),
|
|
IsLatest: true,
|
|
}
|
|
|
|
_, err := db.CreateServer(ctx, nil, serverJSON, officialMeta)
|
|
require.NoError(t, err)
|
|
|
|
tests := []struct {
|
|
name string
|
|
serverName string
|
|
version string
|
|
updatedServer *apiv0.ServerJSON
|
|
expectError bool
|
|
errorType error
|
|
}{
|
|
{
|
|
name: "successful server update",
|
|
serverName: serverName,
|
|
version: version,
|
|
updatedServer: &apiv0.ServerJSON{
|
|
Name: serverName,
|
|
Description: "Updated description",
|
|
Version: version,
|
|
Remotes: []model.Transport{
|
|
{Type: "http", URL: "https://updated.example.com/mcp"},
|
|
},
|
|
},
|
|
},
|
|
{
|
|
name: "update non-existent server",
|
|
serverName: "com.example/non-existent",
|
|
version: "1.0.0",
|
|
updatedServer: &apiv0.ServerJSON{
|
|
Name: "com.example/non-existent",
|
|
Description: "Should fail",
|
|
Version: "1.0.0",
|
|
},
|
|
expectError: true,
|
|
errorType: database.ErrNotFound,
|
|
},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
result, err := db.UpdateServer(ctx, nil, tt.serverName, tt.version, tt.updatedServer)
|
|
|
|
if tt.expectError {
|
|
assert.Error(t, err)
|
|
if tt.errorType != nil {
|
|
assert.ErrorIs(t, err, tt.errorType)
|
|
}
|
|
assert.Nil(t, result)
|
|
} else {
|
|
assert.NoError(t, err)
|
|
assert.NotNil(t, result)
|
|
assert.Equal(t, tt.updatedServer.Description, result.Server.Description)
|
|
assert.NotNil(t, result.Meta.Official)
|
|
assert.NotZero(t, result.Meta.Official.UpdatedAt)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestPostgreSQL_SetServerStatus(t *testing.T) {
|
|
db := database.NewTestDB(t)
|
|
ctx := context.Background()
|
|
|
|
// Setup test data
|
|
serverName := "com.example/status-test-server"
|
|
version := "1.0.0"
|
|
serverJSON := &apiv0.ServerJSON{
|
|
Name: serverName,
|
|
Description: "A server for status testing",
|
|
Version: version,
|
|
}
|
|
officialMeta := &apiv0.RegistryExtensions{
|
|
Status: model.StatusActive,
|
|
PublishedAt: time.Now(),
|
|
UpdatedAt: time.Now(),
|
|
IsLatest: true,
|
|
}
|
|
|
|
_, err := db.CreateServer(ctx, nil, serverJSON, officialMeta)
|
|
require.NoError(t, err)
|
|
|
|
tests := []struct {
|
|
name string
|
|
serverName string
|
|
version string
|
|
newStatus string
|
|
expectError bool
|
|
errorType error
|
|
}{
|
|
{
|
|
name: "active to deprecated",
|
|
serverName: serverName,
|
|
version: version,
|
|
newStatus: string(model.StatusDeprecated),
|
|
},
|
|
{
|
|
name: "invalid status",
|
|
serverName: serverName,
|
|
version: version,
|
|
newStatus: "invalid_status",
|
|
expectError: true,
|
|
},
|
|
{
|
|
name: "non-existent server",
|
|
serverName: "com.example/non-existent",
|
|
version: "1.0.1",
|
|
newStatus: string(model.StatusDeprecated),
|
|
expectError: true,
|
|
errorType: database.ErrNotFound,
|
|
},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
result, err := db.SetServerStatus(ctx, nil, tt.serverName, tt.version, tt.newStatus)
|
|
|
|
if tt.expectError {
|
|
assert.Error(t, err)
|
|
if tt.errorType != nil {
|
|
assert.ErrorIs(t, err, tt.errorType)
|
|
}
|
|
assert.Nil(t, result)
|
|
} else {
|
|
assert.NoError(t, err)
|
|
assert.NotNil(t, result)
|
|
assert.Equal(t, model.Status(tt.newStatus), result.Meta.Official.Status)
|
|
assert.NotZero(t, result.Meta.Official.UpdatedAt)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestPostgreSQL_TransactionHandling(t *testing.T) {
|
|
db := database.NewTestDB(t)
|
|
ctx := context.Background()
|
|
|
|
t.Run("successful transaction", func(t *testing.T) {
|
|
err := db.InTransaction(ctx, func(ctx context.Context, tx pgx.Tx) error {
|
|
serverJSON := &apiv0.ServerJSON{
|
|
Name: "com.example/transaction-success",
|
|
Description: "Transaction test server",
|
|
Version: "1.0.0",
|
|
}
|
|
officialMeta := &apiv0.RegistryExtensions{
|
|
Status: model.StatusActive,
|
|
PublishedAt: time.Now(),
|
|
UpdatedAt: time.Now(),
|
|
IsLatest: true,
|
|
}
|
|
|
|
_, err := db.CreateServer(ctx, tx, serverJSON, officialMeta)
|
|
return err
|
|
})
|
|
|
|
assert.NoError(t, err)
|
|
|
|
// Verify server was created
|
|
result, err := db.GetServerByName(ctx, nil, "com.example/transaction-success")
|
|
assert.NoError(t, err)
|
|
assert.NotNil(t, result)
|
|
})
|
|
|
|
t.Run("failed transaction rollback", func(t *testing.T) {
|
|
err := db.InTransaction(ctx, func(ctx context.Context, tx pgx.Tx) error {
|
|
serverJSON := &apiv0.ServerJSON{
|
|
Name: "com.example/transaction-rollback",
|
|
Description: "Transaction rollback test server",
|
|
Version: "1.0.0",
|
|
}
|
|
officialMeta := &apiv0.RegistryExtensions{
|
|
Status: model.StatusActive,
|
|
PublishedAt: time.Now(),
|
|
UpdatedAt: time.Now(),
|
|
IsLatest: true,
|
|
}
|
|
|
|
_, err := db.CreateServer(ctx, tx, serverJSON, officialMeta)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// Force an error to trigger rollback
|
|
return assert.AnError
|
|
})
|
|
|
|
assert.Error(t, err)
|
|
assert.Equal(t, assert.AnError, err)
|
|
|
|
// Verify server was NOT created due to rollback
|
|
result, err := db.GetServerByName(ctx, nil, "com.example/transaction-rollback")
|
|
assert.Error(t, err)
|
|
assert.ErrorIs(t, err, database.ErrNotFound)
|
|
assert.Nil(t, result)
|
|
})
|
|
}
|
|
|
|
func TestPostgreSQL_ConcurrencyAndLocking(t *testing.T) {
|
|
db := database.NewTestDB(t)
|
|
ctx := context.Background()
|
|
|
|
serverName := "com.example/concurrent-server"
|
|
|
|
// Test advisory locking prevents concurrent publishes
|
|
t.Run("advisory locking prevents race conditions", func(t *testing.T) {
|
|
published := make(chan bool, 2)
|
|
errors := make(chan error, 2)
|
|
|
|
// Launch two concurrent publish operations
|
|
for i := 0; i < 2; i++ {
|
|
go func(version string) {
|
|
err := db.InTransaction(ctx, func(ctx context.Context, tx pgx.Tx) error {
|
|
// Acquire lock
|
|
if err := db.AcquirePublishLock(ctx, tx, serverName); err != nil {
|
|
return err
|
|
}
|
|
|
|
// Simulate some processing time
|
|
time.Sleep(100 * time.Millisecond)
|
|
|
|
serverJSON := &apiv0.ServerJSON{
|
|
Name: serverName,
|
|
Description: "Concurrent test server",
|
|
Version: version,
|
|
}
|
|
officialMeta := &apiv0.RegistryExtensions{
|
|
Status: model.StatusActive,
|
|
PublishedAt: time.Now(),
|
|
UpdatedAt: time.Now(),
|
|
IsLatest: true,
|
|
}
|
|
|
|
_, err := db.CreateServer(ctx, tx, serverJSON, officialMeta)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
published <- true
|
|
return nil
|
|
})
|
|
errors <- err
|
|
}(time.Now().Format("20060102150405.000000"))
|
|
}
|
|
|
|
// Wait for both goroutines to complete
|
|
err1 := <-errors
|
|
err2 := <-errors
|
|
|
|
// One should succeed, one should wait (or fail if timeout)
|
|
successCount := 0
|
|
if err1 == nil {
|
|
successCount++
|
|
}
|
|
if err2 == nil {
|
|
successCount++
|
|
}
|
|
|
|
// At least one should succeed (both can succeed if advisory lock works properly)
|
|
assert.GreaterOrEqual(t, successCount, 1, "At least one concurrent operation should succeed")
|
|
|
|
close(published)
|
|
close(errors)
|
|
})
|
|
}
|
|
|
|
func TestPostgreSQL_HelperMethods(t *testing.T) {
|
|
db := database.NewTestDB(t)
|
|
ctx := context.Background()
|
|
|
|
serverName := "com.example/helper-test-server"
|
|
|
|
// Setup test data with multiple versions
|
|
versions := []string{"1.0.0", "1.1.0", "2.0.0"}
|
|
for _, version := range versions {
|
|
serverJSON := &apiv0.ServerJSON{
|
|
Name: serverName,
|
|
Description: "Helper methods test server",
|
|
Version: version,
|
|
}
|
|
officialMeta := &apiv0.RegistryExtensions{
|
|
Status: model.StatusActive,
|
|
PublishedAt: time.Now(),
|
|
UpdatedAt: time.Now(),
|
|
IsLatest: version == "2.0.0",
|
|
}
|
|
|
|
_, err := db.CreateServer(ctx, nil, serverJSON, officialMeta)
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
t.Run("CountServerVersions", func(t *testing.T) {
|
|
count, err := db.CountServerVersions(ctx, nil, serverName)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, 3, count)
|
|
|
|
// Test non-existent server
|
|
count, err = db.CountServerVersions(ctx, nil, "com.example/non-existent")
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, 0, count)
|
|
})
|
|
|
|
t.Run("CheckVersionExists", func(t *testing.T) {
|
|
exists, err := db.CheckVersionExists(ctx, nil, serverName, "1.1.0")
|
|
assert.NoError(t, err)
|
|
assert.True(t, exists)
|
|
|
|
exists, err = db.CheckVersionExists(ctx, nil, serverName, "3.0.0")
|
|
assert.NoError(t, err)
|
|
assert.False(t, exists)
|
|
})
|
|
|
|
t.Run("GetCurrentLatestVersion", func(t *testing.T) {
|
|
latest, err := db.GetCurrentLatestVersion(ctx, nil, serverName)
|
|
assert.NoError(t, err)
|
|
assert.NotNil(t, latest)
|
|
assert.Equal(t, "2.0.0", latest.Server.Version)
|
|
assert.True(t, latest.Meta.Official.IsLatest)
|
|
})
|
|
|
|
t.Run("GetAllVersionsByServerName", func(t *testing.T) {
|
|
allVersions, err := db.GetAllVersionsByServerName(ctx, nil, serverName)
|
|
assert.NoError(t, err)
|
|
assert.Len(t, allVersions, 3)
|
|
|
|
// Check versions are present
|
|
versionSet := make(map[string]bool)
|
|
for _, server := range allVersions {
|
|
versionSet[server.Server.Version] = true
|
|
}
|
|
for _, expectedVersion := range versions {
|
|
assert.True(t, versionSet[expectedVersion], "Version %s should be present", expectedVersion)
|
|
}
|
|
})
|
|
|
|
t.Run("UnmarkAsLatest", func(t *testing.T) {
|
|
err := db.UnmarkAsLatest(ctx, nil, serverName)
|
|
assert.NoError(t, err)
|
|
|
|
// Verify no version is marked as latest
|
|
latest, err := db.GetCurrentLatestVersion(ctx, nil, serverName)
|
|
assert.Error(t, err)
|
|
assert.ErrorIs(t, err, database.ErrNotFound)
|
|
assert.Nil(t, latest)
|
|
})
|
|
}
|
|
|
|
func TestPostgreSQL_EdgeCases(t *testing.T) {
|
|
db := database.NewTestDB(t)
|
|
ctx := context.Background()
|
|
|
|
t.Run("input validation", func(t *testing.T) {
|
|
// Test nil inputs
|
|
_, err := db.CreateServer(ctx, nil, nil, nil)
|
|
assert.Error(t, err)
|
|
assert.Contains(t, err.Error(), "serverJSON and officialMeta are required")
|
|
|
|
// Test empty required fields
|
|
_, err = db.CreateServer(ctx, nil, &apiv0.ServerJSON{}, &apiv0.RegistryExtensions{})
|
|
assert.Error(t, err)
|
|
assert.Contains(t, err.Error(), "server name and version are required")
|
|
})
|
|
|
|
t.Run("database constraints", func(t *testing.T) {
|
|
// Test server name format constraint (should be caught by database constraint)
|
|
invalidServer := &apiv0.ServerJSON{
|
|
Name: "invalid-name-format", // Missing namespace/name format
|
|
Description: "Invalid server",
|
|
Version: "1.0.0",
|
|
}
|
|
officialMeta := &apiv0.RegistryExtensions{
|
|
Status: model.StatusActive,
|
|
PublishedAt: time.Now(),
|
|
UpdatedAt: time.Now(),
|
|
IsLatest: true,
|
|
}
|
|
|
|
_, err := db.CreateServer(ctx, nil, invalidServer, officialMeta)
|
|
assert.Error(t, err, "Should fail due to server name format constraint")
|
|
})
|
|
|
|
t.Run("pagination edge cases", func(t *testing.T) {
|
|
// Test pagination with no results
|
|
results, cursor, err := db.ListServers(ctx, nil, &database.ServerFilter{
|
|
Name: stringPtr("com.example/non-existent-server"),
|
|
}, "", 10)
|
|
assert.NoError(t, err)
|
|
assert.Empty(t, results)
|
|
assert.Empty(t, cursor)
|
|
|
|
// Test pagination with limit 0 (should use default)
|
|
_, _, err = db.ListServers(ctx, nil, nil, "", 0)
|
|
assert.NoError(t, err)
|
|
// Should still work with default limit
|
|
})
|
|
|
|
t.Run("complex filtering", func(t *testing.T) {
|
|
// Setup test data
|
|
serverName := "com.example/complex-filter-server"
|
|
testTime := time.Now().Add(-1 * time.Hour)
|
|
|
|
_, err := db.CreateServer(ctx, nil, &apiv0.ServerJSON{
|
|
Name: serverName,
|
|
Description: "Complex filter test server",
|
|
Version: "1.0.0",
|
|
Remotes: []model.Transport{
|
|
{Type: "streamable-http", URL: "https://complex.example.com/mcp"},
|
|
},
|
|
}, &apiv0.RegistryExtensions{
|
|
Status: model.StatusActive,
|
|
PublishedAt: testTime,
|
|
UpdatedAt: testTime,
|
|
IsLatest: true,
|
|
})
|
|
require.NoError(t, err)
|
|
|
|
// Test multiple filters combined
|
|
filter := &database.ServerFilter{
|
|
SubstringName: stringPtr("complex"),
|
|
UpdatedSince: timePtr(testTime.Add(-30 * time.Minute)),
|
|
IsLatest: boolPtr(true),
|
|
Version: stringPtr("1.0.0"),
|
|
}
|
|
|
|
results, _, err := db.ListServers(ctx, nil, filter, "", 10)
|
|
assert.NoError(t, err)
|
|
assert.Len(t, results, 1)
|
|
assert.Equal(t, serverName, results[0].Server.Name)
|
|
})
|
|
|
|
t.Run("status transitions", func(t *testing.T) {
|
|
serverName := "com.example/status-transition-server"
|
|
version := "1.0.2"
|
|
|
|
// Create server
|
|
_, err := db.CreateServer(ctx, nil, &apiv0.ServerJSON{
|
|
Name: serverName,
|
|
Description: "Status transition test",
|
|
Version: version,
|
|
}, &apiv0.RegistryExtensions{
|
|
Status: model.StatusActive,
|
|
PublishedAt: time.Now(),
|
|
UpdatedAt: time.Now(),
|
|
IsLatest: true,
|
|
})
|
|
require.NoError(t, err)
|
|
|
|
// Test all valid status transitions
|
|
statuses := []string{
|
|
string(model.StatusDeprecated),
|
|
string(model.StatusDeleted),
|
|
string(model.StatusActive), // Can transition back
|
|
}
|
|
|
|
for _, status := range statuses {
|
|
result, err := db.SetServerStatus(ctx, nil, serverName, version, status)
|
|
assert.NoError(t, err, "Should allow transition to %s", status)
|
|
assert.Equal(t, model.Status(status), result.Meta.Official.Status)
|
|
}
|
|
})
|
|
}
|
|
|
|
func TestPostgreSQL_PerformanceScenarios(t *testing.T) {
|
|
db := database.NewTestDB(t)
|
|
ctx := context.Background()
|
|
|
|
t.Run("many versions management", func(t *testing.T) {
|
|
serverName := "com.example/many-versions-server"
|
|
|
|
// Create many versions (but stay under the limit)
|
|
versionCount := 50
|
|
for i := 0; i < versionCount; i++ {
|
|
_, err := db.CreateServer(ctx, nil, &apiv0.ServerJSON{
|
|
Name: serverName,
|
|
Description: fmt.Sprintf("Version %d", i),
|
|
Version: fmt.Sprintf("1.0.%d", i),
|
|
}, &apiv0.RegistryExtensions{
|
|
Status: model.StatusActive,
|
|
PublishedAt: time.Now(),
|
|
UpdatedAt: time.Now(),
|
|
IsLatest: i == versionCount-1, // Only last one is latest
|
|
})
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
// Test counting versions
|
|
count, err := db.CountServerVersions(ctx, nil, serverName)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, versionCount, count)
|
|
|
|
// Test getting all versions
|
|
allVersions, err := db.GetAllVersionsByServerName(ctx, nil, serverName)
|
|
assert.NoError(t, err)
|
|
assert.Len(t, allVersions, versionCount)
|
|
|
|
// Test only one is marked as latest
|
|
latestCount := 0
|
|
for _, version := range allVersions {
|
|
if version.Meta.Official.IsLatest {
|
|
latestCount++
|
|
}
|
|
}
|
|
assert.Equal(t, 1, latestCount)
|
|
})
|
|
|
|
t.Run("large result pagination", func(t *testing.T) {
|
|
// Create multiple servers for pagination testing
|
|
serverCount := 25
|
|
for i := 0; i < serverCount; i++ {
|
|
_, err := db.CreateServer(ctx, nil, &apiv0.ServerJSON{
|
|
Name: fmt.Sprintf("com.example/pagination-server-%02d", i),
|
|
Description: "Pagination test server",
|
|
Version: "1.0.0",
|
|
}, &apiv0.RegistryExtensions{
|
|
Status: model.StatusActive,
|
|
PublishedAt: time.Now(),
|
|
UpdatedAt: time.Now(),
|
|
IsLatest: true,
|
|
})
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
// Test paginated retrieval
|
|
allResults := []*apiv0.ServerResponse{}
|
|
cursor := ""
|
|
pageSize := 10
|
|
|
|
for {
|
|
results, nextCursor, err := db.ListServers(ctx, nil, nil, cursor, pageSize)
|
|
assert.NoError(t, err)
|
|
allResults = append(allResults, results...)
|
|
|
|
if nextCursor == "" || len(results) < pageSize {
|
|
break
|
|
}
|
|
cursor = nextCursor
|
|
}
|
|
|
|
// Should have retrieved all servers including the ones we just created
|
|
assert.GreaterOrEqual(t, len(allResults), serverCount)
|
|
})
|
|
}
|
|
|
|
// Helper functions for creating pointers to basic types
|
|
func stringPtr(s string) *string {
|
|
return &s
|
|
}
|
|
|
|
func boolPtr(b bool) *bool {
|
|
return &b
|
|
}
|
|
|
|
func timePtr(t time.Time) *time.Time {
|
|
return &t
|
|
}
|