def event_handler(): user_network = UserNetwork() user_network.follow(b'1', b'2') user_network.follow(b'1', b'4') return EventHandler({ b'1': Mock(), b'2': Mock(), b'3': Mock() }, user_network)
def should_follow_unfollow(): network = UserNetwork() network.follow(1, 2) network.follow(1, 3) network.follow(2, 3) assert network[1] == {2, 3} assert network[3] == set() assert network[2] == {3} network.unfollow(1, 3) network.unfollow(1, 3) # no error assert network[1] == {2}
def event_handler(): user_network = UserNetwork() user_network.follow(b'1', b'2') user_network.follow(b'1', b'4') return EventHandler({b'1': Mock(), b'2': Mock(), b'3': Mock()}, user_network)