package server

import (
	"context"
	"testing"

	"github.com/stretchr/testify/assert"
	"tss-rocks-be/internal/config"
)

func TestNewEntClient(t *testing.T) {
	tests := []struct {
		name string
		cfg  *config.Config
	}{
		{
			name: "default sqlite3 config",
			cfg: &config.Config{
				Database: config.DatabaseConfig{
					Driver: "sqlite3",
					DSN:    "file:ent?mode=memory&cache=shared&_fk=1",
				},
			},
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			client := NewEntClient(tt.cfg)
			assert.NotNil(t, client)

			// 验证客户端是否可以正常工作
			err := client.Schema.Create(context.Background())
			assert.NoError(t, err)

			// 清理
			client.Close()
		})
	}
}