StealthOS/subiquity
StealthOS
/
subiquity
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge pull request #1229 from dbungert/api-test-flow 

api test: fix flow test for v2 storage actions
This commit is contained in:
Dan Bungert 2022-03-16 15:41:43 -06:00 committed by GitHub
parent 0550626482 6a378de508
commit d7d1a70b59
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 20 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
subiquity/tests/api/test_api.py
View File

@ -250,8 +250,8 @@ class TestFlow(TestAPI):
resp = await inst.post('/storage/v2/reformat_disk',
disk_id=disk_id)
sda = first(resp['disks'], 'id', disk_id)
self.assertEqual(1, len(sda['partitions']))
self.assertEqual('Gap', sda['partitions'][0]['$type'])
[gap] = sda['partitions']
self.assertEqual('Gap', gap['$type'])
data = {
'disk_id': disk_id,
@ -272,22 +272,28 @@ class TestFlow(TestAPI):
'format': 'ext4',
}
}
resp = await inst.post('/storage/v2/edit_partition', data)
expected = add_resp['disks'][0]['partitions'][1]
actual = resp['disks'][0]['partitions'][1]
edit_resp = await inst.post('/storage/v2/edit_partition', data)
add_sda = first(add_resp['disks'], 'id', disk_id)
add_sda2 = first(add_sda['partitions'], 'number', 2)
edit_sda = first(edit_resp['disks'], 'id', disk_id)
edit_sda2 = first(edit_sda['partitions'], 'number', 2)
for key in 'size', 'number', 'mount', 'boot':
self.assertEqual(expected[key], actual[key])
self.assertEqual('ext4',
resp['disks'][0]['partitions'][1]['format'])
self.assertEqual(add_sda2[key], edit_sda2[key])
self.assertEqual('ext4', edit_sda2['format'])
resp = await inst.post('/storage/v2/delete_partition', data)
sda = first(resp['disks'], 'id', disk_id)
del_resp = await inst.post('/storage/v2/delete_partition', data)
sda = first(del_resp['disks'], 'id', disk_id)
self.assertEqual(2, len(sda['partitions']))
self.assertEqual('Partition', sda['partitions'][0]['$type'])
self.assertEqual('Gap', sda['partitions'][1]['$type'])
resp = await inst.post('/storage/v2/reset')
self.assertEqual(orig_resp, resp)
for type in 'Partition', 'Gap':
pgs = [pg for pg in sda['partitions'] if pg['$type'] == type]
self.assertEqual(len(pgs), 1)
reset_resp = await inst.post('/storage/v2/reset')
self.assertEqual(orig_resp, reset_resp)
choice = {'disk_id': disk_id}
guided_resp = await inst.post('/storage/v2/guided', choice)