Function TestCreateChatNonDefaultOrg

enterprise/coderd/exp_chats_test.go:1080–1148

(t *testing.T)

}

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

	ctx := testutil.Context(t, testutil.WaitLong)

	client, firstUser := coderdenttest.New(t, &coderdenttest.Options{
		Options: &coderdtest.Options{
			DeploymentValues: func() *codersdk.DeploymentValues {
				v := coderdtest.DeploymentValues(t)
				return v
			}(),
		},
		LicenseOptions: &coderdenttest.LicenseOptions{
			Features: license.Features{
				codersdk.FeatureMultipleOrganizations: 1,
			},
		},
	})
	expClient := codersdk.NewExperimentalClient(client)

	provider := createOpenAIProviderForTest(ctx, t, expClient, "test-key", "https://example.com")
	_, err := expClient.CreateChatModelConfig(ctx, codersdk.CreateChatModelConfigRequest{
		Provider:             string(provider.Type),
		AIProviderID:         &provider.ID,
		Model:                "gpt-4o-mini",
		DisplayName:          "Test Model",
		IsDefault:            ptr.Ref(true),
		ContextLimit:         ptr.Ref(int64(1000)),
		CompressionThreshold: ptr.Ref(int32(70)),
	})
	require.NoError(t, err)

	// Create a second (non-default) org via the API.
	secondOrg := coderdenttest.CreateOrganization(t, client, coderdenttest.CreateOrganizationOptions{})

	// Create a member with agents-access in both orgs.
	memberClientRaw, member := coderdtest.CreateAnotherUser(
		t, client, firstUser.OrganizationID,
		rbac.ScopedRoleAgentsAccess(firstUser.OrganizationID),
		rbac.ScopedRoleAgentsAccess(secondOrg.ID),
	)
	memberClient := codersdk.NewExperimentalClient(memberClientRaw)
	// Create a chat in the non-default org.
	chat, err := memberClient.CreateChat(ctx, codersdk.CreateChatRequest{
		OrganizationID: secondOrg.ID,
		Content: []codersdk.ChatInputPart{
			{
				Type: codersdk.ChatInputPartTypeText,
				Text: "hello from non-default org",
			},
		},
	})
	require.NoError(t, err)
	require.Equal(t, secondOrg.ID, chat.OrganizationID)
	require.Equal(t, member.ID, chat.OwnerID)

	// Verify the chat is visible when listing.
	chats, err := memberClient.ListChats(ctx, nil)