def test_cloud_cleaner_server(self):
config = CloudCleanerConfig(args=[])
ALL_RESOURCES["server"].process = Mock()
ALL_RESOURCES["server"].clean = Mock()
cloud_clean(args=[
"--os-auth-url", "http://no.com", "server", "--name", "derp"
],
config=config)
self.assertEqual("server", config.get_resource())
self.assertEqual("derp", config.get_arg("name"))
self.assertEqual(1, len(ALL_RESOURCES["server"].process.mock_calls))
self.assertEqual(0, len(ALL_RESOURCES["server"].clean.mock_calls))
def test_cloud_cleaner_noopts(self):
parser = ArgumentParser()
config_module.DEFAULT_ARGUMENTS = []
config = CloudCleanerConfig(parser=parser)
if version_info.major == 3:
# Raised because no resource
with self.assertRaises(MissingRequiredOptions):
cloud_clean([], config)
else:
with self.assertRaises(SystemExit):
cloud_clean([], config)
self.assertIsNone(config.get_resource())
def test_resource_type(self):
ALL_RESOURCES["server"].process = Mock()
ALL_RESOURCES["server"].clean = Mock()
cloud_clean(args=["--os-auth-url", "http://no.com", "-f", "server"])
self.assertEqual(1, len(ALL_RESOURCES["server"].process.mock_calls))
self.assertEqual(1, len(ALL_RESOURCES["server"].clean.mock_calls))