@@ -1266,6 +1266,11 @@ func (s *MethodTestSuite) TestTemplate() {
1266
1266
dbm .EXPECT ().GetTemplateByID (gomock .Any (), t1 .ID ).Return (t1 , nil ).AnyTimes ()
1267
1267
check .Args (tv .ID ).Asserts (t1 , policy .ActionRead ).Returns (tv )
1268
1268
}))
1269
+ s .Run ("Orphaned/GetTemplateVersionByID" , s .Mocked (func (dbm * dbmock.MockStore , faker * gofakeit.Faker , check * expects ) {
1270
+ tv := testutil .Fake (s .T (), faker , database.TemplateVersion {TemplateID : uuid.NullUUID {Valid : false }})
1271
+ dbm .EXPECT ().GetTemplateVersionByID (gomock .Any (), tv .ID ).Return (tv , nil ).AnyTimes ()
1272
+ check .Args (tv .ID ).Asserts (tv .RBACObjectNoTemplate (), policy .ActionRead ).Returns (tv )
1273
+ }))
1269
1274
s .Run ("GetTemplateVersionsByTemplateID" , s .Mocked (func (dbm * dbmock.MockStore , faker * gofakeit.Faker , check * expects ) {
1270
1275
t1 := testutil .Fake (s .T (), faker , database.Template {})
1271
1276
a := testutil .Fake (s .T (), faker , database.TemplateVersion {TemplateID : uuid.NullUUID {UUID : t1 .ID , Valid : true }})
@@ -1303,7 +1308,7 @@ func (s *MethodTestSuite) TestTemplate() {
1303
1308
check .Args (arg , emptyPreparedAuthorized {}).Asserts ().Returns (slice .New (a ))
1304
1309
}))
1305
1310
s .Run ("InsertTemplate" , s .Mocked (func (dbm * dbmock.MockStore , _ * gofakeit.Faker , check * expects ) {
1306
- arg := database.InsertTemplateParams {Provisioner : "echo" , OrganizationID : uuid .New (), MaxPortSharingLevel : database . AppSharingLevelOwner , CorsBehavior : database . CorsBehaviorSimple }
1311
+ arg := database.InsertTemplateParams {OrganizationID : uuid .New ()}
1307
1312
dbm .EXPECT ().InsertTemplate (gomock .Any (), arg ).Return (nil ).AnyTimes ()
1308
1313
check .Args (arg ).Asserts (rbac .ResourceTemplate .InOrg (arg .OrganizationID ), policy .ActionCreate )
1309
1314
}))
0 commit comments