Movatterモバイル変換

[0]ホーム
URL:

Skip to content

Navigation Menu

Sign in
Appearance settings

Search code, repositories, users, issues, pull requests...

Provide feedback

We read every piece of feedback, and take your input very seriously.

Saved searches

Use saved searches to filter your results more quickly

 Sign up
Appearance settings

Commitaab9e3a

Browse files
authored
chore: Add more objects to dbgen (#6013)
* chore: Add WorkspaceApps to dbgen* GitSSHKey, UserLink, GitAuthLink
1 parentfd2f9dc commitaab9e3a

File tree

2 files changed

+79
-0
lines changed

2 files changed

+79
-0
lines changed

‎coderd/database/dbgen/generator.go

Lines changed: 55 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -186,6 +186,18 @@ func User(t *testing.T, db database.Store, orig database.User) database.User {
186186
returnuser
187187
}
188188

189+
funcGitSSHKey(t*testing.T,db database.Store,orig database.GitSSHKey) database.GitSSHKey {
190+
key,err:=db.InsertGitSSHKey(context.Background(), database.InsertGitSSHKeyParams{
191+
UserID:takeFirst(orig.UserID,uuid.New()),
192+
CreatedAt:takeFirst(orig.CreatedAt,time.Now()),
193+
UpdatedAt:takeFirst(orig.UpdatedAt,time.Now()),
194+
PrivateKey:takeFirst(orig.PrivateKey,""),
195+
PublicKey:takeFirst(orig.PublicKey,""),
196+
})
197+
require.NoError(t,err,"insert ssh key")
198+
returnkey
199+
}
200+
189201
funcOrganization(t*testing.T,db database.Store,orig database.Organization) database.Organization {
190202
org,err:=db.InsertOrganization(context.Background(), database.InsertOrganizationParams{
191203
ID:takeFirst(orig.ID,uuid.New()),
@@ -254,6 +266,34 @@ func ProvisionerJob(t *testing.T, db database.Store, orig database.ProvisionerJo
254266
returnjob
255267
}
256268

269+
funcWorkspaceApp(t*testing.T,db database.Store,orig database.WorkspaceApp) database.WorkspaceApp {
270+
resource,err:=db.InsertWorkspaceApp(context.Background(), database.InsertWorkspaceAppParams{
271+
ID:takeFirst(orig.ID,uuid.New()),
272+
CreatedAt:takeFirst(orig.CreatedAt,time.Now()),
273+
AgentID:takeFirst(orig.AgentID,uuid.New()),
274+
Slug:takeFirst(orig.Slug,namesgenerator.GetRandomName(1)),
275+
DisplayName:takeFirst(orig.DisplayName,namesgenerator.GetRandomName(1)),
276+
Icon:takeFirst(orig.Icon,namesgenerator.GetRandomName(1)),
277+
Command: sql.NullString{
278+
String:takeFirst(orig.Command.String,"ls"),
279+
Valid:orig.Command.Valid,
280+
},
281+
Url: sql.NullString{
282+
String:takeFirst(orig.Url.String),
283+
Valid:orig.Url.Valid,
284+
},
285+
External:orig.External,
286+
Subdomain:orig.Subdomain,
287+
SharingLevel:takeFirst(orig.SharingLevel,database.AppSharingLevelOwner),
288+
HealthcheckUrl:takeFirst(orig.HealthcheckUrl,"https://localhost:8000"),
289+
HealthcheckInterval:takeFirst(orig.HealthcheckInterval,60),
290+
HealthcheckThreshold:takeFirst(orig.HealthcheckThreshold,60),
291+
Health:takeFirst(orig.Health,database.WorkspaceAppHealthHealthy),
292+
})
293+
require.NoError(t,err,"insert app")
294+
returnresource
295+
}
296+
257297
funcWorkspaceResource(t*testing.T,db database.Store,orig database.WorkspaceResource) database.WorkspaceResource {
258298
resource,err:=db.InsertWorkspaceResource(context.Background(), database.InsertWorkspaceResourceParams{
259299
ID:takeFirst(orig.ID,uuid.New()),
@@ -312,6 +352,21 @@ func UserLink(t *testing.T, db database.Store, orig database.UserLink) database.
312352
returnlink
313353
}
314354

355+
funcGitAuthLink(t*testing.T,db database.Store,orig database.GitAuthLink) database.GitAuthLink {
356+
link,err:=db.InsertGitAuthLink(context.Background(), database.InsertGitAuthLinkParams{
357+
ProviderID:takeFirst(orig.ProviderID,uuid.New().String()),
358+
UserID:takeFirst(orig.UserID,uuid.New()),
359+
OAuthAccessToken:takeFirst(orig.OAuthAccessToken,uuid.NewString()),
360+
OAuthRefreshToken:takeFirst(orig.OAuthAccessToken,uuid.NewString()),
361+
OAuthExpiry:takeFirst(orig.OAuthExpiry,time.Now().Add(time.Hour*24)),
362+
CreatedAt:takeFirst(orig.CreatedAt,time.Now()),
363+
UpdatedAt:takeFirst(orig.UpdatedAt,time.Now()),
364+
})
365+
366+
require.NoError(t,err,"insert git auth link")
367+
returnlink
368+
}
369+
315370
funcTemplateVersion(t*testing.T,db database.Store,orig database.TemplateVersion) database.TemplateVersion {
316371
version,err:=db.InsertTemplateVersion(context.Background(), database.InsertTemplateVersionParams{
317372
ID:takeFirst(orig.ID,uuid.New()),

‎coderd/database/dbgen/generator_test.go

Lines changed: 24 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -44,13 +44,30 @@ func TestGenerator(t *testing.T) {
4444
require.Equal(t,exp,must(db.GetUserLinkByLinkedID(context.Background(),exp.LinkedID)))
4545
})
4646

47+
t.Run("GitAuthLink",func(t*testing.T) {
48+
t.Parallel()
49+
db:=dbfake.New()
50+
exp:=dbgen.GitAuthLink(t,db, database.GitAuthLink{})
51+
require.Equal(t,exp,must(db.GetGitAuthLink(context.Background(), database.GetGitAuthLinkParams{
52+
ProviderID:exp.ProviderID,
53+
UserID:exp.UserID,
54+
})))
55+
})
56+
4757
t.Run("WorkspaceResource",func(t*testing.T) {
4858
t.Parallel()
4959
db:=dbfake.New()
5060
exp:=dbgen.WorkspaceResource(t,db, database.WorkspaceResource{})
5161
require.Equal(t,exp,must(db.GetWorkspaceResourceByID(context.Background(),exp.ID)))
5262
})
5363

64+
t.Run("WorkspaceApp",func(t*testing.T) {
65+
t.Parallel()
66+
db:=dbfake.New()
67+
exp:=dbgen.WorkspaceApp(t,db, database.WorkspaceApp{})
68+
require.Equal(t,exp,must(db.GetWorkspaceAppsByAgentID(context.Background(),exp.AgentID))[0])
69+
})
70+
5471
t.Run("WorkspaceResourceMetadatum",func(t*testing.T) {
5572
t.Parallel()
5673
db:=dbfake.New()
@@ -159,6 +176,13 @@ func TestGenerator(t *testing.T) {
159176
exp:=dbgen.User(t,db, database.User{})
160177
require.Equal(t,exp,must(db.GetUserByID(context.Background(),exp.ID)))
161178
})
179+
180+
t.Run("SSHKey",func(t*testing.T) {
181+
t.Parallel()
182+
db:=dbfake.New()
183+
exp:=dbgen.GitSSHKey(t,db, database.GitSSHKey{})
184+
require.Equal(t,exp,must(db.GetGitSSHKey(context.Background(),exp.UserID)))
185+
})
162186
}
163187

164188
funcmust[Tany](valueT,errerror)T {

0 commit comments

Comments
 (0)
[8]ページ先頭
©2009-2025 Movatter.jp