def test_purge_ebs_volumes(self):
conn = Mock()
conn.delete_volume = Mock()
with patch('purge_ebs_volumes.list_ebs_volumes', return_value=self.volume_list):
purge_ebs_volumes(conn, 0, '', False)
self.assertTrue(conn.delete_volume.called)
self.assertEqual(len(self.volume_list), conn.delete_volume.call_count)
def test_purge_ebs_volumes_noop(self):
conn = Mock()
conn.delete_volume = Mock()
with patch('purge_ebs_volumes.list_ebs_volumes',
return_value=self.volume_list):
purge_ebs_volumes(conn, 0, '', True)
self.assertFalse(conn.delete_volume.called)
def test_purge_ebs_volumes(self):
conn = Mock()
conn.delete_volume = Mock()
with patch('purge_ebs_volumes.list_ebs_volumes',
return_value=self.volume_list):
purge_ebs_volumes(conn, 0, '', False)
self.assertTrue(conn.delete_volume.called)
self.assertEqual(len(self.volume_list), conn.delete_volume.call_count)
def test_purge_ebs_volumes_noop(self):
conn = Mock()
conn.delete_volume = Mock()
with patch('purge_ebs_volumes.list_ebs_volumes', return_value=self.volume_list):
purge_ebs_volumes(conn, 0, '', True)
self.assertFalse(conn.delete_volume.called)
