def test_to_from_pkt(self):
pkt = ts.to_pkt('UV', 5, 'still', ['1','2','3'])
task, obs, still, args = ts.from_pkt(pkt)
self.assertEqual(task, 'UV')
self.assertEqual(obs, 5)
self.assertEqual(still, 'still')
self.assertEqual(args, ['1','2','3'])
def test_dbi(self):
self.var = 0
for f in self.dbi.files: self.dbi.files[f] = 'UV_POT'
class NullHandler(ts.TaskHandler):
def handle(me):
task, obs, still, args = me.get_pkt()
t = NullTask(task, obs, still, args, self.dbi)
me.server.append_task(t)
t.run()
self.var += 1
s = ts.TaskServer(self.dbi, handler=NullHandler)
thd = threading.Thread(target=s.start)
thd.start()
try:
sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
sock.sendto(ts.to_pkt('UV',1,'still',[]), ('localhost', ts.STILL_PORT))
while self.var != 1: time.sleep(.6)
self.assertEqual(self.var, 1)
self.assertEqual(self.dbi.get_obs_status(1), 'UV')
finally:
s.shutdown()
thd.join()
def test_to_pkt(self):
pkt = ts.to_pkt('UV', 5, 'still', ['1','2','3'])
self.assertEqual(len(pkt), 7*ts.PKT_LINE_LEN)
self.assertEqual(pkt[:ts.PKT_LINE_LEN], ts.pad('7'))
