127 lines
2.6 KiB
Go
127 lines
2.6 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
"tss-rocks-be/internal/config"
|
|
"tss-rocks-be/internal/server"
|
|
)
|
|
|
|
func TestConfigLoad(t *testing.T) {
|
|
// Create a temporary config file for testing
|
|
tmpConfig := `
|
|
database:
|
|
driver: sqlite3
|
|
dsn: ":memory:"
|
|
server:
|
|
port: 8080
|
|
host: localhost
|
|
storage:
|
|
type: local
|
|
local:
|
|
root_dir: "./testdata"
|
|
`
|
|
tmpFile, err := os.CreateTemp("", "config-*.yaml")
|
|
require.NoError(t, err)
|
|
defer os.Remove(tmpFile.Name())
|
|
|
|
_, err = tmpFile.WriteString(tmpConfig)
|
|
require.NoError(t, err)
|
|
err = tmpFile.Close()
|
|
require.NoError(t, err)
|
|
|
|
// Test config loading
|
|
cfg, err := config.Load(tmpFile.Name())
|
|
require.NoError(t, err)
|
|
assert.Equal(t, "sqlite3", cfg.Database.Driver)
|
|
assert.Equal(t, ":memory:", cfg.Database.DSN)
|
|
assert.Equal(t, 8080, cfg.Server.Port)
|
|
assert.Equal(t, "localhost", cfg.Server.Host)
|
|
assert.Equal(t, "local", cfg.Storage.Type)
|
|
assert.Equal(t, "./testdata", cfg.Storage.Local.RootDir)
|
|
}
|
|
|
|
func TestServerCreation(t *testing.T) {
|
|
cfg := &config.Config{
|
|
Database: config.DatabaseConfig{
|
|
Driver: "sqlite3",
|
|
DSN: ":memory:",
|
|
},
|
|
Server: config.ServerConfig{
|
|
Port: 8080,
|
|
Host: "localhost",
|
|
},
|
|
Storage: config.StorageConfig{
|
|
Type: "local",
|
|
Local: config.LocalStorage{
|
|
RootDir: "./testdata",
|
|
},
|
|
},
|
|
}
|
|
|
|
// Create ent client
|
|
client := server.NewEntClient(cfg)
|
|
require.NotNil(t, client)
|
|
defer client.Close()
|
|
|
|
// Test schema creation
|
|
err := client.Schema.Create(context.Background())
|
|
require.NoError(t, err)
|
|
|
|
// Test server creation
|
|
srv, err := server.New(cfg, client)
|
|
require.NoError(t, err)
|
|
require.NotNil(t, srv)
|
|
}
|
|
|
|
func TestServerStartAndShutdown(t *testing.T) {
|
|
cfg := &config.Config{
|
|
Database: config.DatabaseConfig{
|
|
Driver: "sqlite3",
|
|
DSN: ":memory:",
|
|
},
|
|
Server: config.ServerConfig{
|
|
Port: 0, // Use random available port
|
|
Host: "localhost",
|
|
},
|
|
Storage: config.StorageConfig{
|
|
Type: "local",
|
|
Local: config.LocalStorage{
|
|
RootDir: "./testdata",
|
|
},
|
|
},
|
|
}
|
|
|
|
client := server.NewEntClient(cfg)
|
|
require.NotNil(t, client)
|
|
defer client.Close()
|
|
|
|
err := client.Schema.Create(context.Background())
|
|
require.NoError(t, err)
|
|
|
|
srv, err := server.New(cfg, client)
|
|
require.NoError(t, err)
|
|
|
|
// Start server in goroutine
|
|
go func() {
|
|
err := srv.Start()
|
|
if err != nil {
|
|
t.Logf("Server stopped: %v", err)
|
|
}
|
|
}()
|
|
|
|
// Give server time to start
|
|
time.Sleep(100 * time.Millisecond)
|
|
|
|
// Test graceful shutdown
|
|
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
|
defer cancel()
|
|
|
|
err = srv.Shutdown(ctx)
|
|
assert.NoError(t, err)
|
|
}
|