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:')