trustchain-oc1-orchestral/devicehub-teal
Archived
10
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

fix test for the new api response 201

Browse source
This commit is contained in:
Cayo Puigdefabregas 2022-04-25 14:06:05 +02:00
parent f37e55e3b5
commit 515dad5850
1 changed files with 26 additions and 34 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

60
tests/test_snapshot.py
View file

@ -966,16 +966,17 @@ def test_snapshot_wb_lite(user: UserClient):
"2022-03-31_17h18m51s_ZQMPKKX51K67R68VO2X9RNZL08JPL_snapshot.json" "2022-03-31_17h18m51s_ZQMPKKX51K67R68VO2X9RNZL08JPL_snapshot.json"
) )
body, res = user.post(snapshot, uri="/api/inventory/") body, res = user.post(snapshot, uri="/api/inventory/")
# import pdb; pdb.set_trace()
ssd = [x for x in body['components'] if x['type'] == 'SolidStateDrive'][0] dev = m.Device.query.filter_by(devicehub_id=body['dhid']).one()
ssd = [x for x in dev.components if x.type == 'SolidStateDrive'][0]
assert body['device']['manufacturer'] == 'lenovo' assert dev.manufacturer == 'lenovo'
# assert body['sid'] == "LXVC" assert body['sid'] == "MLKO1Y0R55XZM051WQ5KJM01RY44Q"
assert ssd['serialNumber'] == 's35anx0j401001' assert ssd.serial_number == 's35anx0j401001'
assert res.status == '201 CREATED' assert res.status == '201 CREATED'
assert '00:28:f8:a6:d5:7e' in body['device']['hid'] assert '00:28:f8:a6:d5:7e' in dev.hid
dev = m.Device.query.filter_by(id=body['device']['id']).one()
assert dev.actions[0].power_on_hours == 6032 assert dev.actions[0].power_on_hours == 6032
errors = SnapshotErrors.query.filter().all() errors = SnapshotErrors.query.filter().all()
assert errors == [] assert errors == []
@ -994,7 +995,7 @@ def test_snapshot_wb_lite_qemu(user: UserClient):
assert body['sid'] == "VL0L5" assert body['sid'] == "VL0L5"
assert res.status == '201 CREATED' assert res.status == '201 CREATED'
dev = m.Device.query.filter_by(id=body['device']['id']).one() dev = m.Device.query.filter_by(devicehub_id=body['dhid']).one()
assert dev.manufacturer == 'qemu' assert dev.manufacturer == 'qemu'
assert dev.model == 'standard' assert dev.model == 'standard'
assert dev.serial_number is None assert dev.serial_number is None
@ -1004,7 +1005,6 @@ def test_snapshot_wb_lite_qemu(user: UserClient):
assert dev.components[-1].serial_number == 'qm00001' assert dev.components[-1].serial_number == 'qm00001'
@pytest.mark.mvp @pytest.mark.mvp
@pytest.mark.usefixtures(conftest.app_context.__name__) @pytest.mark.usefixtures(conftest.app_context.__name__)
def test_snapshot_wb_lite_old_snapshots(user: UserClient): def test_snapshot_wb_lite_old_snapshots(user: UserClient):
@ -1036,26 +1036,23 @@ def test_snapshot_wb_lite_old_snapshots(user: UserClient):
body11, res = user.post(snapshot_11, res=Snapshot) body11, res = user.post(snapshot_11, res=Snapshot)
bodyLite, res = user.post(snapshot_lite, uri="/api/inventory/") bodyLite, res = user.post(snapshot_lite, uri="/api/inventory/")
dev = m.Device.query.filter_by(devicehub_id=bodyLite['dhid']).one()
components11 = [] components11 = []
componentsLite = [] componentsLite = []
for c in body11.get('components', []): for c in body11.get('components', []):
if c['type'] in ["HardDrive", "SolidStateDrive"]: if c['type'] in ["HardDrive", "SolidStateDrive"]:
continue continue
components11.append({c.get('model'), c['type'], c.get('manufacturer')}) components11.append({c.get('model'), c['type'], c.get('manufacturer')})
for c in bodyLite.get('components', []): for c in dev.components:
componentsLite.append({c.get('model'), c['type'], c.get('manufacturer')}) componentsLite.append({c.model, c.type, c.manufacturer})
try: try:
assert body11['device'].get('hid') == bodyLite['device'].get('hid') assert body11['device'].get('hid') == dev.hid
if body11['device'].get('hid'): if body11['device'].get('hid'):
assert body11['device']['id'] == bodyLite['device']['id'] assert body11['device']['id'] == dev.id
assert body11['device'].get('serialNumber') == bodyLite['device'].get( assert body11['device'].get('serialNumber') == dev.serial_number
'serialNumber' assert body11['device'].get('model') == dev.model
) assert body11['device'].get('manufacturer') == dev.manufacturer
assert body11['device'].get('model') == bodyLite['device'].get('model')
assert body11['device'].get('manufacturer') == bodyLite['device'].get(
'manufacturer'
)
# wbLite can find more components than wb11 # wbLite can find more components than wb11
assert len(components11) <= len(componentsLite) assert len(components11) <= len(componentsLite)
@ -1094,25 +1091,22 @@ def test_snapshot_errors(user: UserClient):
body11, res = user.post(snapshot_11, res=Snapshot) body11, res = user.post(snapshot_11, res=Snapshot)
assert SnapshotErrors.query.all() == [] assert SnapshotErrors.query.all() == []
bodyLite, res = user.post(snapshot_lite, uri="/api/inventory/") bodyLite, res = user.post(snapshot_lite, uri="/api/inventory/")
dev = m.Device.query.filter_by(devicehub_id=bodyLite['dhid']).one()
assert len(SnapshotErrors.query.all()) == 2 assert len(SnapshotErrors.query.all()) == 2
assert body11['device'].get('hid') == bodyLite['device'].get('hid') assert body11['device'].get('hid') == dev.hid
assert body11['device']['id'] == bodyLite['device']['id'] assert body11['device']['id'] == dev.id
assert body11['device'].get('serialNumber') == bodyLite['device'].get( assert body11['device'].get('serialNumber') == dev.serial_number
'serialNumber' assert body11['device'].get('model') == dev.model
) assert body11['device'].get('manufacturer') == dev.manufacturer
assert body11['device'].get('model') == bodyLite['device'].get('model')
assert body11['device'].get('manufacturer') == bodyLite['device'].get(
'manufacturer'
)
components11 = [] components11 = []
componentsLite = [] componentsLite = []
for c in body11['components']: for c in body11['components']:
if c['type'] == "HardDrive": if c['type'] == "HardDrive":
continue continue
components11.append({c['model'], c['type'], c['manufacturer']}) components11.append({c['model'], c['type'], c['manufacturer']})
for c in bodyLite['components']: for c in dev.components:
componentsLite.append({c['model'], c['type'], c['manufacturer']}) componentsLite.append({c.model, c.type, c.manufacturer})
assert len(components11) == len(componentsLite) assert len(components11) == len(componentsLite)
for c in components11: for c in components11:
@ -1142,7 +1136,7 @@ def test_snapshot_errors_no_serial_number(user: UserClient):
bodyLite, res = user.post(snapshot_lite, uri="/api/inventory/") bodyLite, res = user.post(snapshot_lite, uri="/api/inventory/")
assert res.status_code == 201 assert res.status_code == 201
assert len(SnapshotErrors.query.all()) == 0 assert len(SnapshotErrors.query.all()) == 0
dev = m.Device.query.filter_by(id=bodyLite['device']['id']).one() dev = m.Device.query.filter_by(devicehub_id=bodyLite['dhid']).one()
assert not dev.model assert not dev.model
assert not dev.manufacturer assert not dev.manufacturer
assert not dev.serial_number assert not dev.serial_number
@ -1165,7 +1159,5 @@ def test_snapshot_check_tests_lite(user: UserClient):
bodyLite, res = user.post(snapshot_lite, uri="/api/inventory/") bodyLite, res = user.post(snapshot_lite, uri="/api/inventory/")
assert res.status_code == 201 assert res.status_code == 201
SnapshotErrors.query.all() SnapshotErrors.query.all()
dev = m.Device.query.filter_by(id=bodyLite['device']['id']).one() dev = m.Device.query.filter_by(devicehub_id=bodyLite['dhid']).one()
# import pdb; pdb.set_trace() # import pdb; pdb.set_trace()