go/testing-go-code-with-postgres
1
0
Fork 0
mirror of https://github.com/xorcare/testing-go-code-with-postgres.git synced 2025-06-30 23:23:40 +02:00
Code Issues Releases Activity

Add example of tests with transactional cleanup

Browse Source
This commit is contained in:
Vasiliy Vasilyuk
2024-02-17 18:12:44 +03:00
parent 13e3596692
commit 05d30f6c09
6 changed files with 221 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

57
testingpg/testingpg_test.go
View File

@ -69,9 +69,9 @@ func TestNewPostgres(t *testing.T) {
ctx := context.Background()
// Act
const sql = `CREATE TABLE "no_conflict" (id integer PRIMARY KEY)`
_, err1 := postgres1.DB().ExecContext(ctx, sql)
_, err2 := postgres2.DB().ExecContext(ctx, sql)
const sqlStr = `CREATE TABLE "no_conflict" (id integer PRIMARY KEY)`
_, err1 := postgres1.DB().ExecContext(ctx, sqlStr)
_, err2 := postgres2.DB().ExecContext(ctx, sqlStr)
// Assert
require.NoError(t, err1)
@ -93,3 +93,54 @@ func TestNewPostgres(t *testing.T) {
require.NotEqual(t, url1, url2)
})
}
func TestNewWithTransactionalCleanup(t *testing.T) {
if testing.Short() {
t.Skip("skipping test in short mode")
}
t.Parallel()
t.Run("Successfully obtained a version", func(t *testing.T) {
t.Parallel()
// Arrange
tx := testingpg.NewWithTransactionalCleanup(t)
ctx := context.Background()
// Act
var version string
err := tx.QueryRowContext(ctx, "SELECT version();").Scan(&version)
// Assert
require.NoError(t, err)
require.NotEmpty(t, version)
t.Log(version)
})
t.Run("Changes are not visible in different instances", func(t *testing.T) {
t.Parallel()
// Arrange
ctx := context.Background()
const sqlStr = `CREATE TABLE "no_conflict" (id integer PRIMARY KEY)`
t.Run("Arrange", func(t *testing.T) {
tx := testingpg.NewWithTransactionalCleanup(t)
_, err := tx.ExecContext(ctx, sqlStr)
require.NoError(t, err)
})
var err error
// Act
t.Run("Act", func(t *testing.T) {
tx := testingpg.NewWithTransactionalCleanup(t)
_, err = tx.ExecContext(ctx, sqlStr)
})
// Assert
require.NoError(t, err, "side effects must be isolated for each instance")
})
}