tests/db.py
branchdevel
changeset 205 02ae72e8bab9
parent 203 b543ab9a9724
child 208 5da0f00ef872
--- a/tests/db.py	Thu Mar 01 21:00:31 2012 +0100
+++ b/tests/db.py	Fri Mar 02 02:50:17 2012 +0100
@@ -84,12 +84,32 @@
 
         with self.session() as session:
             o=Offer(name="s2", provider="sipgate", route="basic", typ="sms")
+            session.add(o)
             o=Offer(name="s3", provider="sipgate", route="basic", typ="sms2")
             session.add(o)
         
         with self.session() as session:
             self.assertEqual(session.typs.order_by(Offer.typ).all(),[('sms',),('sms2',)])
 
+    def testOfferRoutes(self):
+        with self.session() as session:
+            o=Offer(name="sipgate_basic", provider="sipgate", route="basic", typ="sms")
+            session.add(o)
+        
+        with self.session() as session:
+            self.assertEqual([o.name for o in Offer.routes(session, "sms").order_by(Offer.name)],["sipgate_basic"])
+
+        with self.session() as session:
+            o=Offer(name="s2", provider="sipgate", route="basic", typ="sms")
+            session.add(o)
+            o=Offer(name="s3", provider="sipgate", route="basic", typ="sms2")
+            session.add(o)
+        
+        with self.session() as session:
+            self.assertEqual([o.name for o in Offer.routes(session, "sms2").order_by(Offer.name)],["s3"])
+            self.assertEqual([o.name for o in Offer.routes(session, "sms").order_by(Offer.name)],["s2","sipgate_basic"])
+
+
 class Has_RightTests(DBTestCase):
     '''test User.has_right'''
     def testSimple(self):