Fix unit tests. (#38)

keikoproj · Apr 16, 2019 · e80c6ec · e80c6ec
1 parent 7217fd7
commit e80c6ec
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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")