Fix test_test_data_storage

2018-10-27 09:01:13 +02:00 · 2018-10-27 09:01:13 +02:00 · a600dd8e54
parent 42c5bee947
commit a600dd8e54
1 changed files with 4 additions and 1 deletions
--- a/tests/test_snapshot.py
+++ b/tests/test_snapshot.py
@ -326,7 +326,10 @@ def test_test_data_storage(user: UserClient):
    """Tests a Snapshot with EraseSectors."""
    s = file('erase-sectors-2-hdd.snapshot')
    snapshot, _ = user.post(res=Snapshot, data=s)
-    incidence_test = next(ev for ev in snapshot['events'] if ev['reallocatedSectorCount'] == 15)
+    incidence_test = next(
+        ev for ev in snapshot['events']
+        if ev.get('reallocatedSectorCount', None) == 15
+    )
    assert incidence_test['incidence']
    assert incidence_test['description'] == 'Warning: Drive failure expected soon.'