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

internal/x/dbx/dsn_testutils_test.go

@@ -0,0 +1,64 @@
+// Copyright © 2023 Ory Corp
+// SPDX-License-Identifier: Apache-2.0
+
+package dbx
+
+import (
+	"testing"
+
+	"github.com/ory/pop/v6"
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+)
+
+//nolint:gosec // Test DSNs with dummy credentials, not real secrets.
+func Test_withDbName(t *testing.T) {
+	type args struct {
+		dsn string
+		db  string
+	}
+	tests := []struct {
+		name string
+		args args
+		want string
+	}{{
+		name: "postgres",
+		args: args{
+			dsn: "postgres://postgres:postgres@localhost:5432/postgres?sslmode=disable",
+			db:  "mydb",
+		},
+		want: "postgres://postgres:postgres@localhost:5432/mydb?sslmode=disable",
+	}, {
+		name: "cockroach",
+		args: args{
+			dsn: "cockroach://root@localhost:49364/defaultdb?sslmode=disable",
+			db:  "foo",
+		},
+		want: "cockroach://root@localhost:49364/foo?sslmode=disable",
+	}, {
+		name: "mysql",
+		args: args{
+			dsn: "mysql://root:secret@(localhost:49394)/mysql?parseTime=true&multiStatements=true",
+			db:  "testdb",
+		},
+		want: "mysql://root:secret@tcp(localhost:49394)/testdb?multiStatements=true&parseTime=true",
+	}}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			assert.Equal(t, tt.want, withDbName(tt.args.dsn, tt.args.db))
+		})
+	}
+}
+
+func Test_GetDSNs_can_connect_to_each_db(t *testing.T) {
+	for _, db := range GetDSNs(t) {
+		t.Run("dsn="+db.Name, func(t *testing.T) {
+			t.Parallel()
+			conn, err := pop.NewConnection(&pop.ConnectionDetails{URL: db.Conn})
+			require.NoError(t, err)
+			assert.NoError(t, conn.Open())
+			assert.NoError(t, Ping(conn))
+			assert.NoError(t, conn.Close())
+		})
+	}
+}