Jay Brown
44 lines
1.2 KiB
Go
44 lines
1.2 KiB
Go
// Package collector_test contains tests for the collector package.
|
|
// Note: MinTextVersion has been removed since text extraction is deprecated.
|
|
package collector_test
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"queryorchestration/internal/collector"
|
|
"queryorchestration/internal/database/repository"
|
|
"queryorchestration/internal/serviceconfig"
|
|
|
|
"github.com/pashagolub/pgxmock/v3"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestGetByClientID(t *testing.T) {
|
|
ctx := t.Context()
|
|
|
|
pool, err := pgxmock.NewPool()
|
|
require.NoError(t, err)
|
|
cfg := &serviceconfig.BaseConfig{}
|
|
cfg.DBPool = pool
|
|
cfg.DBQueries = repository.New(pool)
|
|
|
|
svc := collector.New(cfg)
|
|
|
|
minCleanV := int64(2)
|
|
ogc := collector.Collector{
|
|
ClientID: "hi",
|
|
MinCleanVersion: minCleanV,
|
|
}
|
|
|
|
pool.ExpectQuery("name: GetCollectorByClientID :one").WithArgs(ogc.ClientID).
|
|
WillReturnRows(
|
|
pgxmock.NewRows([]string{"clientId", "minCleanVersion", "activeVersion", "latestVersion"}).
|
|
AddRow(ogc.ClientID, ogc.MinCleanVersion, ogc.ActiveVersion, ogc.LatestVersion),
|
|
)
|
|
|
|
coll, err := svc.GetByClientID(ctx, ogc.ClientID)
|
|
require.NoError(t, err)
|
|
assert.EqualExportedValues(t, ogc, *coll)
|
|
}
|