378b863b2b
first, so that they all work the same way, because it's getting complicated, and second, so that we can do the appropriate thing instead of CREATE SCHEMA for cockroachdb. Change-Id: I27fbaeeb6223a3e06d97bcf692a2d014b31465f7
116 lines
2.8 KiB
Go
116 lines
2.8 KiB
Go
// Copyright (C) 2019 Storj Labs, Inc.
|
|
// See LICENSE for copying information.
|
|
|
|
package pgutil_test
|
|
|
|
import (
|
|
"testing"
|
|
|
|
_ "github.com/lib/pq"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"storj.io/storj/private/dbutil/dbschema"
|
|
"storj.io/storj/private/dbutil/pgutil"
|
|
"storj.io/storj/private/dbutil/pgutil/pgtest"
|
|
"storj.io/storj/private/dbutil/tempdb"
|
|
"storj.io/storj/private/testcontext"
|
|
)
|
|
|
|
const (
|
|
// DefaultPostgresConn is a connstring that works with docker-compose
|
|
DefaultPostgresConn = "postgres://storj:storj-pass@test-postgres/teststorj?sslmode=disable"
|
|
)
|
|
|
|
func TestQueryPostgres(t *testing.T) {
|
|
if *pgtest.ConnStr == "" {
|
|
t.Skip("Postgres flag missing, example: -postgres-test-db=" + DefaultPostgresConn)
|
|
}
|
|
|
|
doQueryTest(t, *pgtest.ConnStr)
|
|
}
|
|
|
|
func TestQueryCockroach(t *testing.T) {
|
|
if *pgtest.CrdbConnStr == "" {
|
|
t.Skip("Cockroach flag missing, example: -cockroach-test-db=" + pgtest.DefaultCrdbConnStr)
|
|
}
|
|
|
|
doQueryTest(t, *pgtest.CrdbConnStr)
|
|
}
|
|
|
|
func doQueryTest(t *testing.T, connStr string) {
|
|
ctx := testcontext.New(t)
|
|
defer ctx.Cleanup()
|
|
|
|
db, err := tempdb.OpenUnique(connStr, "pgutil-query")
|
|
require.NoError(t, err)
|
|
defer ctx.Check(db.Close)
|
|
|
|
emptySchema, err := pgutil.QuerySchema(db)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, &dbschema.Schema{}, emptySchema)
|
|
|
|
_, err = db.Exec(`
|
|
CREATE TABLE users (
|
|
a bigint NOT NULL,
|
|
b bigint NOT NULL,
|
|
c text,
|
|
UNIQUE (c),
|
|
PRIMARY KEY (a)
|
|
);
|
|
CREATE TABLE names (
|
|
users_a bigint REFERENCES users( a ) ON DELETE CASCADE,
|
|
a text NOT NULL,
|
|
x text,
|
|
b text,
|
|
PRIMARY KEY (a, x),
|
|
UNIQUE ( x ),
|
|
UNIQUE ( a, b )
|
|
);
|
|
`)
|
|
require.NoError(t, err)
|
|
|
|
schema, err := pgutil.QuerySchema(db)
|
|
require.NoError(t, err)
|
|
|
|
expected := &dbschema.Schema{
|
|
Tables: []*dbschema.Table{
|
|
{
|
|
Name: "users",
|
|
Columns: []*dbschema.Column{
|
|
{Name: "a", Type: "bigint", IsNullable: false, Reference: nil},
|
|
{Name: "b", Type: "bigint", IsNullable: false, Reference: nil},
|
|
{Name: "c", Type: "text", IsNullable: true, Reference: nil},
|
|
},
|
|
PrimaryKey: []string{"a"},
|
|
Unique: [][]string{
|
|
{"c"},
|
|
},
|
|
},
|
|
{
|
|
Name: "names",
|
|
Columns: []*dbschema.Column{
|
|
{Name: "users_a", Type: "bigint", IsNullable: true,
|
|
Reference: &dbschema.Reference{
|
|
Table: "users",
|
|
Column: "a",
|
|
OnDelete: "CASCADE",
|
|
}},
|
|
{Name: "a", Type: "text", IsNullable: false, Reference: nil},
|
|
{Name: "x", Type: "text", IsNullable: false, Reference: nil}, // not null, because primary key
|
|
{Name: "b", Type: "text", IsNullable: true, Reference: nil},
|
|
},
|
|
PrimaryKey: []string{"a", "x"},
|
|
Unique: [][]string{
|
|
{"a", "b"},
|
|
{"x"},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
expected.Sort()
|
|
schema.Sort()
|
|
assert.Equal(t, expected, schema)
|
|
}
|