Files
query-orchestration/internal/database/repository/client_test.go
T
Michael McGuinness fee71e7740 Merged in feature/postprocessing (pull request #114)
Feature/postprocessing

* tests

* passtest

* fixshorttests

* mosttests

* improvingbasedockerfile

* testspeeds

* testing

* host

* canparallel

* clean

* passfullsuite

* singlepagemax

* test

* findfeatures

* findstables

* tbls

* tablestoo

* tablestoo

* lateraltests

* tableloc

* cleanup

* inlinetable

* childids

* cleanup

* tests
2025-04-22 14:40:16 +00:00

96 lines
2.1 KiB
Go

package repository_test
import (
"context"
"testing"
"queryorchestration/internal/database/repository"
"queryorchestration/internal/serviceconfig"
"queryorchestration/internal/test"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
)
func TestClient(t *testing.T) {
t.Parallel()
if testing.Short() {
t.Skip("Skipping long test in short mode")
}
ctx := context.Background()
cfg := &serviceconfig.BaseConfig{}
net := test.DepNetwork.Get(t, ctx)
_ = test.CreateDB(t, ctx, cfg, net, &test.CreateDatabaseConfig{
RunMigrations: true,
})
queries := cfg.GetDBQueries()
name := "example_name"
clientId := "EXAMPLE"
err := queries.CreateClient(ctx, &repository.CreateClientParams{
Name: name,
ID: clientId,
})
require.NoError(t, err)
client, err := queries.GetClient(ctx, clientId)
require.NoError(t, err)
assert.EqualExportedValues(t, &repository.Fullclient{
ID: clientId,
Name: name,
Cansync: false,
}, client)
name = "updated_client"
err = queries.UpdateClient(ctx, &repository.UpdateClientParams{
ID: clientId,
Name: name,
})
require.NoError(t, err)
client, err = queries.GetClient(ctx, clientId)
require.NoError(t, err)
assert.EqualExportedValues(t, &repository.Fullclient{
ID: clientId,
Name: name,
Cansync: false,
}, client)
err = queries.AddClientCanSync(ctx, &repository.AddClientCanSyncParams{
Clientid: clientId,
Cansync: true,
})
require.NoError(t, err)
client, err = queries.GetClient(ctx, clientId)
require.NoError(t, err)
assert.EqualExportedValues(t, &repository.Fullclient{
ID: clientId,
Name: name,
Cansync: true,
}, client)
err = queries.CreateClient(ctx, &repository.CreateClientParams{
Name: name,
ID: clientId,
})
assert.Error(t, err)
err = queries.CreateClient(ctx, &repository.CreateClientParams{
Name: name,
ID: "definitely different",
})
assert.Error(t, err)
err = queries.CreateClient(ctx, &repository.CreateClientParams{
Name: "100 different",
ID: clientId,
})
assert.Error(t, err)
err = queries.CreateClient(ctx, &repository.CreateClientParams{
Name: "sameid",
ID: "sameid",
})
require.NoError(t, err)
}