[feature] migrate to monorepo
This commit is contained in:
commit
05ddc1f783
267 changed files with 75165 additions and 0 deletions
127
backend/cmd/server/main_test.go
Normal file
127
backend/cmd/server/main_test.go
Normal file
|
@ -0,0 +1,127 @@
|
|||
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)
|
||||
}
|
Loading…
Add table
Add a link
Reference in a new issue