Fix tests

Netflix-Skunkworks · Oct 10, 2023 · c4076f3 · c4076f3
1 parent 80df04b
commit c4076f3
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/tests/netflix/test_evcache.py b/tests/netflix/test_evcache.py
@@ -297,7 +297,7 @@ def test_evcache_ondisk_disk_usage():
 
         assert total_ram > inmemory_qps.data_shape.estimated_state_size_gib.mid
 
-def test_evcache_ondisk_disk_usage():
+def test_evcache_ondisk_high_disk_usage():
     high_disk_usage_rps = CapacityDesires(
         service_tier=0,
         query_pattern=QueryPattern(
@@ -329,7 +329,8 @@ def test_evcache_ondisk_disk_usage():
     )
 
     for candidate in plan:
-        total_disk = candidate.candidate_clusters.zonal[0].instance.drive.size_gib * \
-                    candidate.candidate_clusters.zonal[0].count
+        if candidate.candidate_clusters.zonal[0].instance.drive is not None:
+            total_disk = candidate.candidate_clusters.zonal[0].instance.drive.size_gib * \
+                        candidate.candidate_clusters.zonal[0].count
 
-        assert total_disk > high_disk_usage_rps.data_shape.estimated_state_size_gib.mid
+            assert total_disk > high_disk_usage_rps.data_shape.estimated_state_size_gib.mid