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