474074480a
The enrichment_attempts column dropped its default after backfill, so inserts must supply it explicitly. The e2e seed helper omitted the column and tripped the NOT NULL constraint, failing TestThirdParty_ImportFromCommon and TestCommonThirdParties_QueryWithLogo. Set it to 0 in the seed insert. Signed-off-by: Émile Ré <emile@probo.com>
118 lines
3.3 KiB
Go
118 lines
3.3 KiB
Go
// Copyright (c) 2026 Probo Inc <hello@probo.com>.
|
|
//
|
|
// Permission to use, copy, modify, and/or distribute this software for any
|
|
// purpose with or without fee is hereby granted, provided that the above
|
|
// copyright notice and this permission notice appear in all copies.
|
|
//
|
|
// THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES WITH
|
|
// REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
|
|
// AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY SPECIAL, DIRECT,
|
|
// INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM
|
|
// LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR
|
|
// OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
|
|
// PERFORMANCE OF THIS SOFTWARE.
|
|
|
|
package console_test
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/jackc/pgx/v5"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
"go.probo.inc/probo/e2e/internal/factory"
|
|
"go.probo.inc/probo/e2e/internal/testutil"
|
|
"go.probo.inc/probo/pkg/coredata"
|
|
"go.probo.inc/probo/pkg/gid"
|
|
)
|
|
|
|
func TestCommonThirdParties_QueryWithLogo(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
owner := testutil.NewClient(t, testutil.RoleOwner)
|
|
|
|
name := factory.SafeName("CommonTP")
|
|
id := seedCommonThirdParty(t, name)
|
|
|
|
const query = `
|
|
query($name: String!) {
|
|
commonThirdParties(name: $name) {
|
|
id
|
|
name
|
|
logo {
|
|
downloadUrl
|
|
}
|
|
}
|
|
}
|
|
`
|
|
|
|
var result struct {
|
|
CommonThirdParties []struct {
|
|
ID string `json:"id"`
|
|
Name string `json:"name"`
|
|
Logo *struct {
|
|
DownloadURL string `json:"downloadUrl"`
|
|
} `json:"logo"`
|
|
} `json:"commonThirdParties"`
|
|
}
|
|
|
|
err := owner.Execute(query, map[string]any{"name": name}, &result)
|
|
require.NoError(t, err, "querying commonThirdParties.logo must not surface a resource-not-found error")
|
|
require.Len(t, result.CommonThirdParties, 1)
|
|
assert.Equal(t, id.String(), result.CommonThirdParties[0].ID)
|
|
assert.Equal(t, name, result.CommonThirdParties[0].Name)
|
|
assert.Nil(t, result.CommonThirdParties[0].Logo)
|
|
}
|
|
|
|
func seedCommonThirdParty(t *testing.T, name string) gid.GID {
|
|
t.Helper()
|
|
|
|
ctx := context.Background()
|
|
conn := dialTestPg(t, ctx)
|
|
t.Cleanup(func() { _ = conn.Close(ctx) })
|
|
|
|
id := gid.New(gid.NilTenant, coredata.CommonThirdPartyEntityType)
|
|
slug := "e2e-" + id.String()
|
|
now := time.Now().UTC()
|
|
|
|
_, err := conn.Exec(ctx, `
|
|
INSERT INTO common_third_parties (
|
|
id, name, slug, category, certifications, enrichment_attempts, created_at, updated_at
|
|
) VALUES (
|
|
$1, $2, $3, $4, $5, $6, $7, $8
|
|
)
|
|
`, id, name, slug, "OTHER", []string{}, 0, now, now)
|
|
require.NoError(t, err)
|
|
|
|
t.Cleanup(func() {
|
|
cleanupCtx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
|
defer cancel()
|
|
|
|
cleanupConn := dialTestPg(t, cleanupCtx)
|
|
|
|
defer func() { _ = cleanupConn.Close(cleanupCtx) }()
|
|
|
|
_, err := cleanupConn.Exec(cleanupCtx, `DELETE FROM common_third_parties WHERE id = $1`, id)
|
|
assert.NoError(t, err, "cleanup: cannot delete seeded common third party %s", id)
|
|
})
|
|
|
|
return id
|
|
}
|
|
|
|
func dialTestPg(t *testing.T, ctx context.Context) *pgx.Conn {
|
|
t.Helper()
|
|
|
|
dsn := os.Getenv("PROBO_E2E_PG_URL")
|
|
if dsn == "" {
|
|
dsn = "postgres://probod:probod@localhost:5432/probod_test?sslmode=disable"
|
|
}
|
|
|
|
conn, err := pgx.Connect(ctx, dsn)
|
|
require.NoError(t, err, "cannot connect to e2e test database")
|
|
|
|
return conn
|
|
}
|