@@ -1655,10 +1655,7 @@ func (s *MethodTestSuite) TestWorkspace() {
1655
1655
build := dbgen .WorkspaceBuild (s .T (),db , database.WorkspaceBuild {WorkspaceID :ws .ID ,JobID :uuid .New ()})
1656
1656
res := dbgen .WorkspaceResource (s .T (),db , database.WorkspaceResource {JobID :build .JobID })
1657
1657
agt := dbgen .WorkspaceAgent (s .T (),db , database.WorkspaceAgent {ResourceID :res .ID })
1658
- check .Args (agt .ID ).Asserts (ws ,policy .ActionRead ).Returns (database.Workspace {
1659
- Workspace :ws ,
1660
- TemplateName :tpl .Name ,
1661
- })
1658
+ check .Args (agt .ID ).Asserts (ws ,policy .ActionRead )
1662
1659
}))
1663
1660
s .Run ("GetWorkspaceAgentsInLatestBuildByWorkspaceID" ,s .Subtest (func (db database.Store ,check * expects ) {
1664
1661
tpl := dbgen .Template (s .T (),db , database.Template {})
@@ -1676,7 +1673,7 @@ func (s *MethodTestSuite) TestWorkspace() {
1676
1673
OwnerID :ws .OwnerID ,
1677
1674
Deleted :ws .Deleted ,
1678
1675
Name :ws .Name ,
1679
- }).Asserts (ws ,policy .ActionRead ). Returns ( ws )
1676
+ }).Asserts (ws ,policy .ActionRead )
1680
1677
}))
1681
1678
s .Run ("GetWorkspaceResourceByID" ,s .Subtest (func (db database.Store ,check * expects ) {
1682
1679
ws := dbgen .Workspace (s .T (),db , database.WorkspaceTable {})
@@ -1881,7 +1878,7 @@ func (s *MethodTestSuite) TestWorkspace() {
1881
1878
res := dbgen .WorkspaceResource (s .T (),db , database.WorkspaceResource {JobID :build .JobID })
1882
1879
agt := dbgen .WorkspaceAgent (s .T (),db , database.WorkspaceAgent {ResourceID :res .ID })
1883
1880
app := dbgen .WorkspaceApp (s .T (),db , database.WorkspaceApp {AgentID :agt .ID })
1884
- check .Args (app .ID ).Asserts (ws ,policy .ActionRead ). Returns ( ws )
1881
+ check .Args (app .ID ).Asserts (ws ,policy .ActionRead )
1885
1882
}))
1886
1883
s .Run ("ActivityBumpWorkspace" ,s .Subtest (func (db database.Store ,check * expects ) {
1887
1884
ws := dbgen .Workspace (s .T (),db , database.WorkspaceTable {})
@@ -2457,13 +2454,13 @@ func (s *MethodTestSuite) TestSystemFunctions() {
2457
2454
Asserts (tpl ,policy .ActionRead ).Errors (sql .ErrNoRows )
2458
2455
}))
2459
2456
s .Run ("GetWorkspaceAppsByAgentIDs" ,s .Subtest (func (db database.Store ,check * expects ) {
2460
- aWs := dbgen .Workspace (s .T (),db , database.Workspace {})
2457
+ aWs := dbgen .Workspace (s .T (),db , database.WorkspaceTable {})
2461
2458
aBuild := dbgen .WorkspaceBuild (s .T (),db , database.WorkspaceBuild {WorkspaceID :aWs .ID ,JobID :uuid .New ()})
2462
2459
aRes := dbgen .WorkspaceResource (s .T (),db , database.WorkspaceResource {JobID :aBuild .JobID })
2463
2460
aAgt := dbgen .WorkspaceAgent (s .T (),db , database.WorkspaceAgent {ResourceID :aRes .ID })
2464
2461
a := dbgen .WorkspaceApp (s .T (),db , database.WorkspaceApp {AgentID :aAgt .ID })
2465
2462
2466
- bWs := dbgen .Workspace (s .T (),db , database.Workspace {})
2463
+ bWs := dbgen .Workspace (s .T (),db , database.WorkspaceTable {})
2467
2464
bBuild := dbgen .WorkspaceBuild (s .T (),db , database.WorkspaceBuild {WorkspaceID :bWs .ID ,JobID :uuid .New ()})
2468
2465
bRes := dbgen .WorkspaceResource (s .T (),db , database.WorkspaceResource {JobID :bBuild .JobID })
2469
2466
bAgt := dbgen .WorkspaceAgent (s .T (),db , database.WorkspaceAgent {ResourceID :bRes .ID })
@@ -2478,15 +2475,15 @@ func (s *MethodTestSuite) TestSystemFunctions() {
2478
2475
v := dbgen .TemplateVersion (s .T (),db , database.TemplateVersion {TemplateID : uuid.NullUUID {UUID :tpl .ID ,Valid :true },JobID :uuid .New ()})
2479
2476
tJob := dbgen .ProvisionerJob (s .T (),db ,nil , database.ProvisionerJob {ID :v .JobID ,Type :database .ProvisionerJobTypeTemplateVersionImport })
2480
2477
2481
- ws := dbgen .Workspace (s .T (),db , database.Workspace {})
2478
+ ws := dbgen .Workspace (s .T (),db , database.WorkspaceTable {})
2482
2479
build := dbgen .WorkspaceBuild (s .T (),db , database.WorkspaceBuild {WorkspaceID :ws .ID ,JobID :uuid .New ()})
2483
2480
wJob := dbgen .ProvisionerJob (s .T (),db ,nil , database.ProvisionerJob {ID :build .JobID ,Type :database .ProvisionerJobTypeWorkspaceBuild })
2484
2481
check .Args ([]uuid.UUID {tJob .ID ,wJob .ID }).
2485
2482
Asserts (rbac .ResourceSystem ,policy .ActionRead ).
2486
2483
Returns ([]database.WorkspaceResource {})
2487
2484
}))
2488
2485
s .Run ("GetWorkspaceResourceMetadataByResourceIDs" ,s .Subtest (func (db database.Store ,check * expects ) {
2489
- ws := dbgen .Workspace (s .T (),db , database.Workspace {})
2486
+ ws := dbgen .Workspace (s .T (),db , database.WorkspaceTable {})
2490
2487
build := dbgen .WorkspaceBuild (s .T (),db , database.WorkspaceBuild {WorkspaceID :ws .ID ,JobID :uuid .New ()})
2491
2488
_ = dbgen .ProvisionerJob (s .T (),db ,nil , database.ProvisionerJob {ID :build .JobID ,Type :database .ProvisionerJobTypeWorkspaceBuild })
2492
2489
a := dbgen .WorkspaceResource (s .T (),db , database.WorkspaceResource {JobID :build .JobID })
@@ -2495,7 +2492,7 @@ func (s *MethodTestSuite) TestSystemFunctions() {
2495
2492
Asserts (rbac .ResourceSystem ,policy .ActionRead )
2496
2493
}))
2497
2494
s .Run ("GetWorkspaceAgentsByResourceIDs" ,s .Subtest (func (db database.Store ,check * expects ) {
2498
- ws := dbgen .Workspace (s .T (),db , database.Workspace {})
2495
+ ws := dbgen .Workspace (s .T (),db , database.WorkspaceTable {})
2499
2496
build := dbgen .WorkspaceBuild (s .T (),db , database.WorkspaceBuild {WorkspaceID :ws .ID ,JobID :uuid .New ()})
2500
2497
res := dbgen .WorkspaceResource (s .T (),db , database.WorkspaceResource {JobID :build .JobID })
2501
2498
agt := dbgen .WorkspaceAgent (s .T (),db , database.WorkspaceAgent {ResourceID :res .ID })
@@ -2529,7 +2526,7 @@ func (s *MethodTestSuite) TestSystemFunctions() {
2529
2526
}).Asserts (rbac .ResourceSystem ,policy .ActionCreate )
2530
2527
}))
2531
2528
s .Run ("UpdateWorkspaceAgentConnectionByID" ,s .Subtest (func (db database.Store ,check * expects ) {
2532
- ws := dbgen .Workspace (s .T (),db , database.Workspace {})
2529
+ ws := dbgen .Workspace (s .T (),db , database.WorkspaceTable {})
2533
2530
build := dbgen .WorkspaceBuild (s .T (),db , database.WorkspaceBuild {WorkspaceID :ws .ID ,JobID :uuid .New ()})
2534
2531
res := dbgen .WorkspaceResource (s .T (),db , database.WorkspaceResource {JobID :build .JobID })
2535
2532
agt := dbgen .WorkspaceAgent (s .T (),db , database.WorkspaceAgent {ResourceID :res .ID })
@@ -2772,7 +2769,7 @@ func (s *MethodTestSuite) TestSystemFunctions() {
2772
2769
check .Args (uuid .New ()).Asserts (rbac .ResourceSystem ,policy .ActionRead )
2773
2770
}))
2774
2771
s .Run ("GetJFrogXrayScanByWorkspaceAndAgentID" ,s .Subtest (func (db database.Store ,check * expects ) {
2775
- ws := dbgen .Workspace (s .T (),db , database.Workspace {})
2772
+ ws := dbgen .Workspace (s .T (),db , database.WorkspaceTable {})
2776
2773
agent := dbgen .WorkspaceAgent (s .T (),db , database.WorkspaceAgent {})
2777
2774
2778
2775
err := db .UpsertJFrogXrayScanByWorkspaceAndAgentID (context .Background (), database.UpsertJFrogXrayScanByWorkspaceAndAgentIDParams {