hatch-surf/internal/store/sqlite_repo_test.go

package store

import (
	"context"
	"database/sql"
	"testing"
	"time"
)

func openTestRepo(t *testing.T) Repository {
	t.Helper()
	db, err := sql.Open("sqlite", ":memory:?_journal_mode=WAL&_foreign_keys=on")
	if err != nil {
		t.Fatalf("open in-memory sqlite: %v", err)
	}
	db.SetMaxOpenConns(1)
	if err := migrate(db); err != nil {
		db.Close()
		t.Fatalf("migrate: %v", err)
	}
	repo, err := NewSQLiteRepo(db)
	if err != nil {
		db.Close()
		t.Fatalf("new sqlite repo: %v", err)
	}
	t.Cleanup(func() { repo.Close() })
	return repo
}

func TestCreateAndGetEndpoint(t *testing.T) {
	repo := openTestRepo(t)
	e, err := repo.CreateEndpoint(context.Background(), "test-one")
	if err != nil {
		t.Fatalf("CreateEndpoint: %v", err)
	}
	if e.ID == "" {
		t.Error("expected non-empty ID")
	}
	if e.URL != "test-one" {
		t.Errorf("url: got %q", e.URL)
	}
	if e.CreatedAt == "" || e.UpdatedAt == "" {
		t.Error("expected timestamps")
	}
	got, err := repo.GetEndpoint(context.Background(), e.ID)
	if err != nil {
		t.Fatalf("GetEndpoint: %v", err)
	}
	if got.ID != e.ID || got.URL != e.URL {
		t.Errorf("got %+v", got)
	}
}

func TestGetEndpointNotFound(t *testing.T) {
	repo := openTestRepo(t)
	_, err := repo.GetEndpoint(context.Background(), "nonexistent")
	if err == nil {
		t.Fatal("expected error")
	}
}

func TestCreateEndpointDuplicateURL(t *testing.T) {
	repo := openTestRepo(t)
	ctx := context.Background()
	if _, err := repo.CreateEndpoint(ctx, "dup"); err != nil {
		t.Fatal(err)
	}
	if _, err := repo.CreateEndpoint(ctx, "dup"); err == nil {
		t.Fatal("expected UNIQUE error")
	}
}

func TestAppendAndListRequests(t *testing.T) {
	repo := openTestRepo(t)
	ctx := context.Background()
	e, err := repo.CreateEndpoint(ctx, "reqs-test")
	if err != nil {
		t.Fatal(err)
	}
	r1 := &Request{Method: "POST", Path: "/webhook", Headers: `{"Content-Type":"application/json"}`, Query: "foo=bar", Body: []byte(`{"hello":"world"}`)}
	if err := repo.AppendRequest(ctx, e.ID, r1); err != nil {
		t.Fatal(err)
	}
	time.Sleep(time.Millisecond)
	r2 := &Request{Method: "GET", Path: "/webhook", Headers: `{}`}
	if err := repo.AppendRequest(ctx, e.ID, r2); err != nil {
		t.Fatal(err)
	}
	reqs, err := repo.ListRequests(ctx, e.ID, 10)
	if err != nil {
		t.Fatal(err)
	}
	if len(reqs) != 2 {
		t.Fatalf("got %d requests", len(reqs))
	}
	if reqs[0].Method != "GET" {
		t.Errorf("first: got %s", reqs[0].Method)
	}
	if reqs[1].Method != "POST" {
		t.Errorf("second: got %s", reqs[1].Method)
	}
}

func TestGetRequest(t *testing.T) {
	repo := openTestRepo(t)
	ctx := context.Background()
	e, _ := repo.CreateEndpoint(ctx, "getreq-test")
	r := &Request{Method: "POST", Path: "/webhook", Headers: `{"X-Api-Key":"secret"}`, Query: "a=1", Body: []byte(`{"msg":"hi"}`)}
	if err := repo.AppendRequest(ctx, e.ID, r); err != nil {
		t.Fatal(err)
	}
	got, err := repo.GetRequest(ctx, r.ID)
	if err != nil {
		t.Fatalf("GetRequest: %v", err)
	}
	if got.ID != r.ID {
		t.Errorf("id: got %q want %q", got.ID, r.ID)
	}
	if got.Method != "POST" {
		t.Errorf("method: got %q", got.Method)
	}
	if string(got.Body) != `{"msg":"hi"}` {
		t.Errorf("body: got %q", string(got.Body))
	}
}

func TestGetRequestNotFound(t *testing.T) {
	repo := openTestRepo(t)
	_, err := repo.GetRequest(context.Background(), "nonexistent")
	if err == nil {
		t.Fatal("expected error for nonexistent request")
	}
}

func TestSetAndGetMock(t *testing.T) {
	repo := openTestRepo(t)
	ctx := context.Background()
	e, err := repo.CreateEndpoint(ctx, "mock-test")
	if err != nil {
		t.Fatal(err)
	}
	mock := &MockConfig{
		EndpointID: e.ID,
		Status:     201,
		Headers:    map[string]string{"Content-Type": "application/json"},
		Body:       []byte(`{"mocked": true}`),
	}
	if err := repo.SetMock(ctx, mock); err != nil {
		t.Fatal(err)
	}
	got, err := repo.GetMock(ctx, e.ID)
	if err != nil {
		t.Fatal(err)
	}
	if got.Status != 201 {
		t.Errorf("status: got %d", got.Status)
	}
	if string(got.Body) != `{"mocked": true}` {
		t.Errorf("body: got %q", string(got.Body))
	}
}

func TestGetMockNotFound(t *testing.T) {
	repo := openTestRepo(t)
	_, err := repo.GetMock(context.Background(), "no-mock")
	if err == nil {
		t.Fatal("expected error")
	}
}

func TestListRequestsLimit(t *testing.T) {
	repo := openTestRepo(t)
	ctx := context.Background()
	e, _ := repo.CreateEndpoint(ctx, "limit-test")
	for i := 0; i < 5; i++ {
		repo.AppendRequest(ctx, e.ID, &Request{Method: "GET", Path: "/"})
	}
	reqs, err := repo.ListRequests(ctx, e.ID, 3)
	if err != nil {
		t.Fatal(err)
	}
	if len(reqs) != 3 {
		t.Fatalf("got %d", len(reqs))
	}
}

func TestListRequestsEmpty(t *testing.T) {
	repo := openTestRepo(t)
	ctx := context.Background()
	e, _ := repo.CreateEndpoint(ctx, "empty-test")
	reqs, _ := repo.ListRequests(ctx, e.ID, 10)
	if len(reqs) != 0 {
		t.Errorf("got %d", len(reqs))
	}
}

func TestAppendRequestBodyBinary(t *testing.T) {
	repo := openTestRepo(t)
	ctx := context.Background()
	e, _ := repo.CreateEndpoint(ctx, "binary-test")
	body := []byte{0x00, 0x01, 0x02, 0xFF}
	if err := repo.AppendRequest(ctx, e.ID, &Request{Method: "POST", Path: "/binary", Body: body}); err != nil {
		t.Fatal(err)
	}
	reqs, _ := repo.ListRequests(ctx, e.ID, 1)
	if len(reqs) != 1 {
		t.Fatal("expected 1 request")
	}
	if len(reqs[0].Body) != 4 {
		t.Errorf("body len: %d", len(reqs[0].Body))
	}
}

func TestMigrateIdempotent(t *testing.T) {
	db, err := sql.Open("sqlite", ":memory:?_journal_mode=WAL&_foreign_keys=on")
	if err != nil {
		t.Fatal(err)
	}
	defer db.Close()
	db.SetMaxOpenConns(1)
	if err := migrate(db); err != nil {
		t.Fatal(err)
	}
	if err := migrate(db); err != nil {
		t.Fatal(err)
	}
	var name string
	if err := db.QueryRow("SELECT name FROM sqlite_master WHERE type='table' AND name='endpoints'").Scan(&name); err != nil {
		t.Fatalf("endpoints table: %v", err)
	}
	if err := db.QueryRow("SELECT name FROM sqlite_master WHERE type='table' AND name='requests'").Scan(&name); err != nil {
		t.Fatalf("requests table: %v", err)
	}
}

func TestSearchRequests(t *testing.T) {
	repo := openTestRepo(t)
	ctx := context.Background()
	e, _ := repo.CreateEndpoint(ctx, "search-test")

	// Add requests with different content.
	repo.AppendRequest(ctx, e.ID, &Request{Method: "GET", Path: "/users", Headers: `{"Accept":"json"}`})
	repo.AppendRequest(ctx, e.ID, &Request{Method: "POST", Path: "/orders", Body: []byte(`{"item":"apple"}`)})
	repo.AppendRequest(ctx, e.ID, &Request{Method: "DELETE", Path: "/users/123"})
	repo.AppendRequest(ctx, e.ID, &Request{Method: "PUT", Path: "/products", Query: "category=electronics"})

	// Search for "users".
	reqs, err := repo.SearchRequests(ctx, e.ID, "users", 10)
	if err != nil {
		t.Fatal(err)
	}
	if len(reqs) != 2 {
		t.Errorf("expected 2 requests matching 'users', got %d", len(reqs))
	}

	// Search for "apple" (in body).
	reqs, err = repo.SearchRequests(ctx, e.ID, "apple", 10)
	if err != nil {
		t.Fatal(err)
	}
	if len(reqs) != 1 {
		t.Errorf("expected 1 request matching 'apple', got %d", len(reqs))
	}

	// Search for "electronics" (in query).
	reqs, err = repo.SearchRequests(ctx, e.ID, "electronics", 10)
	if err != nil {
		t.Fatal(err)
	}
	if len(reqs) != 1 {
		t.Errorf("expected 1 request matching 'electronics', got %d", len(reqs))
	}

	// Empty query returns all.
	reqs, err = repo.SearchRequests(ctx, e.ID, "", 10)
	if err != nil {
		t.Fatal(err)
	}
	if len(reqs) != 4 {
		t.Errorf("expected 4 requests with empty query, got %d", len(reqs))
	}

	// Search with limit.
	reqs, err = repo.SearchRequests(ctx, e.ID, "users", 1)
	if err != nil {
		t.Fatal(err)
	}
	if len(reqs) != 1 {
		t.Errorf("expected 1 request with limit=1, got %d", len(reqs))
	}
}

func TestOpenInMemory(t *testing.T) {
	repo, err := Open(":memory:")
	if err != nil {
		t.Fatalf("Open: %v", err)
	}
	defer repo.Close()

	// Verify the repo works.
	e, err := repo.CreateEndpoint(context.Background(), "test")
	if err != nil {
		t.Fatalf("CreateEndpoint: %v", err)
	}
	if e.ID != "test" {
		t.Errorf("expected ID 'test', got %q", e.ID)
	}
}

func TestNewSQLiteRepoNilDB(t *testing.T) {
	_, err := NewSQLiteRepo(nil)
	if err == nil {
		t.Fatal("expected error for nil db")
	}
}