Jeff Mitchell
c1a1f8bb1b
This also then removes support for docker-based containing starting (for now) from platforms other than windows/darwin/linux as there are build issues on several.
92 lines
2.3 KiB
Go
92 lines
2.3 KiB
Go
package oplog
|
|
|
|
import (
|
|
"crypto/rand"
|
|
"testing"
|
|
|
|
"github.com/golang-migrate/migrate/v4"
|
|
"github.com/hashicorp/boundary/internal/docker"
|
|
"github.com/hashicorp/boundary/internal/oplog/oplog_test"
|
|
wrapping "github.com/hashicorp/go-kms-wrapping"
|
|
"github.com/hashicorp/go-kms-wrapping/wrappers/aead"
|
|
"github.com/hashicorp/go-uuid"
|
|
"github.com/jinzhu/gorm"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func testCleanup(t *testing.T, cleanupFunc func() error, db *gorm.DB) {
|
|
t.Helper()
|
|
err := cleanupFunc()
|
|
assert.NoError(t, err)
|
|
err = db.Close()
|
|
assert.NoError(t, err)
|
|
}
|
|
|
|
func testUser(t *testing.T, db *gorm.DB, name, phoneNumber, email string) *oplog_test.TestUser {
|
|
t.Helper()
|
|
u := &oplog_test.TestUser{
|
|
Name: name,
|
|
PhoneNumber: phoneNumber,
|
|
Email: email,
|
|
}
|
|
w := GormWriter{db}
|
|
|
|
err := w.Create(&u)
|
|
require.NoError(t, err)
|
|
return u
|
|
}
|
|
|
|
func testFindUser(t *testing.T, db *gorm.DB, userId uint32) *oplog_test.TestUser {
|
|
t.Helper()
|
|
var foundUser oplog_test.TestUser
|
|
err := db.Where("id = ?", userId).First(&foundUser).Error
|
|
require.NoError(t, err)
|
|
return &foundUser
|
|
}
|
|
|
|
func testId(t *testing.T) string {
|
|
t.Helper()
|
|
id, err := uuid.GenerateUUID()
|
|
require.NoError(t, err)
|
|
return id
|
|
}
|
|
|
|
func testInitDbInDocker(t *testing.T) (cleanup func() error, retURL string, err error) {
|
|
t.Helper()
|
|
cleanup, retURL, _, err = docker.StartDbInDocker("postgres")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
testInitStore(t, cleanup, retURL)
|
|
return
|
|
}
|
|
|
|
// testWrapper initializes an AEAD wrapping.Wrapper for testing the oplog
|
|
func testWrapper(t *testing.T) wrapping.Wrapper {
|
|
t.Helper()
|
|
rootKey := make([]byte, 32)
|
|
n, err := rand.Read(rootKey)
|
|
require.NoError(t, err)
|
|
require.Equal(t, n, 32)
|
|
root := aead.NewWrapper(nil)
|
|
err = root.SetAESGCMKeyBytes(rootKey)
|
|
require.NoError(t, err)
|
|
return root
|
|
}
|
|
|
|
// testInitStore will execute the migrations needed to initialize the store for tests
|
|
func testInitStore(t *testing.T, cleanup func() error, url string) {
|
|
t.Helper()
|
|
// run migrations
|
|
m, err := migrate.New("file://../db/migrations/postgres", url)
|
|
require.NoError(t, err, "Error creating migrations")
|
|
|
|
if err := m.Up(); err != nil && err != migrate.ErrNoChange {
|
|
if err := cleanup(); err != nil {
|
|
t.Fatalf("error cleaning up after migration failure: %v", err)
|
|
}
|
|
require.NoError(t, err, "Error running migrations")
|
|
}
|
|
}
|