Skip to content

Commit 756d29c

Browse files
committed
fixed test accordingly
1 parent f8a9594 commit 756d29c

File tree

1 file changed

+11
-1
lines changed

1 file changed

+11
-1
lines changed

services/autoscaling/tests/unit/test_modules_auto_scaling_dynamic.py

Lines changed: 11 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1769,7 +1769,17 @@ async def test__activate_drained_nodes_with_drained_node(
17691769
updated_cluster = await _activate_drained_nodes(
17701770
initialized_app, cluster_with_drained_nodes, DynamicAutoscaling()
17711771
)
1772-
assert updated_cluster.active_nodes == cluster_with_drained_nodes.drained_nodes
1772+
# they are the same nodes, but the availability might have changed here
1773+
assert updated_cluster.active_nodes != cluster_with_drained_nodes.drained_nodes
1774+
assert (
1775+
updated_cluster.active_nodes[0].assigned_tasks
1776+
== cluster_with_drained_nodes.drained_nodes[0].assigned_tasks
1777+
)
1778+
assert (
1779+
updated_cluster.active_nodes[0].ec2_instance
1780+
== cluster_with_drained_nodes.drained_nodes[0].ec2_instance
1781+
)
1782+
17731783
assert drained_host_node.spec
17741784
mock_docker_tag_node.assert_called_once_with(
17751785
mock.ANY,

0 commit comments

Comments
 (0)