package client_test import ( "context" "errors" "queryorchestration/internal/client" "queryorchestration/internal/database" "queryorchestration/internal/database/repository" "testing" "github.com/google/uuid" "github.com/pashagolub/pgxmock/v3" "github.com/stretchr/testify/assert" ) func TestGet(t *testing.T) { ctx := context.Background() pool, err := pgxmock.NewPool() if err != nil { t.Fatalf("failed to open pgxmock database: %v", err) } queries := repository.New(pool) db := &database.Connection{ Queries: queries, Pool: pool, } svc := client.New(db) id := uuid.New() pool.ExpectQuery("name: GetClient :one").WithArgs(database.MustToDBUUID(id)). WillReturnRows( pgxmock.NewRows([]string{"id", "name", "canSync"}). AddRow(database.MustToDBUUID(id), "client_name", false), ) cli, err := svc.Get(ctx, id) assert.Nil(t, err) assert.EqualExportedValues(t, &client.Client{ ID: id, Name: "client_name", CanSync: false, }, cli) dberr := "database failure" pool.ExpectQuery("name: GetClient :one").WithArgs(database.MustToDBUUID(id)). WillReturnError(errors.New(dberr)) _, err = svc.Get(ctx, id) assert.EqualError(t, err, dberr) }