* refact: Changes required for gorm v2 * refact: Add db.DB wrapper and refact all test fixtures to use it Stop leaking the gorm abstraction within our unit tests. * refact: Migrate direct calls to pq to pgx driver The refactor will remove the remaining direct dependencies on the pq driver which has been deprecated in favor of the pgx driver. It includes the addition of common.SqlOpen which is a wrapper around sql.Open, which is needed to deal with the driver name changing from "postgres" to "pgx" but in our code base the dialect and driver are synonyms.
88 lines
1.9 KiB
Go
88 lines
1.9 KiB
Go
package oplog
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/hashicorp/boundary/internal/db/common"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func Test_testCleanup(t *testing.T) {
|
|
assert := assert.New(t)
|
|
cleanup, db := setup(t)
|
|
testCleanup(t, cleanup, db)
|
|
|
|
err := db.Begin().Error
|
|
assert.Equal("sql: database is closed", err.Error())
|
|
}
|
|
|
|
func Test_testUser(t *testing.T) {
|
|
assert, require := assert.New(t), require.New(t)
|
|
cleanup, db := setup(t)
|
|
defer testCleanup(t, cleanup, db)
|
|
|
|
id := testId(t)
|
|
|
|
u := testUser(t, db, id, id, id)
|
|
require.NotNil(u)
|
|
assert.Equal(id, u.Name)
|
|
assert.Equal(id, u.PhoneNumber)
|
|
assert.Equal(id, u.Email)
|
|
}
|
|
|
|
func Test_testFindUser(t *testing.T) {
|
|
assert, require := assert.New(t), require.New(t)
|
|
cleanup, db := setup(t)
|
|
defer testCleanup(t, cleanup, db)
|
|
id := testId(t)
|
|
u := testUser(t, db, id, id, id)
|
|
require.NotNil(u)
|
|
|
|
found := testFindUser(t, db, u.Id)
|
|
require.NotNil(found)
|
|
assert.Equal(u, found)
|
|
}
|
|
|
|
func Test_testId(t *testing.T) {
|
|
require := require.New(t)
|
|
id := testId(t)
|
|
require.NotNil(id)
|
|
}
|
|
|
|
func Test_testInitDbInDocker(t *testing.T) {
|
|
require := require.New(t)
|
|
cleanup, url, err := testInitDbInDocker(t)
|
|
defer cleanup()
|
|
require.NoError(err)
|
|
require.NotEmpty(url)
|
|
require.NotNil(cleanup)
|
|
}
|
|
|
|
func Test_testWrapper(t *testing.T) {
|
|
w := testWrapper(t)
|
|
require.NotNil(t, w)
|
|
assert.Equal(t, "aead", w.Type())
|
|
}
|
|
|
|
func Test_testInitStore(t *testing.T) {
|
|
assert, require := assert.New(t), require.New(t)
|
|
cleanup, url, err := testInitDbInDocker(t)
|
|
require.NoError(err)
|
|
defer cleanup()
|
|
require.NotEmpty(url)
|
|
|
|
testInitStore(t, cleanup, url)
|
|
|
|
const query = `
|
|
select count(*) from information_schema."tables" t where table_name = 'db_test_user';
|
|
`
|
|
db, err := common.SqlOpen("postgres", url)
|
|
require.NoError(err)
|
|
|
|
var cnt int
|
|
err = db.QueryRow(query).Scan(&cnt)
|
|
require.NoError(err)
|
|
assert.Equal(1, cnt)
|
|
}
|