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 ..test_helpers.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(vRoute("t",None,typ="type"),"t")

        self.assertEqual(vRoute(["t","t"],None,typ="type"),["t"])

        e = self.assertRaises(ValidateException,vRoute, "s", None, 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"))

        e = self.assertRaises(ValidateException,vRoute, "t", "foo", typ="type", allowString=False)

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

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

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

    def testRouteProvider(self):

        with self.session() as session:

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

        self.assertEqual(vRoute("p",None,typ="type"),"p")

    def testRouteDefault(self):

        self.assertEqual(vRoute("default",None,typ="type"),"default")