#!/usr/bin/python3.4
import unittest
import sqlalchemy
from sqlalchemy.engine import create_engine
from sqlalchemy.orm import sessionmaker
import wrpylib.mwdb


class TestMwDb(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_page_table(self):
        Page = wrpylib.mwdb.page_table(self.metadata)
        page = self.session.query(Page).filter(Page.c.page_id==1321).first()
        self.assertEqual(page.page_id, 1321)

    def test_revision_table(self):
        Revision = wrpylib.mwdb.revision_table(self.metadata)
        revision = self.session.query(Revision).filter(Revision.c.rev_id == 666).first()
        self.assertEqual(revision.rev_id, 666)

    def test_text_table(self):
        Text = wrpylib.mwdb.text_table(self.metadata)
        text = self.session.query(Text).filter(Text.c.old_id == 51).first()
        self.assertEqual(text.old_id, 51)

    def test_user_table(self):
        User = wrpylib.mwdb.user_table(self.metadata)
        user = self.session.query(User).filter(User.c.user_id == 5).first()
        self.assertEqual(user.user_id, 5)

    def test_categorylinks_table(self):
        Categorylinks = wrpylib.mwdb.categorylinks_table(self.metadata)
        categorylinks = self.session.query(Categorylinks).filter(Categorylinks.c.cl_from == 229).first()
        self.assertEqual(categorylinks.cl_from, 229)