opensource/coder
3
0
Fork 0
mirror of https://github.com/coder/coder.git synced 2026-09-24 15:04:27 +08:00
Code Issues Packages Projects Releases Wiki Activity

chore: add tallyman events for ai seat tracking (#22689)

Browse Source 
AI seat tracking inserted as heartbeat into usage table.
This commit is contained in:
Steven Masley
2026-03-18 09:30:22 -05:00
committed by GitHub
parent b83b93ea5c
commit 84de391f26
25 changed files with 860 additions and 34 deletions
Download Patch File Download Diff File Expand all files Collapse all files
enterprise/coderd/workspaces_test.go
+4 -4
View File
@@ -4788,7 +4788,7 @@ func TestWorkspaceAITask(t *testing.T) {
wrk := coderdtest.CreateWorkspace(t, client, template.ID)
build := coderdtest.AwaitWorkspaceBuildJobCompleted(t, client, wrk.LatestBuild.ID)
require.Equal(t, codersdk.WorkspaceStatusRunning, build.Status)
require.Len(t, usage.GetEvents(), 0)
require.Len(t, usage.GetDiscreteEvents(), 0)
})
t.Run("CreateTaskWorkspace", func(t *testing.T) {
@@ -4815,7 +4815,7 @@ func TestWorkspaceAITask(t *testing.T) {
build := coderdtest.AwaitWorkspaceBuildJobCompleted(t, client, wrk.LatestBuild.ID)
require.Equal(t, codersdk.WorkspaceStatusRunning, build.Status)
require.Len(t, usage.GetEvents(), 1)
require.Len(t, usage.GetDiscreteEvents(), 1)
usage.Reset() // Clean slate for easy checks
// Stopping the workspace should not create additional usage.
@@ -4825,7 +4825,7 @@ func TestWorkspaceAITask(t *testing.T) {
})
require.NoError(t, err)
coderdtest.AwaitWorkspaceBuildJobCompleted(t, client, build.ID)
require.Len(t, usage.GetEvents(), 0)
require.Len(t, usage.GetDiscreteEvents(), 0)
usage.Reset() // Clean slate for easy checks
// Starting the workspace manually **WILL** create usage, as it's
@@ -4836,6 +4836,6 @@ func TestWorkspaceAITask(t *testing.T) {
})
require.NoError(t, err)
coderdtest.AwaitWorkspaceBuildJobCompleted(t, client, build.ID)
require.Len(t, usage.GetEvents(), 1)
require.Len(t, usage.GetDiscreteEvents(), 1)
})
}