Files
coder/database/pubsub_memory_test.go
Jonathan Yu 599ea2a331 chore: add tparallel linter (#88)
Add the tparallel linter, which checks that subtests of a parallel
test also run in parallel.
2022-01-29 12:21:16 -08:00

36 lines
734 B
Go

package database_test
import (
"context"
"testing"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
"github.com/coder/coder/database"
)
func TestPubsubMemory(t *testing.T) {
t.Parallel()
t.Run("Memory", func(t *testing.T) {
t.Parallel()
pubsub := database.NewPubsubInMemory()
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))
require.NoError(t, err)
}()
message := <-messageChannel
assert.Equal(t, string(message), data)
})
}