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)