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