add osusers.tests.test_tasks.LdapRouterTest

gnuviechadmin/osusers/tests/test_tasks.py

@@ -0,0 +1,22 @@
+from django.test import TestCase
+
+from osusers.tasks import LdapRouter
+
+
+class LdapRouterTest(TestCase):
+    def setUp(self):
+        self.router = LdapRouter()
+        super(LdapRouterTest, self).setUp()
+
+    def test_ldap_tasks_are_routed_to_ldap_queue(self):
+        route = self.router.route_for_task(
+            'some_ldap_task')
+        self.assertEqual(
+            route,
+            {'exchange': 'ldap',
+             'exchange_type': 'direct',
+             'queue': 'ldap'})
+
+    def test_non_ldap_tasks_are_routed_to_default(self):
+        self.assertIsNone(
+            self.router.route_for_task('other'))