snapshot of https://github.com/ory/kratos.git (master, 2026-09-15; blobs > 131072B omitted)

pkg/testhelpers/network.go

@@ -0,0 +1,45 @@
+// 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")
+	}
+}