@@ -13,7 +13,6 @@ import (
13
13
"time"
14
14
15
15
"github.com/google/uuid"
16
- "github.com/moby/moby/pkg/namesgenerator"
17
16
"github.com/sqlc-dev/pqtype"
18
17
"github.com/stretchr/testify/require"
19
18
@@ -25,6 +24,7 @@ import (
25
24
"github.com/coder/coder/v2/coderd/rbac"
26
25
"github.com/coder/coder/v2/coderd/rbac/policy"
27
26
"github.com/coder/coder/v2/cryptorand"
27
+ "github.com/coder/coder/v2/testutil"
28
28
)
29
29
30
30
// All methods take in a 'seed' object. Any provided fields in the seed will be
@@ -82,15 +82,15 @@ func Template(t testing.TB, db database.Store, seed database.Template) database.
82
82
CreatedAt :takeFirst (seed .CreatedAt ,dbtime .Now ()),
83
83
UpdatedAt :takeFirst (seed .UpdatedAt ,dbtime .Now ()),
84
84
OrganizationID :takeFirst (seed .OrganizationID ,uuid .New ()),
85
- Name :takeFirst (seed .Name ,namesgenerator .GetRandomName (1 )),
85
+ Name :takeFirst (seed .Name ,testutil .GetRandomName (t )),
86
86
Provisioner :takeFirst (seed .Provisioner ,database .ProvisionerTypeEcho ),
87
87
ActiveVersionID :takeFirst (seed .ActiveVersionID ,uuid .New ()),
88
- Description :takeFirst (seed .Description ,namesgenerator .GetRandomName (1 )),
88
+ Description :takeFirst (seed .Description ,testutil .GetRandomName (t )),
89
89
CreatedBy :takeFirst (seed .CreatedBy ,uuid .New ()),
90
- Icon :takeFirst (seed .Icon ,namesgenerator .GetRandomName (1 )),
90
+ Icon :takeFirst (seed .Icon ,testutil .GetRandomName (t )),
91
91
UserACL :seed .UserACL ,
92
92
GroupACL :seed .GroupACL ,
93
- DisplayName :takeFirst (seed .DisplayName ,namesgenerator .GetRandomName (1 )),
93
+ DisplayName :takeFirst (seed .DisplayName ,testutil .GetRandomName (t )),
94
94
AllowUserCancelWorkspaceJobs :seed .AllowUserCancelWorkspaceJobs ,
95
95
MaxPortSharingLevel :takeFirst (seed .MaxPortSharingLevel ,database .AppSharingLevelOwner ),
96
96
})
@@ -139,7 +139,7 @@ func APIKey(t testing.TB, db database.Store, seed database.APIKey) (key database
139
139
func WorkspaceAgentPortShare (t testing.TB ,db database.Store ,orig database.WorkspaceAgentPortShare ) database.WorkspaceAgentPortShare {
140
140
ps ,err := db .UpsertWorkspaceAgentPortShare (genCtx , database.UpsertWorkspaceAgentPortShareParams {
141
141
WorkspaceID :takeFirst (orig .WorkspaceID ,uuid .New ()),
142
- AgentName :takeFirst (orig .AgentName ,namesgenerator .GetRandomName (1 )),
142
+ AgentName :takeFirst (orig .AgentName ,testutil .GetRandomName (t )),
143
143
Port :takeFirst (orig .Port ,8080 ),
144
144
ShareLevel :takeFirst (orig .ShareLevel ,database .AppSharingLevelPublic ),
145
145
Protocol :takeFirst (orig .Protocol ,database .PortShareProtocolHttp ),
@@ -153,11 +153,11 @@ func WorkspaceAgent(t testing.TB, db database.Store, orig database.WorkspaceAgen
153
153
ID :takeFirst (orig .ID ,uuid .New ()),
154
154
CreatedAt :takeFirst (orig .CreatedAt ,dbtime .Now ()),
155
155
UpdatedAt :takeFirst (orig .UpdatedAt ,dbtime .Now ()),
156
- Name :takeFirst (orig .Name ,namesgenerator .GetRandomName (1 )),
156
+ Name :takeFirst (orig .Name ,testutil .GetRandomName (t )),
157
157
ResourceID :takeFirst (orig .ResourceID ,uuid .New ()),
158
158
AuthToken :takeFirst (orig .AuthToken ,uuid .New ()),
159
159
AuthInstanceID : sql.NullString {
160
- String :takeFirst (orig .AuthInstanceID .String ,namesgenerator .GetRandomName (1 )),
160
+ String :takeFirst (orig .AuthInstanceID .String ,testutil .GetRandomName (t )),
161
161
Valid :takeFirst (orig .AuthInstanceID .Valid ,true ),
162
162
},
163
163
Architecture :takeFirst (orig .Architecture ,"amd64" ),
@@ -196,7 +196,7 @@ func Workspace(t testing.TB, db database.Store, orig database.Workspace) databas
196
196
OrganizationID :takeFirst (orig .OrganizationID ,uuid .New ()),
197
197
TemplateID :takeFirst (orig .TemplateID ,uuid .New ()),
198
198
LastUsedAt :takeFirst (orig .LastUsedAt ,dbtime .Now ()),
199
- Name :takeFirst (orig .Name ,namesgenerator .GetRandomName (1 )),
199
+ Name :takeFirst (orig .Name ,testutil .GetRandomName (t )),
200
200
AutostartSchedule :orig .AutostartSchedule ,
201
201
Ttl :orig .Ttl ,
202
202
AutomaticUpdates :takeFirst (orig .AutomaticUpdates ,database .AutomaticUpdatesNever ),
@@ -210,8 +210,8 @@ func WorkspaceAgentLogSource(t testing.TB, db database.Store, orig database.Work
210
210
WorkspaceAgentID :takeFirst (orig .WorkspaceAgentID ,uuid .New ()),
211
211
ID : []uuid.UUID {takeFirst (orig .ID ,uuid .New ())},
212
212
CreatedAt :takeFirst (orig .CreatedAt ,dbtime .Now ()),
213
- DisplayName : []string {takeFirst (orig .DisplayName ,namesgenerator .GetRandomName (1 ))},
214
- Icon : []string {takeFirst (orig .Icon ,namesgenerator .GetRandomName (1 ))},
213
+ DisplayName : []string {takeFirst (orig .DisplayName ,testutil .GetRandomName (t ))},
214
+ Icon : []string {takeFirst (orig .Icon ,testutil .GetRandomName (t ))},
215
215
})
216
216
require .NoError (t ,err ,"insert workspace agent log source" )
217
217
return sources [0 ]
@@ -287,9 +287,9 @@ func WorkspaceBuildParameters(t testing.TB, db database.Store, orig []database.W
287
287
func User (t testing.TB ,db database.Store ,orig database.User ) database.User {
288
288
user ,err := db .InsertUser (genCtx , database.InsertUserParams {
289
289
ID :takeFirst (orig .ID ,uuid .New ()),
290
- Email :takeFirst (orig .Email ,namesgenerator .GetRandomName (1 )),
291
- Username :takeFirst (orig .Username ,namesgenerator .GetRandomName (1 )),
292
- Name :takeFirst (orig .Name ,namesgenerator .GetRandomName (1 )),
290
+ Email :takeFirst (orig .Email ,testutil .GetRandomName (t )),
291
+ Username :takeFirst (orig .Username ,testutil .GetRandomName (t )),
292
+ Name :takeFirst (orig .Name ,testutil .GetRandomName (t )),
293
293
HashedPassword :takeFirstSlice (orig .HashedPassword , []byte (must (cryptorand .String (32 )))),
294
294
CreatedAt :takeFirst (orig .CreatedAt ,dbtime .Now ()),
295
295
UpdatedAt :takeFirst (orig .UpdatedAt ,dbtime .Now ()),
@@ -336,9 +336,9 @@ func GitSSHKey(t testing.TB, db database.Store, orig database.GitSSHKey) databas
336
336
func Organization (t testing.TB ,db database.Store ,orig database.Organization ) database.Organization {
337
337
org ,err := db .InsertOrganization (genCtx , database.InsertOrganizationParams {
338
338
ID :takeFirst (orig .ID ,uuid .New ()),
339
- Name :takeFirst (orig .Name ,namesgenerator .GetRandomName (1 )),
340
- DisplayName :takeFirst (orig .Name ,namesgenerator .GetRandomName (1 )),
341
- Description :takeFirst (orig .Description ,namesgenerator .GetRandomName (1 )),
339
+ Name :takeFirst (orig .Name ,testutil .GetRandomName (t )),
340
+ DisplayName :takeFirst (orig .Name ,testutil .GetRandomName (t )),
341
+ Description :takeFirst (orig .Description ,testutil .GetRandomName (t )),
342
342
Icon :takeFirst (orig .Icon ,"" ),
343
343
CreatedAt :takeFirst (orig .CreatedAt ,dbtime .Now ()),
344
344
UpdatedAt :takeFirst (orig .UpdatedAt ,dbtime .Now ()),
@@ -360,7 +360,7 @@ func OrganizationMember(t testing.TB, db database.Store, orig database.Organizat
360
360
}
361
361
362
362
func Group (t testing.TB ,db database.Store ,orig database.Group ) database.Group {
363
- name := takeFirst (orig .Name ,namesgenerator .GetRandomName (1 ))
363
+ name := takeFirst (orig .Name ,testutil .GetRandomName (t ))
364
364
group ,err := db .InsertGroup (genCtx , database.InsertGroupParams {
365
365
ID :takeFirst (orig .ID ,uuid .New ()),
366
366
Name :name ,
@@ -470,7 +470,7 @@ func ProvisionerKey(t testing.TB, db database.Store, orig database.ProvisionerKe
470
470
ID :takeFirst (orig .ID ,uuid .New ()),
471
471
CreatedAt :takeFirst (orig .CreatedAt ,dbtime .Now ()),
472
472
OrganizationID :takeFirst (orig .OrganizationID ,uuid .New ()),
473
- Name :takeFirst (orig .Name ,namesgenerator .GetRandomName (1 )),
473
+ Name :takeFirst (orig .Name ,testutil .GetRandomName (t )),
474
474
HashedSecret :orig .HashedSecret ,
475
475
Tags :orig .Tags ,
476
476
})
@@ -483,9 +483,9 @@ func WorkspaceApp(t testing.TB, db database.Store, orig database.WorkspaceApp) d
483
483
ID :takeFirst (orig .ID ,uuid .New ()),
484
484
CreatedAt :takeFirst (orig .CreatedAt ,dbtime .Now ()),
485
485
AgentID :takeFirst (orig .AgentID ,uuid .New ()),
486
- Slug :takeFirst (orig .Slug ,namesgenerator .GetRandomName (1 )),
487
- DisplayName :takeFirst (orig .DisplayName ,namesgenerator .GetRandomName (1 )),
488
- Icon :takeFirst (orig .Icon ,namesgenerator .GetRandomName (1 )),
486
+ Slug :takeFirst (orig .Slug ,testutil .GetRandomName (t )),
487
+ DisplayName :takeFirst (orig .DisplayName ,testutil .GetRandomName (t )),
488
+ Icon :takeFirst (orig .Icon ,testutil .GetRandomName (t )),
489
489
Command : sql.NullString {
490
490
String :takeFirst (orig .Command .String ,"ls" ),
491
491
Valid :orig .Command .Valid ,
@@ -546,7 +546,7 @@ func WorkspaceResource(t testing.TB, db database.Store, orig database.WorkspaceR
546
546
JobID :takeFirst (orig .JobID ,uuid .New ()),
547
547
Transition :takeFirst (orig .Transition ,database .WorkspaceTransitionStart ),
548
548
Type :takeFirst (orig .Type ,"fake_resource" ),
549
- Name :takeFirst (orig .Name ,namesgenerator .GetRandomName (1 )),
549
+ Name :takeFirst (orig .Name ,testutil .GetRandomName (t )),
550
550
Hide :takeFirst (orig .Hide ,false ),
551
551
Icon :takeFirst (orig .Icon ,"" ),
552
552
InstanceType : sql.NullString {
@@ -562,8 +562,8 @@ func WorkspaceResource(t testing.TB, db database.Store, orig database.WorkspaceR
562
562
func WorkspaceResourceMetadatums (t testing.TB ,db database.Store ,seed database.WorkspaceResourceMetadatum ) []database.WorkspaceResourceMetadatum {
563
563
meta ,err := db .InsertWorkspaceResourceMetadata (genCtx , database.InsertWorkspaceResourceMetadataParams {
564
564
WorkspaceResourceID :takeFirst (seed .WorkspaceResourceID ,uuid .New ()),
565
- Key : []string {takeFirst (seed .Key ,namesgenerator .GetRandomName (1 ))},
566
- Value : []string {takeFirst (seed .Value .String ,namesgenerator .GetRandomName (1 ))},
565
+ Key : []string {takeFirst (seed .Key ,testutil .GetRandomName (t ))},
566
+ Value : []string {takeFirst (seed .Value .String ,testutil .GetRandomName (t ))},
567
567
Sensitive : []bool {takeFirst (seed .Sensitive ,false )},
568
568
})
569
569
require .NoError (t ,err ,"insert meta data" )
@@ -577,9 +577,9 @@ func WorkspaceProxy(t testing.TB, db database.Store, orig database.WorkspaceProx
577
577
578
578
proxy ,err := db .InsertWorkspaceProxy (genCtx , database.InsertWorkspaceProxyParams {
579
579
ID :takeFirst (orig .ID ,uuid .New ()),
580
- Name :takeFirst (orig .Name ,namesgenerator .GetRandomName (1 )),
581
- DisplayName :takeFirst (orig .DisplayName ,namesgenerator .GetRandomName (1 )),
582
- Icon :takeFirst (orig .Icon ,namesgenerator .GetRandomName (1 )),
580
+ Name :takeFirst (orig .Name ,testutil .GetRandomName (t )),
581
+ DisplayName :takeFirst (orig .DisplayName ,testutil .GetRandomName (t )),
582
+ Icon :takeFirst (orig .Icon ,testutil .GetRandomName (t )),
583
583
TokenHashedSecret :hashedSecret [:],
584
584
CreatedAt :takeFirst (orig .CreatedAt ,dbtime .Now ()),
585
585
UpdatedAt :takeFirst (orig .UpdatedAt ,dbtime .Now ()),
@@ -659,9 +659,9 @@ func TemplateVersion(t testing.TB, db database.Store, orig database.TemplateVers
659
659
OrganizationID :takeFirst (orig .OrganizationID ,uuid .New ()),
660
660
CreatedAt :takeFirst (orig .CreatedAt ,dbtime .Now ()),
661
661
UpdatedAt :takeFirst (orig .UpdatedAt ,dbtime .Now ()),
662
- Name :takeFirst (orig .Name ,namesgenerator .GetRandomName (1 )),
662
+ Name :takeFirst (orig .Name ,testutil .GetRandomName (t )),
663
663
Message :orig .Message ,
664
- Readme :takeFirst (orig .Readme ,namesgenerator .GetRandomName (1 )),
664
+ Readme :takeFirst (orig .Readme ,testutil .GetRandomName (t )),
665
665
JobID :takeFirst (orig .JobID ,uuid .New ()),
666
666
CreatedBy :takeFirst (orig .CreatedBy ,uuid .New ()),
667
667
})
@@ -683,11 +683,11 @@ func TemplateVersion(t testing.TB, db database.Store, orig database.TemplateVers
683
683
func TemplateVersionVariable (t testing.TB ,db database.Store ,orig database.TemplateVersionVariable ) database.TemplateVersionVariable {
684
684
version ,err := db .InsertTemplateVersionVariable (genCtx , database.InsertTemplateVersionVariableParams {
685
685
TemplateVersionID :takeFirst (orig .TemplateVersionID ,uuid .New ()),
686
- Name :takeFirst (orig .Name ,namesgenerator .GetRandomName (1 )),
687
- Description :takeFirst (orig .Description ,namesgenerator .GetRandomName (1 )),
686
+ Name :takeFirst (orig .Name ,testutil .GetRandomName (t )),
687
+ Description :takeFirst (orig .Description ,testutil .GetRandomName (t )),
688
688
Type :takeFirst (orig .Type ,"string" ),
689
689
Value :takeFirst (orig .Value ,"" ),
690
- DefaultValue :takeFirst (orig .DefaultValue ,namesgenerator .GetRandomName (1 )),
690
+ DefaultValue :takeFirst (orig .DefaultValue ,testutil .GetRandomName (t )),
691
691
Required :takeFirst (orig .Required ,false ),
692
692
Sensitive :takeFirst (orig .Sensitive ,false ),
693
693
})
@@ -698,8 +698,8 @@ func TemplateVersionVariable(t testing.TB, db database.Store, orig database.Temp
698
698
func TemplateVersionWorkspaceTag (t testing.TB ,db database.Store ,orig database.TemplateVersionWorkspaceTag ) database.TemplateVersionWorkspaceTag {
699
699
workspaceTag ,err := db .InsertTemplateVersionWorkspaceTag (genCtx , database.InsertTemplateVersionWorkspaceTagParams {
700
700
TemplateVersionID :takeFirst (orig .TemplateVersionID ,uuid .New ()),
701
- Key :takeFirst (orig .Key ,namesgenerator .GetRandomName (1 )),
702
- Value :takeFirst (orig .Value ,namesgenerator .GetRandomName (1 )),
701
+ Key :takeFirst (orig .Key ,testutil .GetRandomName (t )),
702
+ Value :takeFirst (orig .Value ,testutil .GetRandomName (t )),
703
703
})
704
704
require .NoError (t ,err ,"insert template version workspace tag" )
705
705
return workspaceTag
@@ -710,20 +710,20 @@ func TemplateVersionParameter(t testing.TB, db database.Store, orig database.Tem
710
710
711
711
version ,err := db .InsertTemplateVersionParameter (genCtx , database.InsertTemplateVersionParameterParams {
712
712
TemplateVersionID :takeFirst (orig .TemplateVersionID ,uuid .New ()),
713
- Name :takeFirst (orig .Name ,namesgenerator .GetRandomName (1 )),
714
- Description :takeFirst (orig .Description ,namesgenerator .GetRandomName (1 )),
713
+ Name :takeFirst (orig .Name ,testutil .GetRandomName (t )),
714
+ Description :takeFirst (orig .Description ,testutil .GetRandomName (t )),
715
715
Type :takeFirst (orig .Type ,"string" ),
716
716
Mutable :takeFirst (orig .Mutable ,false ),
717
- DefaultValue :takeFirst (orig .DefaultValue ,namesgenerator .GetRandomName (1 )),
718
- Icon :takeFirst (orig .Icon ,namesgenerator .GetRandomName (1 )),
717
+ DefaultValue :takeFirst (orig .DefaultValue ,testutil .GetRandomName (t )),
718
+ Icon :takeFirst (orig .Icon ,testutil .GetRandomName (t )),
719
719
Options :takeFirstSlice (orig .Options , []byte ("[]" )),
720
720
ValidationRegex :takeFirst (orig .ValidationRegex ,"" ),
721
721
ValidationMin :takeFirst (orig .ValidationMin , sql.NullInt32 {}),
722
722
ValidationMax :takeFirst (orig .ValidationMax , sql.NullInt32 {}),
723
723
ValidationError :takeFirst (orig .ValidationError ,"" ),
724
724
ValidationMonotonic :takeFirst (orig .ValidationMonotonic ,"" ),
725
725
Required :takeFirst (orig .Required ,false ),
726
- DisplayName :takeFirst (orig .DisplayName ,namesgenerator .GetRandomName (1 )),
726
+ DisplayName :takeFirst (orig .DisplayName ,testutil .GetRandomName (t )),
727
727
DisplayOrder :takeFirst (orig .DisplayOrder ,0 ),
728
728
Ephemeral :takeFirst (orig .Ephemeral ,false ),
729
729
})
@@ -783,7 +783,7 @@ func WorkspaceAgentStat(t testing.TB, db database.Store, orig database.Workspace
783
783
func OAuth2ProviderApp (t testing.TB ,db database.Store ,seed database.OAuth2ProviderApp ) database.OAuth2ProviderApp {
784
784
app ,err := db .InsertOAuth2ProviderApp (genCtx , database.InsertOAuth2ProviderAppParams {
785
785
ID :takeFirst (seed .ID ,uuid .New ()),
786
- Name :takeFirst (seed .Name ,namesgenerator .GetRandomName (1 )),
786
+ Name :takeFirst (seed .Name ,testutil .GetRandomName (t )),
787
787
CreatedAt :takeFirst (seed .CreatedAt ,dbtime .Now ()),
788
788
UpdatedAt :takeFirst (seed .UpdatedAt ,dbtime .Now ()),
789
789
Icon :takeFirst (seed .Icon ,"" ),
@@ -836,8 +836,8 @@ func OAuth2ProviderAppToken(t testing.TB, db database.Store, seed database.OAuth
836
836
837
837
func CustomRole (t testing.TB ,db database.Store ,seed database.CustomRole ) database.CustomRole {
838
838
role ,err := db .UpsertCustomRole (genCtx , database.UpsertCustomRoleParams {
839
- Name :takeFirst (seed .Name ,strings .ToLower (namesgenerator .GetRandomName (1 ))),
840
- DisplayName :namesgenerator .GetRandomName (1 ),
839
+ Name :takeFirst (seed .Name ,strings .ToLower (testutil .GetRandomName (t ))),
840
+ DisplayName :testutil .GetRandomName (t ),
841
841
OrganizationID :seed .OrganizationID ,
842
842
SitePermissions :takeFirstSlice (seed .SitePermissions , []database.CustomRolePermission {}),
843
843
OrgPermissions :takeFirstSlice (seed .SitePermissions , []database.CustomRolePermission {}),