def test_subscribe_producer_composite_rejected(self):
setup = BoboSetup()
sub = StubSubscriberSetup()
setup.add_complex_event(event_def=BoboComplexEvent(
name=NAME_NFA_A, pattern=stub_pattern_1, action=NoAction()))
# this rejects all events passed to Producer
setup.config_producer(NoAction(bool_return=False))
setup.subscribe_producer(NAME_NFA_A, sub)
setup.configure()
producer = setup.get_producer()
producer.setup()
c_event = CompositeEvent(timestamp=EpochNSClock.generate_timestamp(),
name=NAME_NFA_A,
history=BoboHistory(),
data={})
producer.on_decider_complex_event(c_event)
producer.loop()
self.assertEqual(c_event, sub.rejected_producer_event[0])
def test_config_arguments(self):
setup = BoboSetup()
event_def = BoboComplexEvent(NAME_NFA_A, stub_pattern_1)
validator = StrDictValidator()
action_producer = RateLimitAction()
action_forwarder = NoAction()
null_data = BoboNullDataStatic(DATA_DICT_A)
setup.add_complex_event(event_def)
setup.config_receiver(validator)
setup.config_producer(action_producer)
setup.config_forwarder(action_forwarder)
setup.config_null_data(NULL_DATA_DELAY, null_data)
setup.config_distributed(EXCHANGE_NAME, USER_NAME, parameters)
setup.configure()
receiver = setup.get_receiver()
decider = setup.get_decider()
producer = setup.get_producer()
forwarder = setup.get_forwarder()
nullgen = setup.get_null_data_generator()
manager = setup.get_distributed()
self.assertEqual(validator, receiver.get_validator())
self.assertEqual(NAME_NFA_A, decider.get_all_handlers()[0].nfa.name)
self.assertEqual(action_producer, producer._action)
self.assertEqual(action_forwarder, forwarder._action)
self.assertEqual(null_data, nullgen.null_data)
self.assertEqual(receiver, nullgen.receiver)
self.assertEqual(manager.outgoing.decider, decider)
self.assertEqual(manager.outgoing.exchange_name, EXCHANGE_NAME)
self.assertTrue(manager.outgoing.user_id.find(USER_NAME) != -1)
self.assertEqual(manager.outgoing.parameters, parameters)
self.assertEqual(manager.incoming.decider, decider)
self.assertEqual(manager.incoming.exchange_name, EXCHANGE_NAME)
self.assertTrue(manager.incoming.user_id.find(USER_NAME) != -1)
self.assertEqual(manager.incoming.parameters, parameters)
