This commit is contained in:
jay brown
2025-08-04 10:54:22 -07:00
parent 582e5b77c0
commit 56fb9b4ab6
20 changed files with 1026 additions and 28 deletions
+23 -11
View File
@@ -1,12 +1,14 @@
package repository_test
import (
"fmt"
"testing"
"queryorchestration/internal/database/repository"
"queryorchestration/internal/serviceconfig"
"queryorchestration/internal/test"
"github.com/google/uuid"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
)
@@ -24,7 +26,7 @@ func TestDocument(t *testing.T) {
queries := cfg.GetDBQueries()
clientId := "EXAMPLE"
clientId := fmt.Sprintf("EXAMPLE_%s", uuid.New().String()[:8])
err := queries.CreateClient(ctx, &repository.CreateClientParams{
Name: "example_client",
Clientid: clientId,
@@ -44,7 +46,7 @@ func TestDocument(t *testing.T) {
queries := cfg.GetDBQueries()
clientId := "EXAMPLE"
clientId := fmt.Sprintf("EXAMPLE_%s", uuid.New().String()[:8])
err := queries.CreateClient(ctx, &repository.CreateClientParams{
Name: "example_client",
Clientid: clientId,
@@ -55,6 +57,7 @@ func TestDocument(t *testing.T) {
id, err := queries.CreateDocument(ctx, &repository.CreateDocumentParams{
Clientid: clientId,
Hash: hash,
BatchID: nil,
})
require.NoError(t, err)
assert.NotEmpty(t, id)
@@ -74,7 +77,7 @@ func TestDocument(t *testing.T) {
queries := cfg.GetDBQueries()
clientId := "EXAMPLE"
clientId := fmt.Sprintf("EXAMPLE_%s", uuid.New().String()[:8])
err := queries.CreateClient(ctx, &repository.CreateClientParams{
Name: "example_client",
Clientid: clientId,
@@ -85,6 +88,7 @@ func TestDocument(t *testing.T) {
id, err := queries.CreateDocument(ctx, &repository.CreateDocumentParams{
Clientid: clientId,
Hash: hash,
BatchID: nil,
})
require.NoError(t, err)
assert.NotEmpty(t, id)
@@ -92,6 +96,7 @@ func TestDocument(t *testing.T) {
documentTwoID, err := queries.CreateDocument(ctx, &repository.CreateDocumentParams{
Clientid: clientId,
Hash: "example_hash_two",
BatchID: nil,
})
require.NoError(t, err)
assert.NotEmpty(t, documentTwoID)
@@ -109,7 +114,7 @@ func TestDocument(t *testing.T) {
queries := cfg.GetDBQueries()
clientId := "EXAMPLE"
clientId := fmt.Sprintf("EXAMPLE_%s", uuid.New().String()[:8])
err := queries.CreateClient(ctx, &repository.CreateClientParams{
Name: "example_client",
Clientid: clientId,
@@ -120,11 +125,12 @@ func TestDocument(t *testing.T) {
id, err := queries.CreateDocument(ctx, &repository.CreateDocumentParams{
Clientid: clientId,
Hash: hash,
BatchID: nil,
})
require.NoError(t, err)
assert.NotEmpty(t, id)
clientTwoId := "EXAMPLE TWO"
clientTwoId := fmt.Sprintf("EXAMPLE_TWO_%s", uuid.New().String()[:8])
err = queries.CreateClient(ctx, &repository.CreateClientParams{
Name: "example_name_two",
Clientid: clientTwoId,
@@ -135,6 +141,7 @@ func TestDocument(t *testing.T) {
idTwo, err := queries.CreateDocument(ctx, &repository.CreateDocumentParams{
Clientid: clientTwoId,
Hash: hashTwo,
BatchID: nil,
})
require.NoError(t, err)
assert.NotEmpty(t, idTwo)
@@ -155,7 +162,7 @@ func TestDocument(t *testing.T) {
queries := cfg.GetDBQueries()
clientId := "EXAMPLE"
clientId := fmt.Sprintf("EXAMPLE_%s", uuid.New().String()[:8])
err := queries.CreateClient(ctx, &repository.CreateClientParams{
Name: "example_client",
Clientid: clientId,
@@ -166,13 +173,14 @@ func TestDocument(t *testing.T) {
id, err := queries.CreateDocument(ctx, &repository.CreateDocumentParams{
Clientid: clientId,
Hash: hash,
BatchID: nil,
})
require.NoError(t, err)
assert.NotEmpty(t, id)
doc, err := queries.GetDocumentSummary(ctx, id)
require.NoError(t, err)
assert.EqualExportedValues(t, &repository.Document{
assert.EqualExportedValues(t, &repository.GetDocumentSummaryRow{
ID: id,
Clientid: clientId,
Hash: hash,
@@ -187,7 +195,7 @@ func TestDocument(t *testing.T) {
queries := cfg.GetDBQueries()
clientId := "EXAMPLE"
clientId := fmt.Sprintf("EXAMPLE_%s", uuid.New().String()[:8])
err := queries.CreateClient(ctx, &repository.CreateClientParams{
Name: "example_client",
Clientid: clientId,
@@ -198,6 +206,7 @@ func TestDocument(t *testing.T) {
id, err := queries.CreateDocument(ctx, &repository.CreateDocumentParams{
Clientid: clientId,
Hash: hash,
BatchID: nil,
})
require.NoError(t, err)
assert.NotEmpty(t, id)
@@ -220,7 +229,7 @@ func TestDocument(t *testing.T) {
queries := cfg.GetDBQueries()
clientId := "EXAMPLE"
clientId := fmt.Sprintf("EXAMPLE_%s", uuid.New().String()[:8])
err := queries.CreateClient(ctx, &repository.CreateClientParams{
Name: "example_client",
Clientid: clientId,
@@ -231,6 +240,7 @@ func TestDocument(t *testing.T) {
id, err := queries.CreateDocument(ctx, &repository.CreateDocumentParams{
Clientid: clientId,
Hash: hash,
BatchID: nil,
})
require.NoError(t, err)
assert.NotEmpty(t, id)
@@ -251,7 +261,7 @@ func TestDocument(t *testing.T) {
queries := cfg.GetDBQueries()
clientId := "EXAMPLE"
clientId := fmt.Sprintf("EXAMPLE_%s", uuid.New().String()[:8])
err := queries.CreateClient(ctx, &repository.CreateClientParams{
Name: "example_client",
Clientid: clientId,
@@ -262,6 +272,7 @@ func TestDocument(t *testing.T) {
id, err := queries.CreateDocument(ctx, &repository.CreateDocumentParams{
Clientid: clientId,
Hash: hash,
BatchID: nil,
})
require.NoError(t, err)
assert.NotEmpty(t, id)
@@ -292,7 +303,7 @@ func TestDocument(t *testing.T) {
queries := cfg.GetDBQueries()
clientId := "EXAMPLE"
clientId := fmt.Sprintf("EXAMPLE_%s", uuid.New().String()[:8])
err := queries.CreateClient(ctx, &repository.CreateClientParams{
Name: "example_client",
Clientid: clientId,
@@ -303,6 +314,7 @@ func TestDocument(t *testing.T) {
id, err := queries.CreateDocument(ctx, &repository.CreateDocumentParams{
Clientid: clientId,
Hash: hash,
BatchID: nil,
})
require.NoError(t, err)
assert.NotEmpty(t, id)