tests/test_mwdb.py

   1 #!/usr/bin/python3.4
   2 import unittest
   3 import sqlalchemy
   4 from sqlalchemy.engine import create_engine
   5 from sqlalchemy.orm import sessionmaker
   6 import wrpylib.mwdb
   7
   8
   9 class TestMwDb(unittest.TestCase):
  10     @classmethod
  11     def setUpClass(cls):
  12         cls.metadata = sqlalchemy.MetaData()
  13         cls.engine = create_engine('mysql://philipp@localhost:3306/philipp_winterrodeln_wiki?charset=utf8&use_unicode=1')
  14         cls.Session = sessionmaker(bind=cls.engine)
  15
  16     def setUp(self):
  17         self.session = self.Session()
  18
  19     def tearDown(self):
  20         self.session.rollback()
  21
  22     def test_page_table(self):
  23         Page = wrpylib.mwdb.page_table(self.metadata)
  24         page = self.session.query(Page).filter(Page.c.page_id==1321).first()
  25         self.assertEqual(page.page_id, 1321)
  26
  27     def test_revision_table(self):
  28         metadata = sqlalchemy.MetaData()
  29         RevisionTable = wrpylib.mwdb.revision_table(metadata)
  30         revision_table = self.session.query(RevisionTable).filter(RevisionTable.c.rev_id == 666).first()
  31         self.assertEqual(revision_table.rev_id, 666)
  32
  33     def test_text_table(self):
  34         metadata = sqlalchemy.MetaData()
  35         TextTable = wrpylib.mwdb.text_table(metadata)
  36         text_table = self.session.query(TextTable).filter(TextTable.c.old_id == 51).first()
  37         self.assertEqual(text_table.old_id, 51)
  38
  39     def test_user_table(self):
  40         metadata = sqlalchemy.MetaData()
  41         UserTable = wrpylib.mwdb.user_table(metadata)
  42         user_table = self.session.query(UserTable).filter(UserTable.c.user_id == 5).first()
  43         self.assertEqual(user_table.user_id, 5)
  44
  45     def test_categorylinks_table(self):
  46         metadata = sqlalchemy.MetaData()
  47         CategorylinksTable = wrpylib.mwdb.categorylinks_table(metadata)
  48         categorylinks_table = self.session.query(CategorylinksTable).filter(CategorylinksTable.c.cl_from == 229).first()
  49         self.assertEqual(categorylinks_table.cl_from, 229)