mirror of
https://github.com/coder/coder.git
synced 2025-07-06 15:41:45 +00:00
5861e516b9
Refactors our use of `slogtest` to instantiate a "standard logger" across most of our tests. This standard logger incorporates https://github.com/coder/slog/pull/217 to also ignore database query canceled errors by default, which are a source of low-severity flakes. Any test that has set non-default `slogtest.Options` is left alone. In particular, `coderdtest` defaults to ignoring all errors. We might consider revisiting that decision now that we have better tools to target the really common flaky Error logs on shutdown.
71 lines
1.6 KiB
Go
71 lines
1.6 KiB
Go
package awsiamrds_test
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/coder/coder/v2/cli"
|
|
"github.com/coder/coder/v2/coderd/database/awsiamrds"
|
|
"github.com/coder/coder/v2/coderd/database/pubsub"
|
|
"github.com/coder/coder/v2/testutil"
|
|
)
|
|
|
|
func TestDriver(t *testing.T) {
|
|
t.Parallel()
|
|
// Be sure to set AWS_DEFAULT_REGION to the database region as well.
|
|
// Example:
|
|
// export AWS_DEFAULT_REGION=us-east-2;
|
|
// export DBAWSIAMRDS_TEST_URL="postgres://user@host:5432/dbname";
|
|
url := os.Getenv("DBAWSIAMRDS_TEST_URL")
|
|
if url == "" {
|
|
t.Log("skipping test; no DBAWSIAMRDS_TEST_URL set")
|
|
t.Skip()
|
|
}
|
|
|
|
logger := testutil.Logger(t)
|
|
ctx, cancel := context.WithTimeout(context.Background(), testutil.WaitShort)
|
|
defer cancel()
|
|
|
|
sqlDriver, err := awsiamrds.Register(ctx, "postgres")
|
|
require.NoError(t, err)
|
|
|
|
db, err := cli.ConnectToPostgres(ctx, testutil.Logger(t), sqlDriver, url)
|
|
require.NoError(t, err)
|
|
defer func() {
|
|
_ = db.Close()
|
|
}()
|
|
|
|
i, err := db.QueryContext(ctx, "select 1;")
|
|
require.NoError(t, err)
|
|
defer func() {
|
|
_ = i.Close()
|
|
}()
|
|
|
|
require.True(t, i.Next())
|
|
var one int
|
|
require.NoError(t, i.Scan(&one))
|
|
require.Equal(t, 1, one)
|
|
|
|
ps, err := pubsub.New(ctx, logger, db, url)
|
|
require.NoError(t, err)
|
|
|
|
gotChan := make(chan struct{})
|
|
subCancel, err := ps.Subscribe("test", func(_ context.Context, _ []byte) {
|
|
close(gotChan)
|
|
})
|
|
defer subCancel()
|
|
require.NoError(t, err)
|
|
|
|
err = ps.Publish("test", []byte("hello"))
|
|
require.NoError(t, err)
|
|
|
|
select {
|
|
case <-gotChan:
|
|
case <-ctx.Done():
|
|
require.Fail(t, "timed out waiting for message")
|
|
}
|
|
}
|