The order of XML attributes is now retained.
[philipp/winterrodeln/wrpylib.git] / tests / test_wrmwdb.py
1 import os
2 import unittest
3 import sqlalchemy
4 from sqlalchemy.engine import create_engine
5 from sqlalchemy.orm import sessionmaker
6 import wrpylib.wrmwdb
7 from wrpylib.wrmwcache import update_wrreportcache
8
9 TEST_DB = 'mysql://philipp@localhost:3306/philipp_winterrodeln_wrpylib_wrpylib_test?charset=utf8mb4'
10
11
12 def setup_test_database():
13     engine = create_engine(TEST_DB)
14     testdb = os.path.join(os.path.dirname(__file__), 'testdb.sql')
15     with open(testdb, 'r') as f:
16         sql = f.read()
17     with engine.begin() as con:
18         con.execution_options(no_parameters=True).execute(sql)
19
20         # update dates
21         seconds_diff, = con.execute("SELECT TIMESTAMPDIFF(SECOND, '2017-03-31 07:00', now())").fetchone()
22         seconds_diff = int(seconds_diff)
23         con.execute(
24             "UPDATE wrreport SET date_report = DATE(date_report + INTERVAL %s SECOND) WHERE time_report IS NULL",
25             (seconds_diff,))
26         con.execute(
27             "UPDATE wrreport SET date_report = DATE(TIMESTAMP(date_report, time_report) + INTERVAL %s SECOND) WHERE time_report IS NOT NULL",
28             (seconds_diff,))
29         con.execute(
30             "UPDATE wrreport SET time_report = TIME(TIMESTAMP(date_report, time_report) + INTERVAL %s SECOND) WHERE time_report IS NOT NULL",
31             (seconds_diff,))
32         con.execute("UPDATE wrreport SET date_entry = date_entry + INTERVAL %s SECOND", (seconds_diff,))
33         con.execute("UPDATE wrreport SET date_invalid = date_invalid + INTERVAL %s SECOND", (seconds_diff,))
34         con.execute("UPDATE wrreport SET delete_date = delete_date + INTERVAL %s SECOND", (seconds_diff,))
35     return engine
36
37
38 class TestWrMwDb(unittest.TestCase):
39     @classmethod
40     def setUpClass(cls):
41         cls.engine = setup_test_database()
42         cls.Session = sessionmaker(bind=cls.engine)
43
44     def setUp(self):
45         self.metadata = sqlalchemy.MetaData()
46         self.session = self.Session()
47
48     def tearDown(self):
49         self.session.rollback()
50
51     def test_wrreport_table(self):
52         WrReport = wrpylib.wrmwdb.wrreport_table(self.metadata)
53         wrreport = self.session.query(WrReport).filter(WrReport.c.id==19590).first()
54         self.assertEqual(wrreport.id, 19590)
55         self.assertEqual('Pleisenhütte Axamer Lizum', wrreport.page_title)
56
57     def test_wrsledruncache_table(self):
58         WrSledrunCache = wrpylib.wrmwdb.wrsledruncache_table(self.metadata)
59         wrsledruncache = self.session.query(WrSledrunCache).filter(WrSledrunCache.c.page_id==926).first()
60         self.assertEqual(926, wrsledruncache.page_id)
61         self.assertEqual('Hühnerspiel', wrsledruncache.page_title)
62
63     def test_wrinncache_table(self):
64         WrInnCache = wrpylib.wrmwdb.wrinncache_table(self.metadata)
65         wrinncache = self.session.query(WrInnCache).filter(WrInnCache.c.page_id==362).first()
66         self.assertEqual(362, wrinncache.page_id)
67         self.assertEqual('Meißner_Haus_(Gasthaus)', wrinncache.page_title)
68
69     def test_wrreportcache_table(self):
70         update_wrreportcache(self.engine.connect())
71         WrReportCache = wrpylib.wrmwdb.wrreportcache_table(self.metadata)
72         wrreport = self.session.query(WrReportCache).filter(WrReportCache.c.page_id == 237).first()
73         self.assertEqual(237, wrreport.page_id)
74         self.assertEqual('Gleirschalm', wrreport.page_title)
75
76     def test_wrregion_table(self):
77         WrRegion = wrpylib.wrmwdb.wrregion_table(self.metadata)
78         wrregion = self.session.query(WrRegion).filter(WrRegion.c.id == 18).first()
79         self.assertEqual(18, wrregion.id)
80         self.assertEqual('Salzburg', wrregion.name)
81
82     def test_wrregioncache_table(self):
83         WrRegionCache = wrpylib.wrmwdb.wrregioncache_table(self.metadata)
84         wrregioncache = self.session.query(WrRegionCache).filter((WrRegionCache.c.region_id == 27) &
85                                                                  (WrRegionCache.c.page_id == 5)).first()
86         self.assertEqual(27, wrregioncache.region_id)
87         self.assertEqual(5, wrregioncache.page_id)