Function NewDB

coderd/database/dbtestutil/db.go:92–146

(t testing.TB, opts ...Option)

}

func NewDB(t testing.TB, opts ...Option) (database.Store, pubsub.Pubsub) {
	t.Helper()

	o := options{logger: testutil.Logger(t).Named("pubsub")}
	for _, opt := range opts {
		opt(&o)
	}

	var db database.Store
	var ps pubsub.Pubsub

	connectionURL := os.Getenv("CODER_PG_CONNECTION_URL")
	if connectionURL == "" && o.url != "" {
		connectionURL = o.url
	}
	if connectionURL == "" {
		var err error
		connectionURL, err = Open(t)
		require.NoError(t, err)
	}

	if o.fixedTimezone == "" {
		// To make sure we find timezone-related issues, we set the timezone
		// of the database to a non-UTC one.
		// The below was picked due to the following properties:
		// - It has a non-UTC offset
		// - It has a fractional hour UTC offset
		// - It includes a daylight savings time component
		o.fixedTimezone = DefaultTimezone
	}
	dbName := dbNameFromConnectionURL(t, connectionURL)
	setDBTimezone(t, connectionURL, dbName, o.fixedTimezone)

	sqlDB, err := sql.Open("postgres", connectionURL)
	require.NoError(t, err)
	t.Cleanup(func() {
		_ = sqlDB.Close()
	})
	if o.returnSQLDB != nil {
		o.returnSQLDB(sqlDB)
	}
	if o.dumpOnFailure {
		t.Cleanup(func() { DumpOnFailure(t, connectionURL) })
	}
	// Unit tests should not retry serial transaction failures.
	db = database.New(sqlDB, database.WithSerialRetryCount(1))

	ps, err = pubsub.New(context.Background(), o.logger, sqlDB, connectionURL)
	require.NoError(t, err)
	t.Cleanup(func() {
		_ = ps.Close()
	})

	return db, ps
}

// setRandDBTimezone sets the timezone of the database to the given timezone.
// Note that the updated timezone only comes into effect on reconnect, so we