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}
#!/usr/bin/python
import socket
from threading import Thread
from event_handler import EventHandler
from message_queue import MessageQueue
from user_network import UserNetwork
users = {}
queue = MessageQueue()
event_handler = EventHandler(users, UserNetwork())
def bind_server(port, queue_len=1):
serversocket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
serversocket.bind(('localhost', port))
serversocket.listen(queue_len)
return serversocket
def event_listener(port=9090):
serversocket = bind_server(port)
data = None
clientsocket = None
while True:
try:
if not data:
if clientsocket:
print(
"Event stream has stopped. Press Ctrl-C to stop the server"
)
clientsocket.close()
