#!/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)