def test_namehasprefix(self): export = mock.MagicMock() export.name = 'export_frete' notexport = mock.MagicMock() notexport.name = 'blargh' isexport = namehasprefix('export_') self.assertTrue(isexport.apply(export)) self.assertFalse(isexport.apply(notexport))
def test_collect_applies_rule(self): mock_obj1 = mock.MagicMock() mock_obj1.name = 'pref_name_suf' mock_obj2 = mock.MagicMock() mock_obj2.name = 'noprefnosuf' mock_cont = mock.MagicMock() mock_cont.get_objects.return_value = [mock_obj1, mock_obj2] mock_conn = mock.MagicMock() mock_conn.get_all_containers.return_value = [mock_cont] mock_conn.get_container.return_value = mock_cont with mock.patch('cloudfiles.get_connection', return_value=mock_conn): deleted = collect(rule=namehasprefix('pref')&namehassuffix('suf') ,user=mock.ANY, key=mock.ANY) self.assertIn(mock.call.delete_object(mock_obj1.name), mock_cont.method_calls) self.assertNotIn(mock.call.delete_object(mock_obj2.name), mock_cont.method_calls) self.assertItemsEqual([mock_obj1.name], deleted)