Function TestStreamStateCollector

coderd/x/chatd/streamcollector_internal_test.go:22–137

TestStreamStateCollector exercises the four gauges emitted by streamStateCollector against representative map states.

(t *testing.T)

// TestStreamStateCollector exercises the four gauges emitted by
// streamStateCollector against representative map states.
func TestStreamStateCollector(t *testing.T) {
	t.Parallel()

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

		reg := prometheus.NewRegistry()
		server := &Server{}
		reg.MustRegister(&streamStateCollector{server: server})

		assertGauges(t, reg, gaugeExpectations{
			active:      0,
			bufferMax:   0,
			bufferTotal: 0,
			subscribers: 0,
		})
	})

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

		reg := prometheus.NewRegistry()
		server := &Server{}

		server.chatStreams.Store(uuid.New(), &chatStreamState{
			buffer:      make([]bufferedStreamPart, 10),
			subscribers: newSubscribers(t, 2),
		})
		server.chatStreams.Store(uuid.New(), &chatStreamState{
			buffer:      make([]bufferedStreamPart, 25),
			subscribers: map[uuid.UUID]chan codersdk.ChatStreamEvent{},
		})
		server.chatStreams.Store(uuid.New(), &chatStreamState{
			buffer:      nil,
			subscribers: newSubscribers(t, 1),
		})

		reg.MustRegister(&streamStateCollector{server: server})

		assertGauges(t, reg, gaugeExpectations{
			active:      3,
			bufferMax:   25,
			bufferTotal: 35,
			subscribers: 3,
		})
	})

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

		reg := prometheus.NewRegistry()
		server := &Server{}

		server.chatStreams.Store(uuid.New(), "garbage")
		server.chatStreams.Store(uuid.New(), &chatStreamState{
			buffer:      make([]bufferedStreamPart, 5),
			subscribers: newSubscribers(t, 1),
		})