def test___iter__no_suffix(self): my_store = TextStore(MemoryTransport(), prefixed=False, compressed=False) stream = StringIO("content") my_store.add(stream, "foo") self.assertEqual(set(['foo']), set(my_store.__iter__()))
def test_copy_suffixes(self): from_store = self.get_populated_store() to_store = TextStore(MemoryTransport(), prefixed=True, compressed=True) to_store.register_suffix("sig") to_store.copy_all_ids(from_store) self.assertEqual(1, len(to_store)) self.assertEqual(set(["foo"]), set(to_store.__iter__())) self.assertEqual("content", to_store.get("foo").read()) self.assertEqual("signature", to_store.get("foo", "sig").read()) self.assertRaises(KeyError, to_store.get, "missing", "sig")
def test_copy_suffixes(self): from_store = self.get_populated_store() to_store = TextStore(MemoryTransport(), prefixed=True, compressed=True) to_store.register_suffix('sig') to_store.copy_all_ids(from_store) self.assertEqual(1, len(to_store)) self.assertEqual(set(['foo']), set(to_store.__iter__())) self.assertEqual('content', to_store.get('foo').read()) self.assertEqual('signature', to_store.get('foo', 'sig').read()) self.assertRaises(KeyError, to_store.get, 'missing', 'sig')