mcr/internal/grpcserver/registry_test.go

package grpcserver

import (
	"context"
	"testing"

	"google.golang.org/grpc/codes"
	"google.golang.org/grpc/status"

	pb "git.wntrmute.dev/kyle/mcr/gen/mcr/v1"
	"git.wntrmute.dev/kyle/mcr/internal/auth"
)

func adminDeps(t *testing.T) Deps {
	t.Helper()
	return Deps{
		DB: openTestDB(t),
		Validator: &fakeValidator{
			claims: &auth.Claims{Subject: "admin-uuid", AccountType: "human", Roles: []string{"admin"}},
		},
	}
}

func adminCtx() context.Context {
	return withAuth(context.Background(), "admin-token")
}

func TestListRepositoriesEmpty(t *testing.T) {
	deps := adminDeps(t)
	cc := startTestServer(t, deps)
	client := pb.NewRegistryServiceClient(cc)

	resp, err := client.ListRepositories(adminCtx(), &pb.ListRepositoriesRequest{})
	if err != nil {
		t.Fatalf("ListRepositories: %v", err)
	}
	if len(resp.GetRepositories()) != 0 {
		t.Fatalf("expected 0 repos, got %d", len(resp.Repositories))
	}
}

func TestGetRepositoryNotFound(t *testing.T) {
	deps := adminDeps(t)
	cc := startTestServer(t, deps)
	client := pb.NewRegistryServiceClient(cc)

	_, err := client.GetRepository(adminCtx(), &pb.GetRepositoryRequest{Name: "nonexistent"})
	if err == nil {
		t.Fatal("expected error for nonexistent repo")
	}
	st, ok := status.FromError(err)
	if !ok {
		t.Fatalf("expected gRPC status, got %v", err)
	}
	if st.Code() != codes.NotFound {
		t.Fatalf("code: got %v, want NotFound", st.Code())
	}
}

func TestGetRepositoryEmptyName(t *testing.T) {
	deps := adminDeps(t)
	cc := startTestServer(t, deps)
	client := pb.NewRegistryServiceClient(cc)

	_, err := client.GetRepository(adminCtx(), &pb.GetRepositoryRequest{})
	if err == nil {
		t.Fatal("expected error for empty name")
	}
	st, ok := status.FromError(err)
	if !ok {
		t.Fatalf("expected gRPC status, got %v", err)
	}
	if st.Code() != codes.InvalidArgument {
		t.Fatalf("code: got %v, want InvalidArgument", st.Code())
	}
}

func TestDeleteRepositoryNotFound(t *testing.T) {
	deps := adminDeps(t)
	cc := startTestServer(t, deps)
	client := pb.NewRegistryServiceClient(cc)

	_, err := client.DeleteRepository(adminCtx(), &pb.DeleteRepositoryRequest{Name: "nonexistent"})
	if err == nil {
		t.Fatal("expected error for nonexistent repo")
	}
	st, ok := status.FromError(err)
	if !ok {
		t.Fatalf("expected gRPC status, got %v", err)
	}
	if st.Code() != codes.NotFound {
		t.Fatalf("code: got %v, want NotFound", st.Code())
	}
}

func TestDeleteRepositoryEmptyName(t *testing.T) {
	deps := adminDeps(t)
	cc := startTestServer(t, deps)
	client := pb.NewRegistryServiceClient(cc)

	_, err := client.DeleteRepository(adminCtx(), &pb.DeleteRepositoryRequest{})
	if err == nil {
		t.Fatal("expected error for empty name")
	}
	st, ok := status.FromError(err)
	if !ok {
		t.Fatalf("expected gRPC status, got %v", err)
	}
	if st.Code() != codes.InvalidArgument {
		t.Fatalf("code: got %v, want InvalidArgument", st.Code())
	}
}

func TestGCStatusInitial(t *testing.T) {
	deps := adminDeps(t)
	cc := startTestServer(t, deps)
	client := pb.NewRegistryServiceClient(cc)

	resp, err := client.GetGCStatus(adminCtx(), &pb.GetGCStatusRequest{})
	if err != nil {
		t.Fatalf("GetGCStatus: %v", err)
	}
	if resp.Running {
		t.Fatal("expected running=false on startup")
	}
	if resp.LastRun != nil {
		t.Fatal("expected no last_run on startup")
	}
}

func TestGarbageCollectTrigger(t *testing.T) {
	deps := adminDeps(t)
	cc := startTestServer(t, deps)
	client := pb.NewRegistryServiceClient(cc)

	// Trigger GC without a collector (no-op but should return an ID).
	resp, err := client.GarbageCollect(adminCtx(), &pb.GarbageCollectRequest{})
	if err != nil {
		t.Fatalf("GarbageCollect: %v", err)
	}
	if resp.GetId() == "" {
		t.Fatal("expected non-empty GC ID")
	}
}