Function TestGetChatsFilter

coderd/database/querier_test.go:14264–14557

(t *testing.T)

}

func TestGetChatsFilter(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})

	provider := dbgen.AIProviderWithOptionalKey(t, store, database.AIProvider{
		Type: database.AiProviderTypeOpenai,
	}, "test-key")

	modelCfg, err := store.InsertChatModelConfig(ctx, database.InsertChatModelConfigParams{
		Provider:             "openai",
		AIProviderID:         uuid.NullUUID{UUID: provider.ID, Valid: true},
		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)

	// --- helpers ---

	createRoot := func(title string) database.Chat {
		t.Helper()
		chat, err := store.InsertChat(ctx, database.InsertChatParams{
			OrganizationID:    org.ID,
			Status:            database.ChatStatusWaiting,
			ClientType:        database.ChatClientTypeUi,
			OwnerID:           user.ID,
			LastModelConfigID: modelCfg.ID,
			Title:             title,
		})
		require.NoError(t, err)
		return chat
	}

	createChild := func(root database.Chat, title string) database.Chat {
		t.Helper()
		chat, err := store.InsertChat(ctx, database.InsertChatParams{
			OrganizationID:    org.ID,
			Status:            database.ChatStatusWaiting,
			ClientType:        database.ChatClientTypeUi,
			OwnerID:           user.ID,
			LastModelConfigID: modelCfg.ID,
			Title:             title,
			ParentChatID:      uuid.NullUUID{UUID: root.ID, Valid: true},
			RootChatID:        uuid.NullUUID{UUID: root.ID, Valid: true},
		})
		require.NoError(t, err)