Skip to content

Commit ecf392f

Browse files
authored
testserver: volumes: include volume_id (#4224)
## Why Makes it more similar to real volumes, helps with local+cloud tests.
1 parent 4e6ae9c commit ecf392f

File tree

7 files changed

+16
-10
lines changed

7 files changed

+16
-10
lines changed

acceptance/bundle/migrate/basic/out.plan_update.json

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -209,6 +209,7 @@
209209
"schema_name": "myschema",
210210
"storage_location": "s3://deco-uc-prod-isolated-aws-us-east-1/metastore/[UUID]/volumes/[UUID]",
211211
"updated_at": [UNIX_TIME_MILLIS][2],
212+
"volume_id": "[UUID]",
212213
"volume_type": "MANAGED"
213214
},
214215
"changes": {

acceptance/bundle/resources/volumes/change-comment/output.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -50,6 +50,7 @@ Deployment complete!
5050
"schema_name":"myschema",
5151
"storage_location":"s3://deco-uc-prod-isolated-aws-us-east-1/metastore/[UUID]/volumes/[UUID]",
5252
"updated_at":[UNIX_TIME_MILLIS][0],
53+
"volume_id":"[UUID]",
5354
"volume_type":"MANAGED"
5455
}
5556

acceptance/bundle/resources/volumes/change-name/out.plan.direct.json

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,7 @@
2626
"schema_name": "myschema",
2727
"storage_location": "s3://deco-uc-prod-isolated-aws-us-east-1/metastore/[UUID]/volumes/[UUID]",
2828
"updated_at": [UNIX_TIME_MILLIS][0],
29+
"volume_id": "[UUID]",
2930
"volume_type": "MANAGED"
3031
},
3132
"changes": {

acceptance/bundle/resources/volumes/change-name/output.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -70,6 +70,7 @@ Deployment complete!
7070
"storage_location":"s3://deco-uc-prod-isolated-aws-us-east-1/metastore/[UUID]/volumes/[UUID]",
7171
"updated_at":[UNIX_TIME_MILLIS][1],
7272
"updated_by":"[USERNAME]",
73+
"volume_id":"[UUID]",
7374
"volume_type":"MANAGED"
7475
}
7576

acceptance/bundle/resources/volumes/remote-change-name/output.txt

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,7 @@ Deployment complete!
1717
"storage_location":"s3://deco-uc-prod-isolated-aws-us-east-1/metastore/[UUID]/volumes/[UUID]",
1818
"updated_at":[UNIX_TIME_MILLIS][1],
1919
"updated_by":"[USERNAME]",
20+
"volume_id":"[UUID]",
2021
"volume_type":"MANAGED"
2122
}
2223

@@ -32,6 +33,7 @@ Deployment complete!
3233
"storage_location":"s3://deco-uc-prod-isolated-aws-us-east-1/metastore/[UUID]/volumes/[UUID]",
3334
"updated_at":[UNIX_TIME_MILLIS][1],
3435
"updated_by":"[USERNAME]",
36+
"volume_id":"[UUID]",
3537
"volume_type":"MANAGED"
3638
}
3739

libs/testserver/handlers.go

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -388,6 +388,14 @@ func AddDefaultHandlers(server *Server) {
388388
return req.Workspace.VolumesCreate(req)
389389
})
390390

391+
server.Handle("PATCH", "/api/2.1/unity-catalog/volumes/{full_name}", func(req Request) any {
392+
return req.Workspace.VolumesUpdate(req, req.Vars["full_name"])
393+
})
394+
395+
server.Handle("DELETE", "/api/2.1/unity-catalog/volumes/{full_name}", func(req Request) any {
396+
return MapDelete(req.Workspace, req.Workspace.Volumes, req.Vars["full_name"])
397+
})
398+
391399
// Repos:
392400

393401
server.Handle("POST", "/api/2.0/repos", func(req Request) any {
@@ -406,14 +414,6 @@ func AddDefaultHandlers(server *Server) {
406414
return req.Workspace.ReposDelete(req)
407415
})
408416

409-
server.Handle("PATCH", "/api/2.1/unity-catalog/volumes/{full_name}", func(req Request) any {
410-
return req.Workspace.VolumesUpdate(req, req.Vars["full_name"])
411-
})
412-
413-
server.Handle("DELETE", "/api/2.1/unity-catalog/volumes/{full_name}", func(req Request) any {
414-
return MapDelete(req.Workspace, req.Workspace.Volumes, req.Vars["full_name"])
415-
})
416-
417417
// SQL Warehouses:
418418

419419
server.Handle("GET", "/api/2.0/sql/warehouses/{warehouse_id}", func(req Request) any {

libs/testserver/volumes.go

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -29,8 +29,8 @@ func (s *FakeWorkspace) VolumesCreate(req Request) Response {
2929
},
3030
}
3131
}
32-
// QQQ first UUID should be constant per workspace?
33-
volume.StorageLocation = fmt.Sprintf("s3://deco-uc-prod-isolated-aws-us-east-1/metastore/%s/volumes/%s", nextUUID(), nextUUID())
32+
volume.VolumeId = nextUUID()
33+
volume.StorageLocation = fmt.Sprintf("s3://deco-uc-prod-isolated-aws-us-east-1/metastore/%s/volumes/%s", TestMetastore.MetastoreId, volume.VolumeId)
3434

3535
volume.CreatedAt = nowMilli()
3636
volume.UpdatedAt = volume.CreatedAt

0 commit comments

Comments
 (0)