diff -r 448dd8d36839 -r 3929338fd17f iro/tests/model_validate.py --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/iro/tests/model_validate.py Sun Mar 18 14:06:27 2012 +0100 @@ -0,0 +1,57 @@ +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") +