XingfenD
4e467e1136
- Config loader with env var expansion and validation - SQLite database with tasks and toolchain_versions tables - Gitea REST API client for repository and branch operations - K8s client for Job lifecycle management (with noop fallback for local dev) - REST API: tasks CRUD, retry, plugins, version management - Worker/scheduler with polling, concurrency control, and job tracking - Static frontend: generate, tasks, settings pages - Toolchain images: Go and Node.js Dockerfiles with entrypoint scripts - Helm chart with deployment, service, configmap, ingress, PVC - Multi-stage Dockerfile for service - Unit tests for config, database, gitclient, k8s, api, worker - E2E tests for idempotency, version switch, failure recovery
125 lines
2.8 KiB
Go
125 lines
2.8 KiB
Go
package database
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"proto-gen/internal/models"
|
|
)
|
|
|
|
func newTestDB(t *testing.T) *DB {
|
|
t.Helper()
|
|
db, err := New(t.TempDir() + "/test.db")
|
|
if err != nil {
|
|
t.Fatalf("creating test db: %v", err)
|
|
}
|
|
t.Cleanup(func() { db.Close() })
|
|
return db
|
|
}
|
|
|
|
func TestNew_CreatesTables(t *testing.T) {
|
|
db := newTestDB(t)
|
|
var count int
|
|
err := db.conn.QueryRow(`SELECT COUNT(*) FROM sqlite_master WHERE type='table' AND name IN ('tasks', 'toolchain_versions')`).Scan(&count)
|
|
if err != nil {
|
|
t.Fatalf("querying tables: %v", err)
|
|
}
|
|
if count != 2 {
|
|
t.Errorf("expected 2 tables, got %d", count)
|
|
}
|
|
}
|
|
|
|
func TestCreateAndGetTask(t *testing.T) {
|
|
db := newTestDB(t)
|
|
task := &models.Task{
|
|
ID: "test-1",
|
|
Status: models.StatusPending,
|
|
Language: "go",
|
|
ProtoRepo: "yoresee_doc",
|
|
ProtoBranch: "main",
|
|
TargetRepo: "yoresee_doc-gen-go",
|
|
ToolchainImage: "proto-gen-go:v1.0",
|
|
}
|
|
if err := db.CreateTask(task); err != nil {
|
|
t.Fatalf("CreateTask: %v", err)
|
|
}
|
|
|
|
got, err := db.GetTask("test-1")
|
|
if err != nil {
|
|
t.Fatalf("GetTask: %v", err)
|
|
}
|
|
if got.Language != "go" {
|
|
t.Errorf("Language = %q, want %q", got.Language, "go")
|
|
}
|
|
if got.Status != models.StatusPending {
|
|
t.Errorf("Status = %q, want %q", got.Status, models.StatusPending)
|
|
}
|
|
}
|
|
|
|
func TestHasRunningTask(t *testing.T) {
|
|
db := newTestDB(t)
|
|
|
|
has, err := db.HasRunningTask("yoresee_doc-gen-go")
|
|
if err != nil {
|
|
t.Fatalf("HasRunningTask: %v", err)
|
|
}
|
|
if has {
|
|
t.Error("expected no running task")
|
|
}
|
|
|
|
task := &models.Task{
|
|
ID: "test-2",
|
|
Status: models.StatusRunning,
|
|
Language: "go",
|
|
ProtoRepo: "yoresee_doc",
|
|
ProtoBranch: "main",
|
|
TargetRepo: "yoresee_doc-gen-go",
|
|
ToolchainImage: "proto-gen-go:v1.0",
|
|
}
|
|
if err := db.CreateTask(task); err != nil {
|
|
t.Fatalf("CreateTask: %v", err)
|
|
}
|
|
|
|
has, err = db.HasRunningTask("yoresee_doc-gen-go")
|
|
if err != nil {
|
|
t.Fatalf("HasRunningTask: %v", err)
|
|
}
|
|
if !has {
|
|
t.Error("expected running task")
|
|
}
|
|
}
|
|
|
|
func TestListTasks(t *testing.T) {
|
|
db := newTestDB(t)
|
|
for i := 0; i < 3; i++ {
|
|
status := models.StatusPending
|
|
if i == 1 {
|
|
status = models.StatusRunning
|
|
}
|
|
db.CreateTask(&models.Task{
|
|
ID: "t-" + string(rune('0'+i)),
|
|
Status: status,
|
|
Language: "go",
|
|
ProtoRepo: "yoresee_doc",
|
|
ProtoBranch: "main",
|
|
TargetRepo: "repo-" + string(rune('0'+i)),
|
|
ToolchainImage: "img:v1",
|
|
})
|
|
}
|
|
|
|
tasks, err := db.ListTasks("", 10, 0)
|
|
if err != nil {
|
|
t.Fatalf("ListTasks: %v", err)
|
|
}
|
|
if len(tasks) != 3 {
|
|
t.Errorf("len(tasks) = %d, want 3", len(tasks))
|
|
}
|
|
|
|
tasks, err = db.ListTasks(string(models.StatusRunning), 10, 0)
|
|
if err != nil {
|
|
t.Fatalf("ListTasks filtered: %v", err)
|
|
}
|
|
if len(tasks) != 1 {
|
|
t.Errorf("len(running) = %d, want 1", len(tasks))
|
|
}
|
|
}
|