def test_worker_connection(self):
"""
Purpose: Test regular running of worker
Expectation: startup system, hit the reddit user and parse the data, send to mothership
:precondition: Mothership server running
:return:
"""
server = MothershipServer()
server.run()
try:
worker = BasicUserParseWorker(
"https://www.reddit.com/user/alecnin")
worker.send_to_mother()
# connect to mother, so should not raise exception, should run everything else
except ConnectionRefusedError:
self.fail('connection failure')
def test_worker_send_to_mothership(self):
worker = BasicUserParseWorker("https://www.reddit.com/user/Chrikelnel")
test_data = "test_data"
test_root = "root"
# Set up our own socket on the mothership port.
# Precondition: ip in settings.py for mothership is localhost
s = socket.socket()
addr = (settings.MOTHERSHIP['host'], settings.MOTHERSHIP['port'])
s.bind(addr)
s.listen(5)
worker.send_to_mother(test_data, test_root)
worker, address = s.accept()
data = worker.recv(1024)
data = json.loads(data.decode('ascii'))
self.assertEqual(data['data'], test_data)
self.assertEqual(data['root'], test_root)
worker.close()
s.close()