Begin tests

dramatiq/registry.py

@@ -2,19 +2,19 @@
 
 
 class Registry:
-    '''
+    """
     A Registry allows defining a collection of Actors not directly bound to a Broker.
 
     This allows your code to declar Actors before configuring a Broker.
-    '''
+    """
     def __init__(self):
         self.actors = {}
         self.broker = None
 
     def actor(self, fn=None, *, actor_class=Actor, actor_name=None, queue_name="default", priority=0, **options):
-        '''
+        """
         Mimics `actor.actor` decorator, but skips the actor options check, and passes `self` as broker.
-        '''
+        """
 
         def decorator(fn):
             if not _queue_name_re.fullmatch(queue_name):
@@ -41,9 +41,9 @@ def __getattr__(self, name):
         return getattr(self.broker, name)
 
     def declare_actor(self, actor):
-        '''
+        """
         Intercept when Actor class tries to register itself.
-        '''
+        """
         if self.broker:
             self.broker.declare_actor(actor)
         else:

tests/test_registry.py

@@ -0,0 +1,22 @@
+import dramatiq
+from dramatiq.registry import Registry
+
+
+def test_registry_can_be_declared():
+
+    Registry()
+
+
+def test_actor_can_be_declared_on_registry():
+    # When I have a Registry
+    reg = Registry()
+
+    # Given that I've decorated a function with @registry.actor
+    @reg.actor
+    def add(x, y):
+        return x + y
+
+    # I expect that function to become an instance of Actor
+    assert isinstance(add, dramatiq.Actor)
+    assert add.broker is reg
+    assert len(reg.actors) == 1