Function TestChatHasUnread

coderd/database/querier_test.go:14559–14680

(t *testing.T)

}

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

	store, _ := dbtestutil.NewDB(t)
	ctx := context.Background()

	org := dbgen.Organization(t, store, database.Organization{})
	user := dbgen.User(t, store, database.User{})
	dbgen.OrganizationMember(t, store, database.OrganizationMember{UserID: user.ID, OrganizationID: org.ID})

	dbgen.ChatProvider(t, store, database.ChatProvider{
		Provider:             "openai",
		DisplayName:          "OpenAI",
		APIKey:               "test-key",
		Enabled:              true,
		CentralApiKeyEnabled: true,
	})

	modelCfg, err := insertChatModelConfigForTest(ctx, t, store, database.InsertChatModelConfigParams{
		Provider:             "openai",
		Model:                "test-model-" + uuid.NewString(),
		DisplayName:          "Test Model",
		CreatedBy:            uuid.NullUUID{UUID: user.ID, Valid: true},
		UpdatedBy:            uuid.NullUUID{UUID: user.ID, Valid: true},
		Enabled:              true,
		IsDefault:            true,
		ContextLimit:         128000,
		CompressionThreshold: 80,
		Options:              json.RawMessage(`{}`),
	})
	require.NoError(t, err)

	chat, err := store.InsertChat(ctx, database.InsertChatParams{
		OrganizationID:    org.ID,
		Status:            database.ChatStatusWaiting,
		ClientType:        database.ChatClientTypeUi,
		OwnerID:           user.ID,
		LastModelConfigID: modelCfg.ID,
		Title:             "test-chat-" + uuid.NewString(),
	})
	require.NoError(t, err)

	getHasUnread := func() bool {
		rows, err := store.GetChats(ctx, database.GetChatsParams{
			OwnedOnly: true,
			ViewerID:  user.ID,
		})
		require.NoError(t, err)
		for _, row := range rows {
			if row.Chat.ID == chat.ID {
				return row.HasUnread
			}
		}
		t.Fatal("chat not found in GetChats result")
		return false
	}

	// New chat with no messages: not unread.