def test_delete_resources_poller_is_none(self, is_mock): now = datetime.now(tz=pytz.utc) client = ARMClient('subscription_id', 'client_id', 'secret', 'tenant') client.init_services() group = ResourceGroup('juju-bar-1') client.resource.resource_groups.list.return_value = [group] # The resource_groups's storage_account is 4 hours old. storage_account = StorageAccount('abcd-12', now - timedelta(hours=4)) client.storage.storage_accounts.list_by_resource_group.return_value = [ storage_account] count = delete_resources(client, 'juju-bar*', old_age=2, now=now) self.assertEqual(1, count) self.assertEqual(1, client.resource.resource_groups.delete.call_count)
def xxx_test_delete_resources_only_network(self, is_mock): now = datetime.now(tz=pytz.utc) client = ARMClient('subscription_id', 'client_id', 'secret', 'tenant') client.init_services() group = ResourceGroup('juju-bar-1') client.resource.resource_groups.list.return_value = [group] # The resource_group has a network, but nothing else network = Network('juju-bar-network-1') client.network.virtual_networks.list.return_value = [network] poller = FakePoller() client.resource.resource_groups.delete.return_value = poller count = delete_resources(client, 'juju-bar*', old_age=2, now=now) self.assertEqual(1, count) self.assertEqual(1, client.resource.resource_groups.delete.call_count)
def test_delete_resources_poller_is_none(self, is_mock): now = datetime.now(tz=pytz.utc) client = ARMClient('subscription_id', 'client_id', 'secret', 'tenant') client.init_services() group = ResourceGroup('juju-bar-1') client.resource.resource_groups.list.return_value = [group] # The resource_groups's storage_account is 4 hours old. storage_account = StorageAccount('abcd-12', now - timedelta(hours=4)) client.storage.storage_accounts.list_by_resource_group.return_value = [ storage_account ] count = delete_resources(client, 'juju-bar*', old_age=2, now=now) self.assertEqual(1, count) self.assertEqual(1, client.resource.resource_groups.delete.call_count)
def test_delete_resources_old_age_0(self, is_mock): now = datetime.now(tz=pytz.utc) client = ARMClient('subscription_id', 'client_id', 'secret', 'tenant') client.init_services() a_group = ResourceGroup('juju-bar-1') b_group = ResourceGroup('juju-foo-0') client.resource.resource_groups.list.return_value = [a_group, b_group] poller = FakePoller() client.resource.resource_groups.delete.return_value = poller # Delete resource groups that are 0 hours old. # All matched resource_groups are deleted count = delete_resources(client, 'juju-bar*', old_age=0, now=now) self.assertEqual(1, count) self.assertEqual(1, client.resource.resource_groups.delete.call_count) self.assertIs(True, poller.is_done)
def test_delete_resources_found_old(self, is_mock): now = datetime.now(tz=pytz.utc) client = ARMClient('subscription_id', 'client_id', 'secret', 'tenant') client.init_services() group = ResourceGroup('juju-bar-1') client.resource.resource_groups.list.return_value = [group] # The resource_groups's storage_account is 4 hours old. storage_account = StorageAccount('abcd-12', now - timedelta(hours=4)) client.storage.storage_accounts.list_by_resource_group.return_value = [ storage_account] poller = FakePoller() client.resource.resource_groups.delete.return_value = poller # Delete resource groups that are 2 hours old. count = delete_resources(client, 'juju-bar*', old_age=2, now=now) self.assertEqual(1, count) client.resource.resource_groups.delete.assert_called_once_with( 'juju-bar-1') self.assertIs(True, poller.is_done)
def test_delete_resources_found_old(self, is_mock): now = datetime.now(tz=pytz.utc) client = ARMClient('subscription_id', 'client_id', 'secret', 'tenant') client.init_services() group = ResourceGroup('juju-bar-1') client.resource.resource_groups.list.return_value = [group] # The resource_groups's storage_account is 4 hours old. storage_account = StorageAccount('abcd-12', now - timedelta(hours=4)) client.storage.storage_accounts.list_by_resource_group.return_value = [ storage_account ] poller = FakePoller() client.resource.resource_groups.delete.return_value = poller # Delete resource groups that are 2 hours old. count = delete_resources(client, 'juju-bar*', old_age=2, now=now) self.assertEqual(1, count) client.resource.resource_groups.delete.assert_called_once_with( 'juju-bar-1') self.assertIs(True, poller.is_done)