c60cc7ca75
Some checks failed
CI / Run tests and lints (push) Failing after 29s
CI / Build CLI docs (push) Has been skipped
CI / Run Playwright end-to-end tests (mysql) (push) Failing after 3s
CI / Run Playwright end-to-end tests (postgres) (push) Failing after 2s
CI / Run Playwright end-to-end tests (sqlite) (push) Failing after 3s
CodeQL / Analyze (go) (push) Failing after 4s
CodeQL / Analyze (javascript) (push) Failing after 3s
Docker Image Scanners / scanners (push) Failing after 4s
Synchronize Issue Labels / Synchronize Issue Labels (push) Has been skipped
CI / Run end-to-end tests (postgres) (push) Failing after 19s
CI / Run end-to-end tests (sqlite) (push) Failing after 2s
CI / Run Playwright end-to-end tests (cockroach) (push) Failing after 2s
CI / Draft newsletter (push) Has been skipped
CI / Generate release (push) Has been skipped
CI / Pending approval Slack notification (push) Has been skipped
CI / Send newsletter (push) Has been skipped
46 lines
1.3 KiB
Go
46 lines
1.3 KiB
Go
// Copyright © 2023 Ory Corp
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package testhelpers
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/gofrs/uuid"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/ory/kratos/persistence"
|
|
"github.com/ory/x/networkx"
|
|
)
|
|
|
|
func NewNetworkUnlessExisting(t *testing.T, ctx context.Context, p persistence.Persister) (uuid.UUID, persistence.Persister) {
|
|
if n, ok := ctx.Value("network").(*networkx.Network); ok && n != nil {
|
|
return n.ID, p.WithNetworkID(n.ID)
|
|
}
|
|
|
|
n := networkx.NewNetwork()
|
|
require.NoError(t, p.GetConnection(ctx).Create(n))
|
|
return n.ID, p.WithNetworkID(n.ID)
|
|
}
|
|
|
|
func NewNetwork(t testing.TB, ctx context.Context, p persistence.Persister) (uuid.UUID, persistence.Persister) {
|
|
SkipIfNetworkContext(t, ctx)
|
|
|
|
n := networkx.NewNetwork()
|
|
require.NoError(t, p.GetConnection(context.Background()).Create(n))
|
|
return n.ID, p.WithNetworkID(n.ID)
|
|
}
|
|
|
|
func ExistingNetwork(t *testing.T, p persistence.Persister, id uuid.UUID) persistence.Persister {
|
|
require.NoError(t, p.GetConnection(context.Background()).Save(&networkx.Network{ID: id}))
|
|
return p.WithNetworkID(id)
|
|
}
|
|
|
|
func SkipIfNetworkContext(t testing.TB, ctx context.Context) {
|
|
a, b := ctx.Value("network").(*networkx.Network)
|
|
if a != nil && b {
|
|
t.Skip("Network was set in context")
|
|
}
|
|
}
|