diff --git a/isso/tests/test_migration.py b/isso/tests/test_migration.py index ef1af6b..cb4236a 100644 --- a/isso/tests/test_migration.py +++ b/isso/tests/test_migration.py @@ -1,5 +1,10 @@ # -*- encoding: utf-8 -*- +try: + import unittest2 as unittest +except ImportError: + import unittest + import tempfile from os.path import join, dirname @@ -9,16 +14,26 @@ from isso.db import SQLite3 from isso.migrate import Disqus -def test_disqus(): +class TestMigration(unittest.TestCase): - xml = join(dirname(__file__), "disqus.xml") - xxx = tempfile.NamedTemporaryFile() + def test_disqus(self): - db = SQLite3(xxx.name, Config.load(None)) - Disqus(db, xml).migrate() + xml = join(dirname(__file__), "disqus.xml") + xxx = tempfile.NamedTemporaryFile() - assert db.threads["/"]["title"] == "Hello, World!" - assert db.threads["/"]["id"] == 1 + db = SQLite3(xxx.name, Config.load(None)) + Disqus(db, xml).migrate() + + self.assertEqual(db.threads["/"]["title"], "Hello, World!") + self.assertEqual(db.threads["/"]["id"], 1) + + a = db.comments.get(1) + + self.assertEqual(a["author"], "peter") + self.assertEqual(a["email"], "foo@bar.com") + + b = db.comments.get(2) + self.assertEqual(b["parent"] ,a["id"]) a = db.comments.get(1)