Merge pull request #125 from maurerle/raise_double_register

raise exception when trying to register a container which is already registered

mango/container/core.py

@@ -118,6 +118,8 @@ def register(self, agent: Agent, suggested_aid: str = None):
         :return The agent ID
         """
         aid = self._reserve_aid(suggested_aid)
+        if agent.context:
+            raise ValueError("Agent is already registered to a container")
         self._agents[aid] = agent
         agent._do_register(self, aid)
         logger.debug("Successfully registered agent;%s", aid)

tests/unit_tests/core/test_agent.py

@@ -102,6 +102,13 @@ async def test_schedule_acl_message():
     # THEN
     assert agent2.test_counter == 1
 
+def test_register_twice():
+    c = create_tcp_container(addr=("127.0.0.1", 5555))
+    agent = MyAgent()
+    c.register(agent)
+
+    with pytest.raises(ValueError):
+        c.register(agent)
 
 def test_sync_setup_agent():
     # this test is not async and therefore does not provide a running event loop