@@ -2955,34 +2955,38 @@ func (s *MethodTestSuite) TestWorkspace() {
29552955Deadline :b .Deadline ,
29562956}).Asserts (w ,policy .ActionUpdate )
29572957}))
2958- s .Run ("UpdateWorkspaceBuildFlagsByID" ,s .Subtest (func (db database. Store ,check * expects ) {
2959- u := dbgen . User (s .T (),db , database.User {})
2960- o := dbgen . Organization (s .T (),db , database.Organization {})
2961- tpl := dbgen . Template (s .T (),db , database.Template {
2958+ s .Run ("UpdateWorkspaceBuildFlagsByID" ,s .Mocked (func (dbm * dbmock. MockStore , faker * gofakeit. Faker ,check * expects ) {
2959+ u := testutil . Fake (s .T (),faker , database.User {})
2960+ o := testutil . Fake (s .T (),faker , database.Organization {})
2961+ tpl := testutil . Fake (s .T (),faker , database.Template {
29622962OrganizationID :o .ID ,
29632963CreatedBy :u .ID ,
29642964})
2965- tv := dbgen . TemplateVersion (s .T (),db , database.TemplateVersion {
2965+ tv := testutil . Fake (s .T (),faker , database.TemplateVersion {
29662966TemplateID : uuid.NullUUID {UUID :tpl .ID ,Valid :true },
29672967OrganizationID :o .ID ,
29682968CreatedBy :u .ID ,
29692969})
2970- w := dbgen . Workspace (s .T (),db , database.WorkspaceTable {
2970+ w := testutil . Fake (s .T (),faker , database.Workspace {
29712971TemplateID :tpl .ID ,
29722972OrganizationID :o .ID ,
29732973OwnerID :u .ID ,
29742974})
2975- j := dbgen . ProvisionerJob (s .T (),db , nil , database.ProvisionerJob {
2975+ j := testutil . Fake (s .T (),faker , database.ProvisionerJob {
29762976Type :database .ProvisionerJobTypeWorkspaceBuild ,
29772977})
2978- b := dbgen . WorkspaceBuild (s .T (),db , database.WorkspaceBuild {
2978+ b := testutil . Fake (s .T (),faker , database.WorkspaceBuild {
29792979JobID :j .ID ,
29802980WorkspaceID :w .ID ,
29812981TemplateVersionID :tv .ID ,
29822982})
2983- res := dbgen .WorkspaceResource (s .T (),db , database.WorkspaceResource {JobID :b .JobID })
2984- agt := dbgen .WorkspaceAgent (s .T (),db , database.WorkspaceAgent {ResourceID :res .ID })
2985- app := dbgen .WorkspaceApp (s .T (),db , database.WorkspaceApp {AgentID :agt .ID })
2983+ res := testutil .Fake (s .T (),faker , database.WorkspaceResource {JobID :b .JobID })
2984+ agt := testutil .Fake (s .T (),faker , database.WorkspaceAgent {ResourceID :res .ID })
2985+ app := testutil .Fake (s .T (),faker , database.WorkspaceApp {AgentID :agt .ID })
2986+
2987+ dbm .EXPECT ().GetWorkspaceByID (gomock .Any (),w .ID ).Return (w ,nil ).AnyTimes ()
2988+ dbm .EXPECT ().GetWorkspaceBuildByID (gomock .Any (),b .ID ).Return (b ,nil ).AnyTimes ()
2989+ dbm .EXPECT ().UpdateWorkspaceBuildFlagsByID (gomock .Any (),gomock .Any ()).Return (nil ).AnyTimes ()
29862990check .Args (database.UpdateWorkspaceBuildFlagsByIDParams {
29872991ID :b .ID ,
29882992HasAITask : sql.NullBool {Bool :true ,Valid :true },