def test_namehassuffix(self): odp = mock.MagicMock() odp.name = 'presentation.odp' notodp = mock.MagicMock() notodp.name = 'blargh' isodp = namehassuffix('.odp') self.assertTrue(isodp.apply(odp)) self.assertFalse(isodp.apply(notodp))
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)