class TestRepositoryBasic(unittest.TestCase): """ Very basic repository tests """ def _repositoryExists(self): try: self.rep.open() self.fail() except DBNoSuchFileError: pass except RepositoryOpenDeniedError: pass def setUp(self): self.rootdir = os.environ['CHANDLERHOME'] self.testdir = os.path.join(self.rootdir, 'repository', 'tests') self.rep = DBRepository(os.path.join(self.testdir, '__repository__')) def testNonExistent(self): """ The repository should not exist at this point """ self.assert_(not self._repositoryExists()) def testCreate(self): """ Create a repository and make sure it is open """ util.timing.reset() util.timing.begin("repository.TestRepositoryBasic.testCreate") self.rep.create() util.timing.end("repository.TestRepositoryBasic.testCreate") util.timing.results(verbose=False) self.assert_(self.rep.check()) self.assert_(self.rep.isOpen()) def testDestroy(self): """ Create and then delete a repository, verify it doesn't exist """ self.rep.create() self.rep.close() self.rep.delete() self.assert_(not self._repositoryExists()) def testLoadPack(self): """ Minimal test to ensure that we can load packs TODO is there more pack testing we need to do? """ self.rep.create() self.assert_(self.rep.check()) util.timing.reset() util.timing.begin("repository.TestRepositoryBasic.testLoadPack") self.rep.getCurrentView() util.timing.end("repository.TestRepositoryBasic.testLoadPack") util.timing.results(verbose=False) self.assert_(self.rep.check()) def tearDown(self): self.rep.close() self.rep.delete()
def initRepository(directory, options): repository = DBRepository(directory) kwds = { "stderr": options.stderr, "ramdb": options.ramdb, "create": True, "recover": options.recover, "exclusive": True, "refcounted": True, } if options.restore: kwds["restore"] = options.restore while True: try: if options.encrypt: from getpass import getpass kwds["password"] = getpass("password> ") if options.create: repository.create(**kwds) else: repository.open(**kwds) except RepositoryPasswordError, e: if options.encrypt: print e.args[0] else: options.encrypt = True continue except RepositoryVersionError: repository.close() raise
class RepositoryTestCase(TestCase): logLevel = logging.WARNING # a nice quiet default def _setup(self, ramdb=True): schema.reset() self.rootdir = os.environ['CHANDLERHOME'] self.chandlerPack = os.path.join(self.rootdir, 'repository', 'packs', 'chandler.pack') handler = \ logging.FileHandler(os.path.join(self.rootdir,'chandler.log')) formatter = \ logging.Formatter('%(asctime)s %(name)s %(levelname)s %(message)s') handler.setFormatter(formatter) root = logging.getLogger() root.addHandler(handler) self.ramdb = ramdb def _openRepository(self, ramdb=True): preloadedRepositoryPath = os.path.join(self.testdir, '__preloaded_repository__') self.rep = DBRepository(os.path.join(self.testdir, '__repository__')) if os.path.exists(preloadedRepositoryPath): self.ramdb = False self.rep.open(ramdb=False, restore=preloadedRepositoryPath, refcounted=True) self.rep.logger.setLevel(self.logLevel) self.rep.logger.info('Using preloaded repository') else: self.rep.create(ramdb=self.ramdb, refcounted=True) self.rep.logger.setLevel(self.logLevel) self.rep.loadPack(self.chandlerPack) self.rep.commit() self.manager = \ ParcelManager.get(self.rep.view, \ path=[os.path.join(self.rootdir, 'parcels')]) def setUp(self, ramdb=True): self._setup(ramdb) self.testdir = os.path.join(self.rootdir, 'repository', 'tests') self._openRepository(ramdb) def tearDown(self): self.rep.close() self.rep.logger.debug('RAMDB = %s', self.ramdb) if not self.ramdb: self.rep.delete() def _reopenRepository(self): self.rep.commit() if self.ramdb: self.rep.closeView() self.rep.openView() else: self.rep.close() self.rep = DBRepository(os.path.join(self.testdir, '__repository__')) self.rep.open() self.manager = \ ParcelManager.get(self.rep.view, \ path=[os.path.join(self.rootdir, 'parcels')]) def _find(self, path): return self.rep.findPath(path) def loadParcel(self, namespace): self.loadParcels([namespace]) def loadParcels(self, namespaces=None): self.manager.loadParcels(namespaces) if namespaces: for namespace in namespaces: parcelItem = self.manager.lookup(namespace) self.assert_(parcelItem) # print "Loaded namespace %s as item %s" % \ # (namespace, parcelItem.itsPath) _KIND_KIND = Path("//Schema/Core/Kind") _ITEM_KIND = Path("//Schema/Core/Item") # Repository specific assertions def assertIsRoot(self, item): self.assert_(item in list(self.rep.iterRoots())) def assertItemPathEqual(self, item, string): self.assertEqual(str(item.itsPath), string)
class RepositoryTestCase(TestCase): logLevel = logging.WARNING # a nice quiet default def _setup(self, ramdb=True): schema.reset() self.rootdir = os.environ['CHANDLERHOME'] self.chandlerPack = os.path.join(self.rootdir, 'repository', 'packs', 'chandler.pack') handler = \ logging.FileHandler(os.path.join(self.rootdir,'chandler.log')) formatter = \ logging.Formatter('%(asctime)s %(name)s %(levelname)s %(message)s') handler.setFormatter(formatter) root = logging.getLogger() root.addHandler(handler) self.ramdb = ramdb def _openRepository(self, ramdb=True): preloadedRepositoryPath = os.path.join(self.testdir, '__preloaded_repository__') self.rep = DBRepository(os.path.join(self.testdir, '__repository__')) if os.path.exists(preloadedRepositoryPath): self.ramdb = False self.rep.open(ramdb=False, restore=preloadedRepositoryPath, refcounted=True) self.rep.logger.setLevel(self.logLevel) self.rep.logger.info('Using preloaded repository') else: self.rep.create(ramdb=self.ramdb, refcounted=True) self.rep.logger.setLevel(self.logLevel) self.rep.loadPack(self.chandlerPack) self.rep.commit() self.manager = \ ParcelManager.get(self.rep.view, \ path=[os.path.join(self.rootdir, 'parcels')]) def setUp(self, ramdb=True): self._setup(ramdb) self.testdir = os.path.join(self.rootdir, 'repository', 'tests') self._openRepository(ramdb) def tearDown(self): self.rep.close() self.rep.logger.debug('RAMDB = %s', self.ramdb) if not self.ramdb: self.rep.delete() def _reopenRepository(self): self.rep.commit() if self.ramdb: self.rep.closeView() self.rep.openView() else: self.rep.close() self.rep = DBRepository( os.path.join(self.testdir, '__repository__')) self.rep.open() self.manager = \ ParcelManager.get(self.rep.view, \ path=[os.path.join(self.rootdir, 'parcels')]) def _find(self, path): return self.rep.findPath(path) def loadParcel(self, namespace): self.loadParcels([namespace]) def loadParcels(self, namespaces=None): self.manager.loadParcels(namespaces) if namespaces: for namespace in namespaces: parcelItem = self.manager.lookup(namespace) self.assert_(parcelItem) # print "Loaded namespace %s as item %s" % \ # (namespace, parcelItem.itsPath) _KIND_KIND = Path("//Schema/Core/Kind") _ITEM_KIND = Path("//Schema/Core/Item") # Repository specific assertions def assertIsRoot(self, item): self.assert_(item in list(self.rep.iterRoots())) def assertItemPathEqual(self, item, string): self.assertEqual(str(item.itsPath), string)