@@ -3989,9 +3989,34 @@ func (s *MethodTestSuite) TestSystemFunctions() {
3989
3989
}).Asserts (rbac .ResourceSystem ,policy .ActionCreate )
3990
3990
}))
3991
3991
s .Run ("InsertWorkspaceAgentDevcontainers" ,s .Subtest (func (db database.Store ,check * expects ) {
3992
- a := dbgen .WorkspaceAgent (s .T (),db , database.WorkspaceAgent {})
3992
+ u := dbgen .User (s .T (),db , database.User {})
3993
+ o := dbgen .Organization (s .T (),db , database.Organization {})
3994
+ tpl := dbgen .Template (s .T (),db , database.Template {
3995
+ OrganizationID :o .ID ,
3996
+ CreatedBy :u .ID ,
3997
+ })
3998
+ tv := dbgen .TemplateVersion (s .T (),db , database.TemplateVersion {
3999
+ TemplateID : uuid.NullUUID {UUID :tpl .ID ,Valid :true },
4000
+ OrganizationID :o .ID ,
4001
+ CreatedBy :u .ID ,
4002
+ })
4003
+ w := dbgen .Workspace (s .T (),db , database.WorkspaceTable {
4004
+ TemplateID :tpl .ID ,
4005
+ OrganizationID :o .ID ,
4006
+ OwnerID :u .ID ,
4007
+ })
4008
+ j := dbgen .ProvisionerJob (s .T (),db ,nil , database.ProvisionerJob {
4009
+ Type :database .ProvisionerJobTypeWorkspaceBuild ,
4010
+ })
4011
+ b := dbgen .WorkspaceBuild (s .T (),db , database.WorkspaceBuild {
4012
+ JobID :j .ID ,
4013
+ WorkspaceID :w .ID ,
4014
+ TemplateVersionID :tv .ID ,
4015
+ })
4016
+ res := dbgen .WorkspaceResource (s .T (),db , database.WorkspaceResource {JobID :b .JobID })
4017
+ agt := dbgen .WorkspaceAgent (s .T (),db , database.WorkspaceAgent {ResourceID :res .ID })
3993
4018
check .Args (database.InsertWorkspaceAgentDevcontainersParams {
3994
- WorkspaceAgentID :a .ID ,
4019
+ WorkspaceAgentID :agt .ID ,
3995
4020
}).Asserts (rbac .ResourceSystem ,policy .ActionCreate )
3996
4021
}))
3997
4022
s .Run ("UpdateWorkspaceAgentConnectionByID" ,s .Subtest (func (db database.Store ,check * expects ) {