Made tests in test_wrmwdb more powerful.
[philipp/winterrodeln/wrpylib.git] / tests / test_wrmwdb.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.wrmwdb
7
8
9 class TestWrMwDb(unittest.TestCase):
10     @classmethod
11     def setUpClass(cls):
12         cls.engine = create_engine('mysql://philipp@localhost:3306/philipp_winterrodeln_wiki?charset=utf8&use_unicode=1')
13         cls.Session = sessionmaker(bind=cls.engine)
14
15     def setUp(self):
16         self.metadata = sqlalchemy.MetaData()
17         self.session = self.Session()
18
19     def tearDown(self):
20         self.session.rollback()
21
22     def test_wrreport_table(self):
23         WrReport = wrpylib.wrmwdb.wrreport_table(self.metadata)
24         wrreport = self.session.query(WrReport).filter(WrReport.c.id==130).first()
25         self.assertEqual(wrreport.id, 130)
26         self.assertEqual(wrreport.page_title, 'Simmeringalm')
27
28     def test_wrsledruncache_table(self):
29         WrSledrunCache = wrpylib.wrmwdb.wrsledruncache_table(self.metadata)
30         wrsledruncache = self.session.query(WrSledrunCache).filter(WrSledrunCache.c.page_id==226).first()
31         self.assertEqual(wrsledruncache.page_id, 226)
32         self.assertEqual(wrsledruncache.page_title, 'Axamer_Lizum')
33
34     def test_wrinncache_table(self):
35         WrInnCache = wrpylib.wrmwdb.wrinncache_table(self.metadata)
36         wrinncache = self.session.query(WrInnCache).filter(WrInnCache.c.page_id==324).first()
37         self.assertEqual(wrinncache.page_id, 324)
38         self.assertEqual(wrinncache.page_title, 'Alpengasthaus_Gleinserhof')