#!/usr/bin/python3.4 import unittest import sqlalchemy from sqlalchemy.engine import create_engine from sqlalchemy.orm import sessionmaker import wrpylib.wrmwdb class TestWrMwDb(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_wrreport_table(self): WrReport = wrpylib.wrmwdb.wrreport_table(self.metadata) wrreport = self.session.query(WrReport).filter(WrReport.c.id==130).first() self.assertEqual(wrreport.id, 130) self.assertEqual(wrreport.page_title, 'Simmeringalm') def test_wrsledruncache_table(self): WrSledrunCache = wrpylib.wrmwdb.wrsledruncache_table(self.metadata) wrsledruncache = self.session.query(WrSledrunCache).filter(WrSledrunCache.c.page_id==226).first() self.assertEqual(wrsledruncache.page_id, 226) self.assertEqual(wrsledruncache.page_title, 'Axamer_Lizum') def test_wrinncache_table(self): WrInnCache = wrpylib.wrmwdb.wrinncache_table(self.metadata) wrinncache = self.session.query(WrInnCache).filter(WrInnCache.c.page_id==324).first() self.assertEqual(wrinncache.page_id, 324) self.assertEqual(wrinncache.page_title, 'Alpengasthaus_Gleinserhof') def test_wrreportcache_table(self): WrReportCache = wrpylib.wrmwdb.wrreportcache_table(self.metadata) wrreport = self.session.query(WrReportCache).filter(WrReportCache.c.page_id==226).first() self.assertEqual(wrreport.page_title, 'Axamer Lizum') def test_wrregion_table(self): WrRegion = wrpylib.wrmwdb.wrregion_table(self.metadata) wrregion = self.session.query(WrRegion).filter(WrRegion.c.id==18).first() self.assertEqual(wrregion.id, 18) self.assertEqual(wrregion.name, 'Salzburg') def test_wrregioncache_table(self): WrRegionCache = wrpylib.wrmwdb.wrregioncache_table(self.metadata) wrregioncache = self.session.query(WrRegionCache).filter(WrRegionCache.c.id==43545).first() self.assertEqual(wrregioncache.id, 43545) self.assertEqual(wrregioncache.region_id, 27) self.assertEqual(wrregioncache.page_id, 54)