class TestUUIDGen(unittest.TestCase, SimpleKVTest): def setUp(self): self.store = UUIDDecorator(DictStore()) def test_put_generates_uuid_form(self): key = self.store.put(None, 'some_data') self.assertRegexpMatches(key, UUID_REGEXP) def test_put_file_generates_uuid_form(self): key = self.store.put_file(None, open('/dev/null', 'rb')) self.assertRegexpMatches(key, UUID_REGEXP) tmpfile = tempfile.NamedTemporaryFile(delete=False) try: tmpfile.close() key2 = self.store.put_file(None, tmpfile.name) self.assertRegexpMatches(key2, UUID_REGEXP) finally: if os.path.exists(tmpfile.name): os.unlink(tmpfile.name) def test_put_generates_valid_uuid(self): key = self.store.put(None, 'some_data') uuid.UUID(hex=key) def test_put_file_generates_valid_uuid(self): key = self.store.put_file(None, open('/dev/null', 'rb')) uuid.UUID(hex=key) tmpfile = tempfile.NamedTemporaryFile(delete=False) try: tmpfile.close() key2 = self.store.put_file(None, tmpfile.name) uuid.UUID(hex=key2) finally: if os.path.exists(tmpfile.name): os.unlink(tmpfile.name)
def ustore(self, store): return UUIDDecorator(store)
def setUp(self): self.store = UUIDDecorator(DictStore())
def templated_uuidstore(self, store, idgen_template): return UUIDDecorator(store, idgen_template)
def setUp(self): self.tmpdir = tempfile.mkdtemp() self.store = UUIDDecorator(FilesystemStore(self.tmpdir))