From ef36dcebcd0779622c3b6bbe28028954b2c27ac5 Mon Sep 17 00:00:00 2001 From: Curtis Maloney Date: Mon, 13 Jul 2020 17:31:48 +1000 Subject: [PATCH] Begin tests --- tests/test_registry.py | 22 ++++++++++++++++++++++ 1 file changed, 22 insertions(+) create mode 100644 tests/test_registry.py diff --git a/tests/test_registry.py b/tests/test_registry.py new file mode 100644 index 00000000..b6e6c05d --- /dev/null +++ b/tests/test_registry.py @@ -0,0 +1,22 @@ +import dramatiq +from dramatiq.registry import Registry + + +def test_registry_can_be_declared(): + + reg = 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