Function TestGetAuthorizedChats

coderd/database/querier_test.go:1238–1483

(t *testing.T)

}

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

	sqlDB := testSQLDB(t)
	err := migrations.Up(sqlDB)
	require.NoError(t, err)
	db := database.New(sqlDB)
	authorizer := rbac.NewStrictCachingAuthorizer(prometheus.NewRegistry())

	// Create users with different roles.
	owner := dbgen.User(t, db, database.User{
		RBACRoles: []string{rbac.RoleOwner().String()},
	})
	member := dbgen.User(t, db, database.User{})
	secondMember := dbgen.User(t, db, database.User{})

	org := dbgen.Organization(t, db, database.Organization{})
	dbgen.OrganizationMember(t, db, database.OrganizationMember{UserID: owner.ID, OrganizationID: org.ID})
	dbgen.OrganizationMember(t, db, database.OrganizationMember{UserID: member.ID, OrganizationID: org.ID, Roles: []string{rbac.RoleAgentsAccess()}})
	dbgen.OrganizationMember(t, db, database.OrganizationMember{UserID: secondMember.ID, OrganizationID: org.ID, Roles: []string{rbac.RoleAgentsAccess()}})

	// Create FK dependencies: a chat provider and model config.
	_ = dbgen.ChatProvider(t, db, database.ChatProvider{
		Provider:    "openai",
		DisplayName: "OpenAI",
	})
	modelCfg := dbgen.ChatModelConfig(t, db, database.ChatModelConfig{
		Provider:             "openai",
		Model:                "test-model",
		CreatedBy:            uuid.NullUUID{UUID: owner.ID, Valid: true},
		UpdatedBy:            uuid.NullUUID{UUID: owner.ID, Valid: true},
		IsDefault:            true,
		CompressionThreshold: 80,
	})

	// Create 3 chats owned by owner.
	for i := range 3 {
		dbgen.Chat(t, db, database.Chat{
			OrganizationID:    org.ID,
			OwnerID:           owner.ID,
			LastModelConfigID: modelCfg.ID,
			Title:             fmt.Sprintf("owner chat %d", i+1),
		})
	}

	// Create 2 chats owned by member.
	for i := range 2 {
		dbgen.Chat(t, db, database.Chat{
			OrganizationID:    org.ID,
			OwnerID:           member.ID,
			LastModelConfigID: modelCfg.ID,
			Title:             fmt.Sprintf("member chat %d", i+1),
		})
	}