def test_all_actors_can_be_stopped_through_registry(a_actor_refs,
b_actor_refs):
assert len(ActorRegistry.get_all()) == 8
ActorRegistry.stop_all(block=True)
assert len(ActorRegistry.get_all()) == 0
def test_actor_may_be_registered_manually(actor_ref):
ActorRegistry.unregister(actor_ref)
assert actor_ref not in ActorRegistry.get_all()
ActorRegistry.register(actor_ref)
assert actor_ref in ActorRegistry.get_all()
def test_actor_may_be_registered_manually(actor_ref):
ActorRegistry.unregister(actor_ref)
assert actor_ref not in ActorRegistry.get_all()
ActorRegistry.register(actor_ref)
assert actor_ref in ActorRegistry.get_all()
def test_actor_may_be_unregistered_multiple_times_without_error(self):
ActorRegistry.unregister(self.ref)
self.assert_(self.ref not in ActorRegistry.get_all())
ActorRegistry.unregister(self.ref)
self.assert_(self.ref not in ActorRegistry.get_all())
ActorRegistry.register(self.ref)
self.assert_(self.ref in ActorRegistry.get_all())
def test_actor_may_be_unregistered_multiple_times_without_error(self):
ActorRegistry.unregister(self.ref)
self.assertTrue(self.ref not in ActorRegistry.get_all())
ActorRegistry.unregister(self.ref)
self.assertTrue(self.ref not in ActorRegistry.get_all())
ActorRegistry.register(self.ref)
self.assertTrue(self.ref in ActorRegistry.get_all())
def test_actor_may_be_unregistered_multiple_times_without_error(actor_ref):
ActorRegistry.unregister(actor_ref)
assert actor_ref not in ActorRegistry.get_all()
ActorRegistry.unregister(actor_ref)
assert actor_ref not in ActorRegistry.get_all()
ActorRegistry.register(actor_ref)
assert actor_ref in ActorRegistry.get_all()
def test_all_actors_are_stopped_on_base_exception(events, actor_ref):
assert len(ActorRegistry.get_all()) == 1
assert not events.on_stop_was_called.is_set()
actor_ref.tell({'command': 'raise base exception'})
events.on_stop_was_called.wait(5)
assert events.on_stop_was_called.is_set()
assert len(ActorRegistry.get_all()) == 0
events.on_stop_was_called.wait(5)
assert events.on_stop_was_called.is_set()
assert len(ActorRegistry.get_all()) == 0
def test_all_actors_are_stopped_on_base_exception(events, actor_ref):
assert len(ActorRegistry.get_all()) == 1
assert not events.on_stop_was_called.is_set()
actor_ref.tell({'command': 'raise base exception'})
events.on_stop_was_called.wait(5)
assert events.on_stop_was_called.is_set()
assert len(ActorRegistry.get_all()) == 0
events.on_stop_was_called.wait(5)
assert events.on_stop_was_called.is_set()
assert len(ActorRegistry.get_all()) == 0
def test_actor_is_stopped_when_unhandled_exceptions_are_raised(
actor_ref, events):
assert not events.on_failure_was_called.is_set()
actor_ref.tell({'command': 'raise exception'})
events.on_failure_was_called.wait(5)
assert events.on_failure_was_called.is_set()
assert len(ActorRegistry.get_all()) == 0
def test_actor_processes_all_messages_before_stop_on_self_stops_it(self):
self.actor_ref.ask({'command': 'message self then stop'})
self.greetings_was_received.wait(5)
self.assertTrue(self.greetings_was_received.is_set())
self.on_stop_was_called.wait(5)
self.assertEqual(0, len(ActorRegistry.get_all()))
def test_actor_processes_all_messages_before_stop_on_self_stops_it(self):
self.actor_ref.ask({'command': 'message self then stop'})
self.greetings_was_received.wait(5)
self.assertTrue(self.greetings_was_received.is_set())
self.on_stop_was_called.wait(5)
self.assertEqual(0, len(ActorRegistry.get_all()))
def test_actor_processes_all_messages_before_stop_on_self_stops_it(
actor_ref, events):
actor_ref.ask({'command': 'message self then stop'})
events.greetings_was_received.wait(5)
assert events.greetings_was_received.is_set()
events.on_stop_was_called.wait(5)
assert len(ActorRegistry.get_all()) == 0
def test_actor_processes_all_messages_before_stop_on_self_stops_it(
actor_ref, events
):
actor_ref.ask({'command': 'message self then stop'})
events.greetings_was_received.wait(5)
assert events.greetings_was_received.is_set()
events.on_stop_was_called.wait(5)
assert len(ActorRegistry.get_all()) == 0
def test_actor_is_stopped_when_unhandled_exceptions_are_raised(
actor_ref, events
):
assert not events.on_failure_was_called.is_set()
actor_ref.tell({'command': 'raise exception'})
events.on_failure_was_called.wait(5)
assert events.on_failure_was_called.is_set()
assert len(ActorRegistry.get_all()) == 0
def test_broadcast_sends_message_to_all_actors_if_no_target(
a_actor_refs, b_actor_refs):
ActorRegistry.broadcast({'command': 'foo'})
running_actors = ActorRegistry.get_all()
assert running_actors
for actor_ref in running_actors:
received_messages = actor_ref.proxy().received_messages.get()
assert {'command': 'foo'} in received_messages
def test_broadcast_sends_message_to_all_actors_if_no_target(
a_actor_refs, b_actor_refs
):
ActorRegistry.broadcast({'command': 'foo'})
running_actors = ActorRegistry.get_all()
assert running_actors
for actor_ref in running_actors:
received_messages = actor_ref.proxy().received_messages.get()
assert {'command': 'foo'} in received_messages
def iap(pid):
g = Greeter.start()
g.actor_urn = pid
ActorRegistry.unregister(g)
ActorRegistry.register(g)
#greeter2 = Greeter.start(1).proxy()
#greeter3 = Greeter.start(2).proxy()
#greeter4 = Greeter.start(3).proxy()
#greeter5 = Greeter.start(4).proxy()
#a = ActorRegistry.get_all()
#for i in a:
# ppid = i.proxy().pid.get()
# print (ppid, type(ppid))
# if ppid == '123':
# print('haha, got you')
# else:
# print(ppid)
for i in ActorRegistry.get_all():
print ('here')
print(i.proxy().actor_urn.get(), type(i.proxy().actor_urn.get()))
if i.proxy().actor_urn.get() == pid:
print ('haha, got you')
print ('=============')
a = ActorRegistry.get_by_urn(pid)
print (a)
print ('=============')
for i in ActorRegistry.get_all():
print(i)
print (i.__dict__)
print ('here')
print(i.proxy().actor_urn.get(), type(i.proxy().actor_urn.get()))
if i.proxy().actor_urn.get() == 'asdf':
print ('haha, got you')
#future = greeter.pprint({'message':'%s' % pid})
return 'as'
def test_all_actors_are_stopped_on_base_exception(self):
start_event = self.event_class()
stop_event = self.event_class()
fail_event = self.event_class()
registered_event = self.event_class()
greetings_event = self.event_class()
self.AnActor.start(
on_start_was_called=start_event,
on_stop_was_called=stop_event,
on_failure_was_called=fail_event,
actor_was_registered_before_on_start_was_called=registered_event,
greetings_was_received=greetings_event)
self.assertEqual(2, len(ActorRegistry.get_all()))
self.assertFalse(self.on_stop_was_called.is_set())
self.actor_ref.tell({'command': 'raise base exception'})
self.on_stop_was_called.wait(5)
self.assertTrue(self.on_stop_was_called.is_set())
self.assert_(1 >= len(ActorRegistry.get_all()))
stop_event.wait(5)
self.assertTrue(stop_event.is_set())
self.assertEqual(0, len(ActorRegistry.get_all()))
def test_all_actors_are_stopped_on_base_exception(self):
start_event = self.event_class()
stop_event = self.event_class()
fail_event = self.event_class()
registered_event = self.event_class()
greetings_event = self.event_class()
self.AnActor.start(
on_start_was_called=start_event,
on_stop_was_called=stop_event,
on_failure_was_called=fail_event,
actor_was_registered_before_on_start_was_called=registered_event,
greetings_was_received=greetings_event)
self.assertEqual(2, len(ActorRegistry.get_all()))
self.assertFalse(self.on_stop_was_called.is_set())
self.actor_ref.tell({'command': 'raise base exception'})
self.on_stop_was_called.wait(5)
self.assertTrue(self.on_stop_was_called.is_set())
self.assert_(1 >= len(ActorRegistry.get_all()))
stop_event.wait(5)
self.assertTrue(stop_event.is_set())
self.assertEqual(0, len(ActorRegistry.get_all()))
def main():
parser = argparse.ArgumentParser()
parser.add_argument('inputFiles',
nargs='+',
help='One or more reparsed log files [reParseLog.py]')
args = parser.parse_args()
flushCounter = 0
distributorListener = startSystem()
for inputFile in args.inputFiles:
with open(inputFile, 'r') as data:
print('Playing:', inputFile, 'flushCounter, distributorListener =',
flushCounter, distributorListener)
jsonData = json.load(data)
flushCounter, distributorListener = procesTweetStream(
jsonData, flushCounter, distributorListener)
distributorListener.flush().get()
distributorListener.stop()
for i in ActorRegistry.get_all():
i.stop()
def test_actor_is_unregistered_when_stopped(actor_ref):
assert actor_ref in ActorRegistry.get_all()
actor_ref.stop()
assert actor_ref not in ActorRegistry.get_all()
def test_actor_is_unregistered_when_stopped(self):
self.assert_(self.ref in ActorRegistry.get_all())
self.ref.stop()
self.assert_(self.ref not in ActorRegistry.get_all())
def test_actor_may_be_registered_manually(self):
ActorRegistry.unregister(self.ref)
self.assert_(self.ref not in ActorRegistry.get_all())
ActorRegistry.register(self.ref)
self.assert_(self.ref in ActorRegistry.get_all())
def test_all_actors_can_be_stopped_through_registry(a_actor_refs, b_actor_refs):
assert len(ActorRegistry.get_all()) == 8
ActorRegistry.stop_all(block=True)
assert len(ActorRegistry.get_all()) == 0
def test_actor_is_registered_when_started(self):
self.assert_(self.ref in ActorRegistry.get_all())
def test_actor_is_unregistered_when_stopped(actor_ref):
assert actor_ref in ActorRegistry.get_all()
actor_ref.stop()
assert actor_ref not in ActorRegistry.get_all()
def test_actor_start_already_registered(actor_class):
actor_ref = actor_class().start()
assert len(ActorRegistry.get_all()) == 1
assert actor_ref in ActorRegistry.get_all()
ActorRegistry.stop_all()
def test_actor_may_be_registered_manually(self):
ActorRegistry.unregister(self.ref)
self.assertTrue(self.ref not in ActorRegistry.get_all())
ActorRegistry.register(self.ref)
self.assertTrue(self.ref in ActorRegistry.get_all())
def test_actor_is_registered_when_started(actor_ref):
assert actor_ref in ActorRegistry.get_all()
def test_all_actors_can_be_stopped_through_registry(self):
self.assertEquals(9, len(ActorRegistry.get_all()))
ActorRegistry.stop_all(block=True)
self.assertEquals(0, len(ActorRegistry.get_all()))
def test_broadcast_sends_message_to_all_actors_if_no_target(self):
ActorRegistry.broadcast({'command': 'foo'})
for actor_ref in ActorRegistry.get_all():
received_messages = actor_ref.proxy().received_messages.get()
self.assertTrue({'command': 'foo'} in received_messages)
def test_actor_is_stopped_when_unhandled_exceptions_are_raised(self):
self.assertFalse(self.on_failure_was_called.is_set())
self.actor_ref.tell({'command': 'raise exception'})
self.on_failure_was_called.wait(5)
self.assertTrue(self.on_failure_was_called.is_set())
self.assertEqual(0, len(ActorRegistry.get_all()))
def test_actor_is_unregistered_when_stopped(self):
self.assertTrue(self.ref in ActorRegistry.get_all())
self.ref.stop()
self.assertTrue(self.ref not in ActorRegistry.get_all())
def test_actor_is_registered_when_started(self):
self.assertTrue(self.ref in ActorRegistry.get_all())
def test_actor_is_registered_when_started(actor_ref):
assert actor_ref in ActorRegistry.get_all()
def test_actor_is_stopped_when_unhandled_exceptions_are_raised(self):
self.assertFalse(self.on_failure_was_called.is_set())
self.actor_ref.tell({'command': 'raise exception'})
self.on_failure_was_called.wait(5)
self.assertTrue(self.on_failure_was_called.is_set())
self.assertEqual(0, len(ActorRegistry.get_all()))
def test_broadcast_sends_message_to_all_actors_if_no_target(self):
ActorRegistry.broadcast({'command': 'foo'})
for actor_ref in ActorRegistry.get_all():
received_messages = actor_ref.proxy().received_messages.get()
self.assert_({'command': 'foo'} in received_messages)
def test_all_actors_can_be_stopped_through_registry(self):
self.assertEqual(9, len(ActorRegistry.get_all()))
ActorRegistry.stop_all(block=True)
self.assertEqual(0, len(ActorRegistry.get_all()))
