def test_emit_event():
count = 0
sub = Subscription(stream_id, stream_partition, 'api_key',
lambda: print('done'))
def test():
nonlocal count
count += 1
sub.on(EventConstant.SUBSCRIBED, test)
sub.set_state(EventConstant.SUBSCRIBED)
assert count == 1
def test_update_state():
sub = Subscription(stream_id, stream_partition, 'api_key',
lambda: print('done'))
sub.set_state(EventConstant.SUBSCRIBED)
assert sub.get_state() == EventConstant.SUBSCRIBED