//go:build linux

package database_test

import (

"context"

"database/sql"

"testing"

"github.com/stretchr/testify/assert"

"github.com/stretchr/testify/require"

"github.com/coder/coder/coderd/database"

"github.com/coder/coder/coderd/database/postgres"

)

// nolint:tparallel,paralleltest

func TestPubsub(t *testing.T) {

t.Parallel()

if testing.Short() {

t.Skip()

return

}

t.Run("Postgres", func(t *testing.T) {

ctx, cancelFunc := context.WithCancel(context.Background())

defer cancelFunc()

connectionURL, closePg, err := postgres.Open()

require.NoError(t, err)

defer closePg()

db, err := sql.Open("postgres", connectionURL)

require.NoError(t, err)

defer db.Close()

pubsub, err := database.NewPubsub(ctx, db, connectionURL)

require.NoError(t, err)

defer pubsub.Close()

event := "test"

data := "testing"

messageChannel := make(chan []byte)

cancelFunc, err = pubsub.Subscribe(event, func(ctx context.Context, message []byte) {

messageChannel <- message

})

require.NoError(t, err)

defer cancelFunc()

go func() {

err = pubsub.Publish(event, []byte(data))

assert.NoError(t, err)

}()

message := <-messageChannel

assert.Equal(t, string(message), data)

})

t.Run("PostgresCloseCancel", func(t *testing.T) {

ctx, cancelFunc := context.WithCancel(context.Background())

defer cancelFunc()

connectionURL, closePg, err := postgres.Open()

require.NoError(t, err)

defer closePg()

db, err := sql.Open("postgres", connectionURL)

require.NoError(t, err)

defer db.Close()

pubsub, err := database.NewPubsub(ctx, db, connectionURL)

require.NoError(t, err)

defer pubsub.Close()

cancelFunc()

})

}