@@ -7,10 +7,11 @@ import (
77
88"github.com/stretchr/testify/require"
99
10+ "cdr.dev/slog"
1011"cdr.dev/slog/sloggers/slogtest"
11-
1212"github.com/coder/coder/v2/cli"
13- awsrdsiam"github.com/coder/coder/v2/coderd/database/awsiamrds"
13+ "github.com/coder/coder/v2/coderd/database/awsiamrds"
14+ "github.com/coder/coder/v2/coderd/database/pubsub"
1415"github.com/coder/coder/v2/testutil"
1516)
1617
@@ -25,10 +26,11 @@ func TestDriver(t *testing.T) {
2526t .Skip ()
2627}
2728
29+ logger := slogtest .Make (t ,nil ).Leveled (slog .LevelDebug )
2830ctx ,cancel := context .WithTimeout (context .Background (),testutil .WaitShort )
2931defer cancel ()
3032
31- sqlDriver ,err := awsrdsiam .Register (ctx ,"postgres" )
33+ sqlDriver ,err := awsiamrds .Register (ctx ,"postgres" )
3234require .NoError (t ,err )
3335
3436db ,err := cli .ConnectToPostgres (ctx ,slogtest .Make (t ,nil ),sqlDriver ,url )
@@ -47,4 +49,19 @@ func TestDriver(t *testing.T) {
4749var one int
4850require .NoError (t ,i .Scan (& one ))
4951require .Equal (t ,1 ,one )
52+
53+ ps ,err := pubsub .New (ctx ,logger ,db ,url )
54+ require .NoError (t ,err )
55+
56+ gotChan := make (chan struct {})
57+ subCancel ,err := ps .Subscribe ("test" ,func (_ context.Context ,_ []byte ) {
58+ close (gotChan )
59+ })
60+ defer subCancel ()
61+ require .NoError (t ,err )
62+
63+ err = ps .Publish ("test" , []byte ("hello" ))
64+ require .NoError (t ,err )
65+
66+ <- gotChan
5067}