4 from sqlalchemy.engine import create_engine
5 from sqlalchemy.orm import sessionmaker
9 class TestMwDb(unittest.TestCase):
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)
17 self.session = self.Session()
20 self.session.rollback()
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)
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)
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)
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)
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)