Function TestInsertChatMessages

coderd/database/querier_test.go:10895–11091

(t *testing.T)

}

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

	insertModelConfig := func(
		t *testing.T,
		store database.Store,
		ctx context.Context,
		userID uuid.UUID,
		provider string,
		model string,
		displayName string,
		isDefault bool,
	) database.ChatModelConfig {
		t.Helper()

		modelConfig, err := insertChatModelConfigForTest(ctx, t, store, database.InsertChatModelConfigParams{
			Provider:             provider,
			Model:                model,
			DisplayName:          displayName,
			CreatedBy:            uuid.NullUUID{UUID: userID, Valid: true},
			UpdatedBy:            uuid.NullUUID{UUID: userID, Valid: true},
			Enabled:              true,
			IsDefault:            isDefault,
			ContextLimit:         128000,
			CompressionThreshold: 80,
			Options:              json.RawMessage(`{}`),
		})
		require.NoError(t, err)

		return modelConfig
	}

	setupChat := func(t *testing.T) (database.Store, context.Context, database.User, database.Chat, string, database.ChatModelConfig) {
		t.Helper()

		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 := "openai"

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

		modelConfigA := insertModelConfig(
			t,
			store,
			ctx,
			user.ID,
			provider,
			"test-model-a-"+uuid.NewString(),