Fix die bug

examples/random_delays/random_delays.py

@@ -4,20 +4,34 @@ Example of a fully programmatic simulation, without definition files.
 '''
 from soil import Simulation, agents
 from soil.time import Delta
-from networkx import Graph
 from random import expovariate
 import logging
 
 
 
 class MyAgent(agents.FSM):
+    '''
+    An agent that first does a ping
+    '''
+
+    defaults = {'pong_counts': 2}
 
     @agents.default_state
     @agents.state
-    def neutral(self):
-        self.info('I am running')
+    def ping(self):
+        self.info('Ping')
+        return self.pong, Delta(expovariate(1/16))
+
+    @agents.state
+    def pong(self):
+        self.info('Pong')
+        self.pong_counts -= 1
+        self.info(str(self.pong_counts))
+        if self.pong_counts < 1:
+            return self.die()
         return None, Delta(expovariate(1/16))
 
+
 s = Simulation(name='Programmatic',
                network_agents=[{'agent_type': MyAgent, 'id': 0}],
                topology={'nodes': [{'id': 0}], 'links': []},