def testBasic(self):
match_sample_to_structure.match_sample_to_structure(self.details_file, self.structure_file, self.out_dir)
self.assertTrue(os.path.exists(self.out_file))
with open(self.out_file, 'r') as x:
results = x.read()
self.assertEqual(self.output, results)
reciever_ip = 'tcp://192.168.11.203:5559'
sinker_ip = 'tcp://192.168.11.203:5558'
context = zmq.Context()
# Get reciever
receiver = context.socket(zmq.PULL)
receiver.connect(reciever_ip)
sinker = context.socket(zmq.PUSH)
sinker.connect(sinker_ip)
while True:
data = receiver.recv_json()
sample_file = data['sample_file']
structure_file = data['structure_file']
out_dir = data['out_dir']
match_sample_to_structure.match_sample_to_structure(sample_file, structure_file, out_dir)
sys.stdout.write('working on %s\n' % sample_file)
sys.stdout.flush()
sinker.send_json({
'sender' : 'worker',
'body' : "%s\tcompleted" % structure_file
})
