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

Commit1113d8f

Browse files
committed
Move generate functions and no more generics
1 parentbc88e41 commit1113d8f

File tree

3 files changed

+272
-0
lines changed

3 files changed

+272
-0
lines changed
Lines changed: 221 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,221 @@
1+
package fakegen
2+
3+
import (
4+
"context"
5+
"crypto/sha256"
6+
"database/sql"
7+
"encoding/hex"
8+
"testing"
9+
"time"
10+
11+
"github.com/coder/coder/cryptorand"
12+
"github.com/tabbed/pqtype"
13+
14+
"github.com/coder/coder/coderd/database"
15+
"github.com/google/uuid"
16+
"github.com/moby/moby/pkg/namesgenerator"
17+
"github.com/stretchr/testify/require"
18+
)
19+
20+
funcTemplate(t*testing.T,db database.Store,seed database.Template) database.Template {
21+
template,err:=db.InsertTemplate(context.Background(), database.InsertTemplateParams{
22+
ID:takeFirst(seed.ID,uuid.New()),
23+
CreatedAt:takeFirst(seed.CreatedAt,time.Now()),
24+
UpdatedAt:takeFirst(seed.UpdatedAt,time.Now()),
25+
OrganizationID:takeFirst(seed.OrganizationID,uuid.New()),
26+
Name:takeFirst(seed.Name,namesgenerator.GetRandomName(1)),
27+
Provisioner:takeFirst(seed.Provisioner,database.ProvisionerTypeEcho),
28+
ActiveVersionID:takeFirst(seed.ActiveVersionID,uuid.New()),
29+
Description:takeFirst(seed.Description,namesgenerator.GetRandomName(1)),
30+
DefaultTTL:takeFirst(seed.DefaultTTL,3600),
31+
CreatedBy:takeFirst(seed.CreatedBy,uuid.New()),
32+
Icon:takeFirst(seed.Icon,namesgenerator.GetRandomName(1)),
33+
UserACL:seed.UserACL,
34+
GroupACL:seed.GroupACL,
35+
DisplayName:takeFirst(seed.DisplayName,namesgenerator.GetRandomName(1)),
36+
AllowUserCancelWorkspaceJobs:takeFirst(seed.AllowUserCancelWorkspaceJobs,true),
37+
})
38+
require.NoError(t,err,"insert template")
39+
returntemplate
40+
}
41+
42+
funcAPIKey(t*testing.T,db database.Store,seed database.APIKey) (key database.APIKey,secretstring) {
43+
id,_:=cryptorand.String(10)
44+
secret,_=cryptorand.String(22)
45+
hashed:=sha256.Sum256([]byte(secret))
46+
47+
key,err:=db.InsertAPIKey(context.Background(), database.InsertAPIKeyParams{
48+
ID:takeFirst(seed.ID,id),
49+
// 0 defaults to 86400 at the db layer
50+
LifetimeSeconds:takeFirst(seed.LifetimeSeconds,0),
51+
HashedSecret:takeFirstBytes(seed.HashedSecret,hashed[:]),
52+
IPAddress: pqtype.Inet{},
53+
UserID:takeFirst(seed.UserID,uuid.New()),
54+
LastUsed:takeFirst(seed.LastUsed,time.Now()),
55+
ExpiresAt:takeFirst(seed.ExpiresAt,time.Now().Add(time.Hour)),
56+
CreatedAt:takeFirst(seed.CreatedAt,time.Now()),
57+
UpdatedAt:takeFirst(seed.UpdatedAt,time.Now()),
58+
LoginType:takeFirst(seed.LoginType,database.LoginTypePassword),
59+
Scope:takeFirst(seed.Scope,database.APIKeyScopeAll),
60+
})
61+
require.NoError(t,err,"insert api key")
62+
returnkey,secret
63+
}
64+
65+
funcWorkspace(t*testing.T,db database.Store,orig database.Workspace) database.Workspace {
66+
workspace,err:=db.InsertWorkspace(context.Background(), database.InsertWorkspaceParams{
67+
ID:takeFirst(orig.ID,uuid.New()),
68+
OwnerID:takeFirst(orig.OwnerID,uuid.New()),
69+
CreatedAt:takeFirst(orig.CreatedAt,time.Now()),
70+
UpdatedAt:takeFirst(orig.UpdatedAt,time.Now()),
71+
OrganizationID:takeFirst(orig.OrganizationID,uuid.New()),
72+
TemplateID:takeFirst(orig.TemplateID,uuid.New()),
73+
Name:takeFirst(orig.Name,namesgenerator.GetRandomName(1)),
74+
AutostartSchedule:orig.AutostartSchedule,
75+
Ttl:orig.Ttl,
76+
})
77+
require.NoError(t,err,"insert workspace")
78+
returnworkspace
79+
}
80+
81+
funcWorkspaceBuild(t*testing.T,db database.Store,orig database.WorkspaceBuild) database.WorkspaceBuild {
82+
build,err:=db.InsertWorkspaceBuild(context.Background(), database.InsertWorkspaceBuildParams{
83+
ID:takeFirst(orig.ID,uuid.New()),
84+
CreatedAt:takeFirst(orig.CreatedAt,time.Now()),
85+
UpdatedAt:takeFirst(orig.UpdatedAt,time.Now()),
86+
WorkspaceID:takeFirst(orig.WorkspaceID,uuid.New()),
87+
TemplateVersionID:takeFirst(orig.TemplateVersionID,uuid.New()),
88+
BuildNumber:takeFirst(orig.BuildNumber,0),
89+
Transition:takeFirst(orig.Transition,database.WorkspaceTransitionStart),
90+
InitiatorID:takeFirst(orig.InitiatorID,uuid.New()),
91+
JobID:takeFirst(orig.JobID,uuid.New()),
92+
ProvisionerState:takeFirstBytes(orig.ProvisionerState, []byte{}),
93+
Deadline:takeFirst(orig.Deadline,time.Now().Add(time.Hour)),
94+
Reason:takeFirst(orig.Reason,database.BuildReasonInitiator),
95+
})
96+
require.NoError(t,err,"insert workspace build")
97+
returnbuild
98+
}
99+
100+
funcUser(t*testing.T,db database.Store,orig database.User) database.User {
101+
user,err:=db.InsertUser(context.Background(), database.InsertUserParams{
102+
ID:takeFirst(orig.ID,uuid.New()),
103+
Email:takeFirst(orig.Email,namesgenerator.GetRandomName(1)),
104+
Username:takeFirst(orig.Username,namesgenerator.GetRandomName(1)),
105+
HashedPassword:takeFirstBytes(orig.HashedPassword, []byte{}),
106+
CreatedAt:takeFirst(orig.CreatedAt,time.Now()),
107+
UpdatedAt:takeFirst(orig.UpdatedAt,time.Now()),
108+
RBACRoles: []string{},
109+
LoginType:takeFirst(orig.LoginType,database.LoginTypePassword),
110+
})
111+
require.NoError(t,err,"insert user")
112+
returnuser
113+
}
114+
115+
funcOrganization(t*testing.T,db database.Store,ctx context.Context,orig database.Organization) database.Organization {
116+
org,err:=db.InsertOrganization(context.Background(), database.InsertOrganizationParams{
117+
ID:takeFirst(orig.ID,uuid.New()),
118+
Name:takeFirst(orig.Name,namesgenerator.GetRandomName(1)),
119+
Description:takeFirst(orig.Description,namesgenerator.GetRandomName(1)),
120+
CreatedAt:takeFirst(orig.CreatedAt,time.Now()),
121+
UpdatedAt:takeFirst(orig.UpdatedAt,time.Now()),
122+
})
123+
require.NoError(t,err,"insert organization")
124+
returnorg
125+
}
126+
127+
funcGroup(t*testing.T,db database.Store,ctx context.Context,orig database.Group) database.Group {
128+
group,err:=db.InsertGroup(context.Background(), database.InsertGroupParams{
129+
ID:takeFirst(orig.ID,uuid.New()),
130+
Name:takeFirst(orig.Name,namesgenerator.GetRandomName(1)),
131+
OrganizationID:takeFirst(orig.OrganizationID,uuid.New()),
132+
AvatarURL:takeFirst(orig.AvatarURL,"https://logo.example.com"),
133+
QuotaAllowance:takeFirst(orig.QuotaAllowance,0),
134+
})
135+
require.NoError(t,err,"insert group")
136+
returngroup
137+
}
138+
139+
funcProvisionerJob(t*testing.T,db database.Store,ctx context.Context,orig database.ProvisionerJob) database.ProvisionerJob {
140+
job,err:=db.InsertProvisionerJob(context.Background(), database.InsertProvisionerJobParams{
141+
ID:takeFirst(orig.ID,uuid.New()),
142+
CreatedAt:takeFirst(orig.CreatedAt,time.Now()),
143+
UpdatedAt:takeFirst(orig.UpdatedAt,time.Now()),
144+
OrganizationID:takeFirst(orig.OrganizationID,uuid.New()),
145+
InitiatorID:takeFirst(orig.InitiatorID,uuid.New()),
146+
Provisioner:takeFirst(orig.Provisioner,database.ProvisionerTypeEcho),
147+
StorageMethod:takeFirst(orig.StorageMethod,database.ProvisionerStorageMethodFile),
148+
FileID:takeFirst(orig.FileID,uuid.New()),
149+
Type:takeFirst(orig.Type,database.ProvisionerJobTypeWorkspaceBuild),
150+
Input:takeFirstBytes(orig.Input, []byte("{}")),
151+
Tags:orig.Tags,
152+
})
153+
require.NoError(t,err,"insert job")
154+
returnjob
155+
}
156+
157+
funcWorkspaceResource(t*testing.T,db database.Store,ctx context.Context,orig database.WorkspaceResource) database.WorkspaceResource {
158+
resource,err:=db.InsertWorkspaceResource(context.Background(), database.InsertWorkspaceResourceParams{
159+
ID:takeFirst(orig.ID,uuid.New()),
160+
CreatedAt:takeFirst(orig.CreatedAt,time.Now()),
161+
JobID:takeFirst(orig.JobID,uuid.New()),
162+
Transition:takeFirst(orig.Transition,database.WorkspaceTransitionStart),
163+
Type:takeFirst(orig.Type,"fake_resource"),
164+
Name:takeFirst(orig.Name,namesgenerator.GetRandomName(1)),
165+
Hide:takeFirst(orig.Hide,false),
166+
Icon:takeFirst(orig.Icon,""),
167+
InstanceType: sql.NullString{
168+
String:takeFirst(orig.InstanceType.String,""),
169+
Valid:takeFirst(orig.InstanceType.Valid,false),
170+
},
171+
DailyCost:takeFirst(orig.DailyCost,0),
172+
})
173+
require.NoError(t,err,"insert resource")
174+
returnresource
175+
}
176+
177+
funcFile(t*testing.T,db database.Store,ctx context.Context,orig database.File) database.File {
178+
file,err:=db.InsertFile(context.Background(), database.InsertFileParams{
179+
ID:takeFirst(orig.ID,uuid.New()),
180+
Hash:takeFirst(orig.Hash,hex.EncodeToString(make([]byte,32))),
181+
CreatedAt:takeFirst(orig.CreatedAt,time.Now()),
182+
CreatedBy:takeFirst(orig.CreatedBy,uuid.New()),
183+
Mimetype:takeFirst(orig.Mimetype,"application/x-tar"),
184+
Data:takeFirstBytes(orig.Data, []byte{}),
185+
})
186+
require.NoError(t,err,"insert file")
187+
returnfile
188+
}
189+
190+
funcUserLink(t*testing.T,db database.Store,ctx context.Context,orig database.UserLink) database.UserLink {
191+
link,err:=db.InsertUserLink(context.Background(), database.InsertUserLinkParams{
192+
UserID:takeFirst(orig.UserID,uuid.New()),
193+
LoginType:takeFirst(orig.LoginType,database.LoginTypeGithub),
194+
LinkedID:takeFirst(orig.LinkedID),
195+
OAuthAccessToken:takeFirst(orig.OAuthAccessToken,uuid.NewString()),
196+
OAuthRefreshToken:takeFirst(orig.OAuthAccessToken,uuid.NewString()),
197+
OAuthExpiry:takeFirst(orig.OAuthExpiry,time.Now().Add(time.Hour*24)),
198+
})
199+
200+
require.NoError(t,err,"insert link")
201+
returnlink
202+
}
203+
204+
funcTemplateVersion(t*testing.T,db database.Store,orig database.TemplateVersion) database.TemplateVersion {
205+
version,err:=db.InsertTemplateVersion(context.Background(), database.InsertTemplateVersionParams{
206+
ID:takeFirst(orig.ID,uuid.New()),
207+
TemplateID: uuid.NullUUID{
208+
UUID:takeFirst(orig.TemplateID.UUID,uuid.New()),
209+
Valid:takeFirst(orig.TemplateID.Valid,true),
210+
},
211+
OrganizationID:takeFirst(orig.OrganizationID,uuid.New()),
212+
CreatedAt:takeFirst(orig.CreatedAt,time.Now()),
213+
UpdatedAt:takeFirst(orig.UpdatedAt,time.Now()),
214+
Name:takeFirst(orig.Name,namesgenerator.GetRandomName(1)),
215+
Readme:takeFirst(orig.Readme,namesgenerator.GetRandomName(1)),
216+
JobID:takeFirst(orig.JobID,uuid.New()),
217+
CreatedBy:takeFirst(orig.CreatedBy,uuid.New()),
218+
})
219+
require.NoError(t,err,"insert template version")
220+
returnversion
221+
}
Lines changed: 29 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,29 @@
1+
package fakegen
2+
3+
// takeFirstBytes implements takeFirst for []byte.
4+
// []byte is not a comparable type.
5+
functakeFirstBytes(values...[]byte) []byte {
6+
returntakeFirstF(values,func(v []byte)bool {
7+
returnlen(v)!=0
8+
})
9+
}
10+
11+
// takeFirstF takes the first value that returns true
12+
functakeFirstF[Valueany](values []Value,takefunc(vValue)bool)Value {
13+
varemptyValue
14+
for_,v:=rangevalues {
15+
iftake(v) {
16+
returnv
17+
}
18+
}
19+
// If all empty, return empty
20+
returnempty
21+
}
22+
23+
// takeFirst will take the first non-empty value.
24+
functakeFirst[Valuecomparable](values...Value)Value {
25+
varemptyValue
26+
returntakeFirstF(values,func(vValue)bool {
27+
returnv!=empty
28+
})
29+
}

‎coderd/database/databasefake/generator.go

Lines changed: 22 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -60,6 +60,28 @@ func Generate[Object Supported](t *testing.T, db database.Store, seed Object) Ob
6060
returnv
6161
}
6262

63+
funcTemplate(t*testing.T,db database.Store,seed database.Template) database.Template {
64+
template,err:=db.InsertTemplate(context.Background(), database.InsertTemplateParams{
65+
ID:takeFirst(seed.ID,uuid.New()),
66+
CreatedAt:takeFirst(seed.CreatedAt,time.Now()),
67+
UpdatedAt:takeFirst(seed.UpdatedAt,time.Now()),
68+
OrganizationID:takeFirst(seed.OrganizationID,uuid.New()),
69+
Name:takeFirst(seed.Name,namesgenerator.GetRandomName(1)),
70+
Provisioner:takeFirst(seed.Provisioner,database.ProvisionerTypeEcho),
71+
ActiveVersionID:takeFirst(seed.ActiveVersionID,uuid.New()),
72+
Description:takeFirst(seed.Description,namesgenerator.GetRandomName(1)),
73+
DefaultTTL:takeFirst(seed.DefaultTTL,3600),
74+
CreatedBy:takeFirst(seed.CreatedBy,uuid.New()),
75+
Icon:takeFirst(seed.Icon,namesgenerator.GetRandomName(1)),
76+
UserACL:seed.UserACL,
77+
GroupACL:seed.GroupACL,
78+
DisplayName:takeFirst(seed.DisplayName,namesgenerator.GetRandomName(1)),
79+
AllowUserCancelWorkspaceJobs:takeFirst(seed.AllowUserCancelWorkspaceJobs,true),
80+
})
81+
require.NoError(t,err,"insert template")
82+
returntemplate
83+
}
84+
6385
funcgenerate(t*testing.T,db database.Store,seedinterface{})interface{} {
6486
t.Helper()
6587

0 commit comments

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