cleanup

Author: sanderegg

services/autoscaling/tests/unit/test_modules_auto_scaling_dynamic.py

@@ -1299,7 +1299,7 @@ async def test_cluster_adapts_machines_on_the_fly(  # noqa: PLR0915
 
     analyzed_cluster = assert_cluster_state(
         spied_cluster_analysis,
-        expected_calls=1,
+        expected_calls=3,
         expected_num_machines=app_settings.AUTOSCALING_EC2_INSTANCES.EC2_INSTANCES_MAX_INSTANCES,
     )
     assert analyzed_cluster.active_nodes
@@ -1391,6 +1391,11 @@ async def test_cluster_adapts_machines_on_the_fly(  # noqa: PLR0915
         await auto_scale_cluster(
             app=initialized_app, auto_scaling_mode=DynamicAutoscaling()
         )
+    analyzed_cluster = assert_cluster_state(
+        spied_cluster_analysis,
+        expected_calls=1,
+        expected_num_machines=app_settings.AUTOSCALING_EC2_INSTANCES.EC2_INSTANCES_MAX_INSTANCES,
+    )
     mock_docker_tag_node.assert_called_with(
         mock.ANY,
         analyzed_cluster.drained_nodes[-1].node,