def test_offers(self): context = Context(['answer'], FAIL) # Test cases where it should fail. self.assertRaises(ValueError, context.register_handler, 'foo', lambda x: 'bar') self.assertRaises(ValueError, context.register_nostate_handler, 'foo', lambda: 'bar') self.assertRaises(ValueError, context.emmit_action, 'bar') # Test case where it shouldn't fail. context.register_handler('answer', lambda x: 42) self.assertEqual(context.emmit_action('answer'), [42])
def test_context(self): register_handler('bar', lambda x: 'bar') cont = Context() class Bar(ActionHandler): def __init__(self): ActionHandler.__init__(self, cont) @register_method('bar') def baar(self, state): return 42 bar = Bar() self.assertEqual(cont.emmit_action('bar'), [42]) self.assertEqual(emmit_action('bar'), ['bar'])