Skip to content

Commit 10bb649

Browse files
committed
Use generated types in test helper entities
1 parent 80a5d1a commit 10bb649

File tree

1 file changed

+67
-40
lines changed

1 file changed

+67
-40
lines changed

site/src/testHelpers/entities.ts

Lines changed: 67 additions & 40 deletions
Original file line numberDiff line numberDiff line change
@@ -1,44 +1,37 @@
1-
import {
2-
BuildInfoResponse,
3-
Organization,
4-
Provisioner,
5-
Template,
6-
UserAgent,
7-
UserResponse,
8-
Workspace,
9-
WorkspaceAgent,
10-
WorkspaceAutostartRequest,
11-
WorkspaceResource,
12-
} from "../api/types"
13-
import { AuthMethods, Role } from "../api/typesGenerated"
14-
15-
export const MockSessionToken = { session_token: "my-session-token" }
16-
17-
export const MockAPIKey = { key: "my-api-key" }
18-
19-
export const MockBuildInfo: BuildInfoResponse = {
1+
import * as Types from "../api/types"
2+
import * as TypesGen from "../api/typesGenerated"
3+
4+
export const MockSessionToken: TypesGen.LoginWithPasswordResponse = {
5+
session_token: "my-session-token",
6+
}
7+
8+
export const MockAPIKey: TypesGen.GenerateAPIKeyResponse = {
9+
key: "my-api-key",
10+
}
11+
12+
export const MockBuildInfo: TypesGen.BuildInfoResponse = {
2013
external_url: "file:///mock-url",
2114
version: "v99.999.9999+c9cdf14",
2215
}
2316

24-
export const MockAdminRole: Role = {
17+
export const MockAdminRole: TypesGen.Role = {
2518
name: "admin",
2619
display_name: "Admin",
2720
}
2821

29-
export const MockMemberRole: Role = {
22+
export const MockMemberRole: TypesGen.Role = {
3023
name: "member",
3124
display_name: "Member",
3225
}
3326

34-
export const MockAuditorRole: Role = {
27+
export const MockAuditorRole: TypesGen.Role = {
3528
name: "auditor",
3629
display_name: "Auditor",
3730
}
3831

3932
export const MockSiteRoles = [MockAdminRole, MockAuditorRole, MockMemberRole]
4033

41-
export const MockUser: UserResponse = {
34+
export const MockUser: TypesGen.User = {
4235
id: "test-user",
4336
username: "TestUser",
4437
email: "test@coder.com",
@@ -48,7 +41,7 @@ export const MockUser: UserResponse = {
4841
roles: [MockAdminRole, MockMemberRole],
4942
}
5043

51-
export const MockUser2: UserResponse = {
44+
export const MockUser2: TypesGen.User = {
5245
id: "test-user-2",
5346
username: "TestUser2",
5447
email: "test2@coder.com",
@@ -58,80 +51,114 @@ export const MockUser2: UserResponse = {
5851
roles: [MockMemberRole],
5952
}
6053

61-
export const MockOrganization: Organization = {
54+
export const MockOrganization: TypesGen.Organization = {
6255
id: "test-org",
6356
name: "Test Organization",
6457
created_at: "",
6558
updated_at: "",
6659
}
6760

68-
export const MockProvisioner: Provisioner = {
61+
export const MockProvisioner: TypesGen.ProvisionerDaemon = {
62+
created_at: "",
6963
id: "test-provisioner",
7064
name: "Test Provisioner",
65+
provisioners: [],
66+
}
67+
68+
export const MockProvisionerJob: TypesGen.ProvisionerJob = {
69+
created_at: "",
70+
id: "test-provisioner-job",
71+
status: "succeeded",
7172
}
7273

73-
export const MockTemplate: Template = {
74+
export const MockTemplate: TypesGen.Template = {
7475
id: "test-template",
7576
created_at: "",
7677
updated_at: "",
7778
organization_id: MockOrganization.id,
7879
name: "Test Template",
7980
provisioner: MockProvisioner.id,
8081
active_version_id: "",
82+
workspace_owner_count: 1,
8183
}
8284

83-
export const MockWorkspaceAutostartDisabled: WorkspaceAutostartRequest = {
85+
export const MockWorkspaceAutostartDisabled: TypesGen.UpdateWorkspaceAutostartRequest = {
8486
schedule: "",
8587
}
8688

87-
export const MockWorkspaceAutostartEnabled: WorkspaceAutostartRequest = {
89+
export const MockWorkspaceAutostartEnabled: TypesGen.UpdateWorkspaceAutostartRequest = {
8890
// Runs at 9:30am Monday through Friday using Canada/Eastern
8991
// (America/Toronto) time
9092
schedule: "CRON_TZ=Canada/Eastern 30 9 * * 1-5",
9193
}
9294

93-
export const MockWorkspaceAutostopDisabled: WorkspaceAutostartRequest = {
95+
export const MockWorkspaceAutostopDisabled: TypesGen.UpdateWorkspaceAutostartRequest = {
9496
schedule: "",
9597
}
9698

97-
export const MockWorkspaceAutostopEnabled: WorkspaceAutostartRequest = {
99+
export const MockWorkspaceAutostopEnabled: TypesGen.UpdateWorkspaceAutostartRequest = {
98100
// Runs at 9:30pm Monday through Friday using America/Toronto
99101
schedule: "CRON_TZ=America/Toronto 30 21 * * 1-5",
100102
}
101103

102-
export const MockWorkspace: Workspace = {
104+
export const MockWorkspaceBuild: TypesGen.WorkspaceBuild = {
105+
after_id: "",
106+
before_id: "",
107+
created_at: "",
108+
id: "test-workspace-build",
109+
initiator_id: "",
110+
job: MockProvisionerJob,
111+
name: "a-workspace-build",
112+
template_version_id: "",
113+
transition: "start",
114+
updated_at: "",
115+
workspace_id: "test-workspace",
116+
}
117+
118+
export const MockWorkspace: TypesGen.Workspace = {
103119
id: "test-workspace",
104120
name: "Test-Workspace",
105121
created_at: "",
106122
updated_at: "",
107123
template_id: MockTemplate.id,
124+
template_name: MockTemplate.name,
125+
outdated: false,
108126
owner_id: MockUser.id,
109127
autostart_schedule: MockWorkspaceAutostartEnabled.schedule,
110128
autostop_schedule: MockWorkspaceAutostopEnabled.schedule,
111-
latest_build: {
112-
id: "test-workspace-build",
113-
},
129+
latest_build: MockWorkspaceBuild,
114130
}
115131

116-
export const MockWorkspaceAgent: WorkspaceAgent = {
132+
export const MockWorkspaceAgent: TypesGen.WorkspaceAgent = {
133+
architecture: "amd64",
134+
created_at: "",
135+
environment_variables: {},
117136
id: "test-workspace-agent",
118137
name: "a-workspace-agent",
119138
operating_system: "linux",
139+
resource_id: "",
140+
status: "connected",
141+
updated_at: "",
120142
}
121143

122-
export const MockWorkspaceResource: WorkspaceResource = {
123-
id: "test-workspace-resource",
144+
export const MockWorkspaceResource: TypesGen.WorkspaceResource = {
124145
agents: [MockWorkspaceAgent],
146+
created_at: "",
147+
id: "test-workspace-resource",
148+
job_id: "",
149+
name: "a-workspace-resource",
150+
type: "google_compute_disk",
151+
workspace_transition: "start",
125152
}
126153

127-
export const MockUserAgent: UserAgent = {
154+
export const MockUserAgent: Types.UserAgent = {
128155
browser: "Chrome 99.0.4844",
129156
device: "Other",
130157
ip_address: "11.22.33.44",
131158
os: "Windows 10",
132159
}
133160

134-
export const MockAuthMethods: AuthMethods = {
161+
export const MockAuthMethods: TypesGen.AuthMethods = {
135162
password: true,
136163
github: false,
137164
}

0 commit comments

Comments
 (0)