#!/usr/bin/python3.4 import unittest import sqlalchemy from sqlalchemy.engine import create_engine from sqlalchemy.orm import sessionmaker import wrpylib.mwdb class TestMwDb(unittest.TestCase): @classmethod def setUpClass(cls): cls.engine = create_engine('mysql://philipp@localhost:3306/philipp_winterrodeln_wiki?charset=utf8&use_unicode=1') cls.Session = sessionmaker(bind=cls.engine) def setUp(self): self.metadata = sqlalchemy.MetaData() self.session = self.Session() def tearDown(self): self.session.rollback() def test_page_table(self): Page = wrpylib.mwdb.page_table(self.metadata) page = self.session.query(Page).filter(Page.c.page_id==1321).first() self.assertEqual(page.page_id, 1321) def test_revision_table(self): Revision = wrpylib.mwdb.revision_table(self.metadata) revision = self.session.query(Revision).filter(Revision.c.rev_id == 666).first() self.assertEqual(revision.rev_id, 666) def test_text_table(self): Text = wrpylib.mwdb.text_table(self.metadata) text = self.session.query(Text).filter(Text.c.old_id == 51).first() self.assertEqual(text.old_id, 51) def test_user_table(self): User = wrpylib.mwdb.user_table(self.metadata) user = self.session.query(User).filter(User.c.user_id == 5).first() self.assertEqual(user.user_id, 5) def test_categorylinks_table(self): Categorylinks = wrpylib.mwdb.categorylinks_table(self.metadata) categorylinks = self.session.query(Categorylinks).filter(Categorylinks.c.cl_from == 229).first() self.assertEqual(categorylinks.cl_from, 229)