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 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)
