@@ -1047,6 +1047,11 @@ func (s *MethodTestSuite) TestTemplate() {
1047
1047
dbm .EXPECT ().GetTemplateByID (gomock .Any (), t1 .ID ).Return (t1 , nil ).AnyTimes ()
1048
1048
check .Args (tv .ID ).Asserts (t1 , policy .ActionRead ).Returns (tv )
1049
1049
}))
1050
+ s .Run ("Orphaned/GetTemplateVersionByID" , s .Mocked (func (dbm * dbmock.MockStore , faker * gofakeit.Faker , check * expects ) {
1051
+ tv := testutil .Fake (s .T (), faker , database.TemplateVersion {TemplateID : uuid.NullUUID {Valid : false }})
1052
+ dbm .EXPECT ().GetTemplateVersionByID (gomock .Any (), tv .ID ).Return (tv , nil ).AnyTimes ()
1053
+ check .Args (tv .ID ).Asserts (tv .RBACObjectNoTemplate (), policy .ActionRead ).Returns (tv )
1054
+ }))
1050
1055
s .Run ("GetTemplateVersionsByTemplateID" , s .Mocked (func (dbm * dbmock.MockStore , faker * gofakeit.Faker , check * expects ) {
1051
1056
t1 := testutil .Fake (s .T (), faker , database.Template {})
1052
1057
a := testutil .Fake (s .T (), faker , database.TemplateVersion {TemplateID : uuid.NullUUID {UUID : t1 .ID , Valid : true }})
@@ -1076,7 +1081,7 @@ func (s *MethodTestSuite) TestTemplate() {
1076
1081
check .Args (arg , emptyPreparedAuthorized {}).Asserts ().Returns (slice .New (a ))
1077
1082
}))
1078
1083
s .Run ("InsertTemplate" , s .Mocked (func (dbm * dbmock.MockStore , _ * gofakeit.Faker , check * expects ) {
1079
- arg := database.InsertTemplateParams {Provisioner : "echo" , OrganizationID : uuid .New (), MaxPortSharingLevel : database . AppSharingLevelOwner , CorsBehavior : database . CorsBehaviorSimple }
1084
+ arg := database.InsertTemplateParams {OrganizationID : uuid .New ()}
1080
1085
dbm .EXPECT ().InsertTemplate (gomock .Any (), arg ).Return (nil ).AnyTimes ()
1081
1086
check .Args (arg ).Asserts (rbac .ResourceTemplate .InOrg (arg .OrganizationID ), policy .ActionCreate )
1082
1087
}))
0 commit comments