zombiezen / postgrestest

A Go test harness that starts an ephemeral PostgreSQL server
Apache License 2.0
53 stars 3 forks source link
golang postgres postgresql testing

zombiezen.com/go/postgrestest

Reference Contributor Covenant

Package postgrestest provides a test harness that starts an ephemeral PostgreSQL server. It is tested on macOS, Linux, and Windows. It can cut down the overhead of PostgreSQL in tests up to 90% compared to spinning up a postgres Docker container: starting a server with this package takes roughly 650 milliseconds and creating a database takes roughly 20 milliseconds.

Example

func TestApp(t *testing.T) {
    // Start up the PostgreSQL server. This can take a few seconds, so better to
    // do it once per test run.
    ctx := context.Background()
    srv, err := postgrestest.Start(ctx)
    if err != nil {
        t.Fatal(err)
    }
    t.Cleanup(srv.Cleanup)

    // Each of your subtests can have their own database:
    t.Run("Test1", func(t *testing.T) {
        db, err := srv.NewDatabase(ctx)
        if err != nil {
            t.Fatal(err)
        }
        if _, err := db.Exec(`CREATE TABLE foo (id SERIAL PRIMARY KEY);`); err != nil {
            t.Fatal(err)
        }
        // ...
    })

    t.Run("Test2", func(t *testing.T) {
        db, err := srv.NewDatabase(ctx)
        if err != nil {
            t.Fatal(err)
        }
        if _, err := db.Exec(`CREATE TABLE foo (id SERIAL PRIMARY KEY);`); err != nil {
            t.Fatal(err)
        }
        // ...
    })
}

Installation

PostgreSQL must be installed locally for this package to work. See the PostgreSQL Downloads page for instructions on how to obtain PostgreSQL for your operating system.

To install the package:

go get zombiezen.com/go/postgrestest

License

Apache 2.0