#from mock import patch, Mock

from twisted.trial import unittest

from sqlalchemy import create_engine

import os

from iro import install

from iro import config

from iro.model.schema import Base, Offer

from iro.model.utils import WithSession

from iro.offer.provider import providers, Provider

from ..test_helpers.dbtestcase import md, SampleDatabase

from ..test_helpers.utils import DummyObserver

class TestInstallation(unittest.TestCase):

    '''test install script'''

    def setUp(self):

        md.setUp()

        if not hasattr(md,"db2"):

            md.db2=SampleDatabase("test2","test2",'%s/my.cnf'%md.tdir)

            md.dburl2='mysql://test2:test@localhost/test2?unix_socket=%s/socket'%md.tdir

            md.db2.create()

        self.log = DummyObserver()

        self.log.start()

        self.engine = None

    def tearDown(self):

        self.log.stop()

        if self.engine:

            Base.metadata.drop_all(self.engine)

            self.engine = None

        try:

            os.remove("iro.conf")

        except OSError as e:

            if e.errno != 2:

                raise

    def testCheckConfig(self):

        self.assertEqual(install.checkConfig(),False)

        with open("iro.conf",'w') as fp:

            fp.write("""[main]

dburl=foo""")

        self.assertEqual(install.checkConfig(),False)

        self.assertEqual(len(self.log.e),1)

        self.assertEqual(self.log.e[0]['message'], ("Error while processing config file: Option 'port' in section 'main' is missing.",))

        with open("iro.conf",'w') as fp:

            fp.write("""[main]

dburl=foo

port=123456

""")

        self.assertEqual(install.checkConfig(),True)

    def testCheckDatabase(self):

        config.main.dburl=md.dburl2

        self.assertTrue(install.checkDatabaseConnection())

        self.assertFalse(install.checkDatabase())

        self.engine = create_engine(md.dburl2)

        Base.metadata.create_all(self.engine)

        self.assertTrue(install.checkDatabase())

    def testCheckDatabaseConnection(self):

        config.main.dburl="mysql://t@localhost/test"

        self.assertFalse(install.checkDatabaseConnection())

        self.assertEqual(len(self.log.e),1)

        self.assertTrue(self.log.e[0]['message'][0].startswith("Error while trying to connect to database\n"))

        config.main.dburl="sqlite://"

        self.assertTrue(install.checkDatabaseConnection())

    def testCreateDatabase(self):

        config.main.dburl=md.dburl2

        self.assertTrue(install.checkDatabaseConnection())

        self.assertFalse(install.checkDatabase())

        install.createDatabase()

        self.assertTrue(install.checkDatabase())

    def testCreateSampleConfig(self):

        install.createSampleConfig()

        with open("iro.conf",'r') as fp:

            c = fp.read()

        with open(os.path.abspath("../iro.conf.inst"),"r") as fp:

            self.assertEqual(c,fp.read())

    def testNoOverwrite(self):

        with open("iro.conf","w") as fp:

            fp.write("muhaha")

        install.createSampleConfig()

        with open("iro.conf",'r') as fp:

            self.assertEqual(fp.read(),"muhaha")

        self.assertEqual(len(self.log.e),1)

        self.assertEqual(self.log.e[0]['message'], ("iro.conf exists and will not be overwritten.",))