Function TestChatPinOrderQueries

coderd/database/querier_test.go:12032–12215

(t *testing.T)

}

func TestChatPinOrderQueries(t *testing.T) {
	t.Parallel()
	if testing.Short() {
		t.SkipNow()
	}

	setup := func(t *testing.T) (context.Context, database.Store, uuid.UUID, uuid.UUID, uuid.UUID) {
		t.Helper()

		db, _ := dbtestutil.NewDB(t)
		org := dbgen.Organization(t, db, database.Organization{})
		owner := dbgen.User(t, db, database.User{})
		dbgen.OrganizationMember(t, db, database.OrganizationMember{UserID: owner.ID, OrganizationID: org.ID})

		// Use background context for fixture setup so the
		// timed test context doesn't tick during DB init.
		bg := context.Background()
		dbgen.ChatProvider(t, db, database.ChatProvider{
			Provider:             "openai",
			DisplayName:          "OpenAI",
			APIKey:               "test-key",
			Enabled:              true,
			CentralApiKeyEnabled: true,
		})

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

		ctx := testutil.Context(t, testutil.WaitMedium)
		return ctx, db, owner.ID, modelCfg.ID, org.ID
	}

	createChat := func(t *testing.T, ctx context.Context, db database.Store, ownerID, modelCfgID, orgID uuid.UUID, title string) database.Chat {
		t.Helper()

		chat, err := db.InsertChat(ctx, database.InsertChatParams{
			OrganizationID:    orgID,
			Status:            database.ChatStatusWaiting,
			ClientType:        database.ChatClientTypeUi,
			OwnerID:           ownerID,
			LastModelConfigID: modelCfgID,
			Title:             title,
		})
		require.NoError(t, err)
		return chat
	}