Mercurial > wikked
diff tests/test_db.py @ 225:ebb12ff21cb2
Updated unit tests to be able to run.
author | Ludovic Chabant <ludovic@chabant.com> |
---|---|
date | Wed, 12 Mar 2014 23:02:40 -0700 |
parents | 2733871775cd |
children |
line wrap: on
line diff
--- a/tests/test_db.py Mon Mar 10 16:47:21 2014 -0700 +++ b/tests/test_db.py Wed Mar 12 23:02:40 2014 -0700 @@ -1,47 +1,17 @@ -import os.path from tests import WikkedTest -from mock import MockFileSystem -from wikked.fs import FileSystem -from wikked.db import SQLiteDatabase class DatabaseTest(WikkedTest): - def tearDown(self): - if hasattr(self, 'wiki') and self.wiki: - self.wiki.db.close() - WikkedTest.tearDown(self) - def testEmpty(self): - self.wiki = self._getWikiFromStructure({}) - self.assertEqual([], list(self.wiki.getPageUrls())) + wiki = self._getWikiFromStructure({}) + self.assertEqual([], list(wiki.getPageUrls())) def testOnePage(self): - self.wiki = self._getWikiFromStructure({ - 'foo.txt': 'A test page.' + wiki = self._getWikiFromStructure({ + '/foo.txt': 'A test page.' }) - self.assertEqual(['foo'], list(self.wiki.getPageUrls())) - page = self.wiki.getPage('foo') - self.assertEqual('foo', page.url) - self.assertEqual(os.path.join(self.root, 'foo.txt'), page.path) + self.assertEqual(['/foo'], list(wiki.getPageUrls())) + page = wiki.getPage('/foo') + self.assertEqual('/foo', page.url) + self.assertEqual('/foo.txt', page.path) self.assertEqual('A test page.', page.raw_text) - - def _getWikiFromStructure(self, structure): - MockFileSystem.save_structure(self.root, structure) - wiki = self.getWiki( - db_factory=self._dbFactory, - fs_factory=self._fsFactory - ) - - # Open the DB before we do anything so that it will be closed - # only on `tearDown` (memory DBs are discarded when the - # connection is closed. - wiki.db.open() - - wiki.start() - return wiki - - def _fsFactory(self, config): - return FileSystem(self.root) - - def _dbFactory(self, config): - return SQLiteDatabase(':memory:')