mirror of
https://github.com/hashicorp/vault.git
synced 2025-08-13 01:57:03 +02:00
118 lines
3.1 KiB
Go
118 lines
3.1 KiB
Go
package dbtesting
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/hashicorp/vault/sdk/database/dbplugin/v5"
|
|
)
|
|
|
|
func getRequestTimeout(t *testing.T) time.Duration {
|
|
rawDur := os.Getenv("VAULT_TEST_DATABASE_REQUEST_TIMEOUT")
|
|
if rawDur == "" {
|
|
// Note: we incremented the default timeout from 5 to 10 seconds in a bid
|
|
// to fix sporadic failures of mssql_test.go tests TestInitialize() and
|
|
// TestUpdateUser_password().
|
|
|
|
return 10 * time.Second
|
|
}
|
|
|
|
dur, err := time.ParseDuration(rawDur)
|
|
if err != nil {
|
|
t.Fatalf("Failed to parse custom request timeout %q: %s", rawDur, err)
|
|
}
|
|
return dur
|
|
}
|
|
|
|
// AssertInitializeCircleCiTest help to diagnose CircleCI failures within AssertInitialize for mssql tests failing
|
|
// with "Failed to initialize: error verifying connection ...". This will now mark a test as failed instead of being fatal
|
|
func AssertInitializeCircleCiTest(t *testing.T, db dbplugin.Database, req dbplugin.InitializeRequest) dbplugin.InitializeResponse {
|
|
t.Helper()
|
|
maxAttempts := 5
|
|
var resp dbplugin.InitializeResponse
|
|
var err error
|
|
|
|
for i := 1; i <= maxAttempts; i++ {
|
|
resp, err = verifyInitialize(t, db, req)
|
|
if err != nil {
|
|
t.Errorf("Failed AssertInitialize attempt: %d with error:\n%+v\n", i, err)
|
|
time.Sleep(1 * time.Second)
|
|
continue
|
|
}
|
|
|
|
if i > 1 {
|
|
t.Logf("AssertInitialize worked the %d time around with a 1 second sleep", i)
|
|
}
|
|
break
|
|
}
|
|
|
|
return resp
|
|
}
|
|
|
|
func AssertInitialize(t *testing.T, db dbplugin.Database, req dbplugin.InitializeRequest) dbplugin.InitializeResponse {
|
|
t.Helper()
|
|
resp, err := verifyInitialize(t, db, req)
|
|
if err != nil {
|
|
t.Fatalf("Failed to initialize: %s", err)
|
|
}
|
|
return resp
|
|
}
|
|
|
|
func verifyInitialize(t *testing.T, db dbplugin.Database, req dbplugin.InitializeRequest) (dbplugin.InitializeResponse, error) {
|
|
ctx, cancel := context.WithTimeout(context.Background(), getRequestTimeout(t))
|
|
defer cancel()
|
|
|
|
return db.Initialize(ctx, req)
|
|
}
|
|
|
|
func AssertNewUser(t *testing.T, db dbplugin.Database, req dbplugin.NewUserRequest) dbplugin.NewUserResponse {
|
|
t.Helper()
|
|
|
|
ctx, cancel := context.WithTimeout(context.Background(), getRequestTimeout(t))
|
|
defer cancel()
|
|
|
|
resp, err := db.NewUser(ctx, req)
|
|
if err != nil {
|
|
t.Fatalf("Failed to create new user: %s", err)
|
|
}
|
|
|
|
if resp.Username == "" {
|
|
t.Fatalf("Missing username from NewUser response")
|
|
}
|
|
return resp
|
|
}
|
|
|
|
func AssertUpdateUser(t *testing.T, db dbplugin.Database, req dbplugin.UpdateUserRequest) {
|
|
t.Helper()
|
|
|
|
ctx, cancel := context.WithTimeout(context.Background(), getRequestTimeout(t))
|
|
defer cancel()
|
|
|
|
_, err := db.UpdateUser(ctx, req)
|
|
if err != nil {
|
|
t.Fatalf("Failed to update user: %s", err)
|
|
}
|
|
}
|
|
|
|
func AssertDeleteUser(t *testing.T, db dbplugin.Database, req dbplugin.DeleteUserRequest) {
|
|
t.Helper()
|
|
|
|
ctx, cancel := context.WithTimeout(context.Background(), getRequestTimeout(t))
|
|
defer cancel()
|
|
|
|
_, err := db.DeleteUser(ctx, req)
|
|
if err != nil {
|
|
t.Fatalf("Failed to delete user %q: %s", req.Username, err)
|
|
}
|
|
}
|
|
|
|
func AssertClose(t *testing.T, db dbplugin.Database) {
|
|
t.Helper()
|
|
err := db.Close()
|
|
if err != nil {
|
|
t.Fatalf("Failed to close database: %s", err)
|
|
}
|
|
}
|