#!/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.metadata = sqlalchemy.MetaData() 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.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): metadata = sqlalchemy.MetaData() RevisionTable = wrpylib.mwdb.revision_table(metadata) revision_table = self.session.query(RevisionTable).filter(RevisionTable.c.rev_id == 666).first() self.assertEqual(revision_table.rev_id, 666) def test_text_table(self): metadata = sqlalchemy.MetaData() TextTable = wrpylib.mwdb.text_table(metadata) text_table = self.session.query(TextTable).filter(TextTable.c.old_id == 51).first() self.assertEqual(text_table.old_id, 51) def test_user_table(self): metadata = sqlalchemy.MetaData() UserTable = wrpylib.mwdb.user_table(metadata) user_table = self.session.query(UserTable).filter(UserTable.c.user_id == 5).first() self.assertEqual(user_table.user_id, 5) def test_categorylinks_table(self): metadata = sqlalchemy.MetaData() CategorylinksTable = wrpylib.mwdb.categorylinks_table(metadata) categorylinks_table = self.session.query(CategorylinksTable).filter(CategorylinksTable.c.cl_from == 229).first() self.assertEqual(categorylinks_table.cl_from, 229)