Make delete a little bit handier.
Enable the ability to determine remote instances created by the user
without needing to specify instance names.
Bug: 117474343
Test: acloud delete
acloud delete --all
acloud delete --instance_names ...
atest acloud_test
Change-Id: Ia76fdb134092721b7d84a2f4c731f775d48a961c
diff --git a/internal/lib/utils_test.py b/internal/lib/utils_test.py
index aad23c1..9d62077 100644
--- a/internal/lib/utils_test.py
+++ b/internal/lib/utils_test.py
@@ -226,6 +226,24 @@
mock.call(16)
])
+ @mock.patch("__builtin__.raw_input")
+ def testGetAnswerFromList(self, mock_raw_input):
+ """Test GetAnswerFromList."""
+ answer_list = ["image1.zip", "image2.zip", "image3.zip"]
+ mock_raw_input.return_value = 0
+ with self.assertRaises(SystemExit):
+ utils.GetAnswerFromList(answer_list)
+ mock_raw_input.side_effect = [1, 2, 3, 1]
+ self.assertEqual(utils.GetAnswerFromList(answer_list),
+ ["image1.zip"])
+ self.assertEqual(utils.GetAnswerFromList(answer_list),
+ ["image2.zip"])
+ self.assertEqual(utils.GetAnswerFromList(answer_list),
+ ["image3.zip"])
+ self.assertEqual(utils.GetAnswerFromList(answer_list,
+ enable_choose_all=True),
+ answer_list)
+
if __name__ == "__main__":
unittest.main()