"""How to test wrfeed:

WRFEED_SETTINGS=../wrfeed/test.cfg python3 -m unittest tests/test_wrfeed.py
"""
import unittest
import xml.etree.ElementTree as et
import sqlalchemy
from sqlalchemy.engine.url import URL, make_url
import wrfeed


class WrfeedTestBase(unittest.TestCase):
    def setUp(self):
        wrfeed.app.config['TESTING'] = True
        self.app = wrfeed.app.test_client()


class TestNoDbWrfeed(WrfeedTestBase):
    def test_root(self):
        result = self.app.get('/')
        self.assertTrue(result.data.startswith(b'<?xml version='))
        self.assertIn(b'page about feeds', result.data)


class TestDbWrfeed(WrfeedTestBase):
    def setUp(self):
        super().setUp()
        with wrfeed.app.app_context():
            # create database
            uri = make_url(wrfeed.app.config['DATABASE_URI'])
            self.uri = uri
            uri_nodb = URL(uri.drivername, host=uri.host, port=uri.port, username=uri.username, password=uri.password, query=uri.query)
            engine_nodb = sqlalchemy.create_engine(uri_nodb)
            engine_nodb.execute("DROP DATABASE IF EXISTS {}".format(uri.database))
            engine_nodb.execute("CREATE DATABASE {} CHARACTER SET = '{}'".format(uri.database, 'UTF8'))

            # connect to database
            self.engine = sqlalchemy.create_engine(uri)
            con = self.engine.connect()

            # fill database
            with open('tests/testdb.sql', 'r') as f:
                sql = f.read()
            con.execute(sql)

            # update dates
            seconds_diff, = con.execute("SELECT TIMESTAMPDIFF(SECOND, '2017-03-31 07:00', now())").fetchone()
            seconds_diff = int(seconds_diff)
            con.execute("UPDATE wrreport SET date_report = DATE(date_report + INTERVAL %s SECOND) WHERE time_report IS NULL", (seconds_diff,))
            con.execute("UPDATE wrreport SET date_report = DATE(TIMESTAMP(date_report, time_report) + INTERVAL %s SECOND) WHERE time_report IS NOT NULL", (seconds_diff,))
            con.execute("UPDATE wrreport SET time_report = TIME(TIMESTAMP(date_report, time_report) + INTERVAL %s SECOND) WHERE time_report IS NOT NULL", (seconds_diff,))
            con.execute("UPDATE wrreport SET date_entry = date_entry + INTERVAL %s SECOND", (seconds_diff,))
            con.execute("UPDATE wrreport SET date_invalid = date_invalid + INTERVAL %s SECOND", (seconds_diff,))
            con.execute("UPDATE wrreport SET delete_date = delete_date + INTERVAL %s SECOND", (seconds_diff,))


    def tearDown(self):
        with wrfeed.app.app_context():
            self.engine.execute("DROP DATABASE {}".format(self.uri.database))

    def test_alle(self):
        result = self.app.get('/berichte/alle')
        self.assertTrue(result.data.startswith(b'<?xml'))
        root = et.fromstring(result.data)
        self.assertIn('Rodelbahnberichte', root.find('{http://www.w3.org/2005/Atom}title').text)
        self.assertEqual(len(root.findall('{http://www.w3.org/2005/Atom}entry')), 7)

    def test_bahn(self):
        bahn_count = {
            'haunold': 1,
            'pleisenhütte_axamer_lizum': 2,
            'prantner_alm': 0,
            'birgitzer_alm_(vom_adelshof)': 1,
        }
        for bahn, count in bahn_count.items():
            with self.subTest(bahn=bahn):
                result = self.app.get('/berichte/bahn/{}'.format(bahn))
                self.assertTrue(result.data.startswith(b'<?xml'))
                root = et.fromstring(result.data)
                self.assertIn('Rodelbahnberichte', root.find('{http://www.w3.org/2005/Atom}title').text)
                self.assertEqual(len(root.findall('{http://www.w3.org/2005/Atom}entry')), count, '"{}" not found {} times'.format(bahn, count))

    def test_bahnen(self):
        bahnen_count = {
            '761': 2,
            '793': 0,
            '100': 0,
            '1417': 1,
            '5': 1,
            '228': 1,
            '793+100': 0,
            '761+1417+5+228': 5,
        }
        for bahnen, count in bahnen_count.items():
            with self.subTest(bahnen=bahnen):
                result = self.app.get('/berichte/bahnen/{}'.format(bahnen))
                self.assertTrue(result.data.startswith(b'<?xml'))
                root = et.fromstring(result.data)
                self.assertIn('Rodelbahnberichte', root.find('{http://www.w3.org/2005/Atom}title').text)
                self.assertEqual(len(root.findall('{http://www.w3.org/2005/Atom}entry')), count, 'bahnen "{}" not found {} times'.format(bahnen, count))

    def test_region(self):
        region_count = {
            'tirol': 6,
            'schweiz': 1,
            'innsbruck': 3,
        }
        for region, count in region_count.items():
            with self.subTest(region=region):
                result = self.app.get('/berichte/region/{}'.format(region))
                self.assertTrue(result.data.startswith(b'<?xml'))
                root = et.fromstring(result.data)
                self.assertIn('Rodelbahnberichte', root.find('{http://www.w3.org/2005/Atom}title').text)
                self.assertEqual(len(root.findall('{http://www.w3.org/2005/Atom}entry')), count, '"{}" not found {} times'.format(region, count))