diff --git a/cloud_provider/aws/aws_minion_manager_test.py b/cloud_provider/aws/aws_minion_manager_test.py index daf13b5..d775a61 100644 --- a/cloud_provider/aws/aws_minion_manager_test.py +++ b/cloud_provider/aws/aws_minion_manager_test.py @@ -355,7 +355,7 @@ def test_cordon(self, mock_get_name_for_instance, mock_check_call): mock_get_name_for_instance.return_value = "ip-of-fake-node-name" awsmm.cordon_node("ip-of-fake-node") mock_check_call.assert_called_with(['kubectl', 'drain', 'ip-of-fake-node-name', - '--ignore-daemonsets=true', '--delete-local-data=true']) + '--ignore-daemonsets=true', '--delete-local-data=true', '--force', '--grace-period=-1']) mock_check_call.side_effect = [Exception("Test"), True] awsmm.cordon_node("ip-of-fake-node")