from iro.model.schema import Offer

from iro.model.decorators import vRoute, vTyp

from iro.model.pool import data

from iro.error import ValidateException

from .dbtestcase import DBTestCase

class DummyPool():

    def run(self, f,*a,**k):

        return f(*a,**k)

class ModelVaidatorTest(DBTestCase):

    """tests for the model vaidators"""

    def setUp(self):

        DBTestCase.setUp(self)

        self.pool = data.pool

        data.pool = DummyPool()

    def tearDown(self):

        data.pool = self.pool

        self.pool = None

        DBTestCase.tearDown(self)

    def testTyp(self):

        with self.session() as session:

            session.add(Offer(name="t",provider="p",typ="type"))

        with self.session() as session:

            self.assertEqual(vTyp("type",None),"type")

            e = self.assertRaises(ValidateException,vTyp, "sss", None)

            self.assertEqual(str(e),'700:Typ sss is not valid.')

    def testRoute(self):

        with self.session() as session:

            session.add(Offer(name="t",provider="p",typ="type"))

        self.assertEqual(str(e),'700:Route s is not valid.')

    def testRouteAllow(self):

        with self.session() as session:

            session.add(Offer(name="t",provider="p",typ="type"))

        self.assertEqual(str(e),'700:foo must be a list of routes.')

        self.assertEqual(str(e),'700:foo must be a route - No list of routes.')