from iro.model.schema import Offer

from iro.model.decorators import vRoute

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):

        pass

    testTyp.todo = "To implement"

    def testRoute(self):

        with self.session() as session:

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

        vR = vRoute('type')

        self.assertEqual(vR("t",None),"t")

        self.assertEqual(vR(["t","t"],None),["t"])

        e = self.assertRaises(ValidateException,vR, "s", None)

        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"))

        vR = vRoute('type')

        e = self.assertRaises(ValidateException,vR, "t", "foo", allowString=False)

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

        e = self.assertRaises(ValidateException,vR, ["t"], "foo", allowList=False)

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