Function TestWatcher_ClosedAfterDial

coderd/workspaceconnwatcher/watcher_test.go:429–472

(t *testing.T)

}

func TestWatcher_ClosedAfterDial(t *testing.T) {
	t.Parallel()

	ctx := testutil.Context(t, testutil.WaitShort)
	logger := testutil.Logger(t)
	h := newHarness(ctx, t, logger)

	h.db.EXPECT().GetLatestWorkspaceBuildWithStatusByWorkspaceID(gomock.Any(), h.workspace.ID).
		Times(1).
		Return(database.GetLatestWorkspaceBuildWithStatusByWorkspaceIDRow{
			Transition:  database.WorkspaceTransitionStop,
			BuildNumber: 1,
			JobStatus:   database.ProvisionerJobStatusSucceeded,
			WorkspaceTable: database.WorkspaceTable{
				ID:             h.workspace.ID,
				OwnerID:        userID,
				OrganizationID: orgID,
			},
		}, nil)

	dec, err := h.Dial(ctx, "wss://local.test/")
	require.NoError(t, err)
	events := dec.Chan()
	_ = testutil.RequireReceive(ctx, t, events)

	closed := make(chan struct{})
	go func() {
		defer close(closed)
		h.watcher.Close()
	}()

	e := testutil.RequireReceive(ctx, t, events)
	require.NotNil(t, e.Error)
	require.Equal(t, workspacesdk.WatchErrorServerShutdown, e.Error.Code)
	require.True(t, e.Error.Retryable)

	select {
	case <-ctx.Done():
		t.Fatal("context timed out")
	case _, ok := <-events:
		require.False(t, ok, "socket not closed")
	}
	testutil.TryReceive(ctx, t, closed)
}