def test_dbi(self): self.var = 0 for f in self.dbi.files: self.dbi.files[f] = 'UV_POT' # Jon : HARDWF 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, port=TEST_PORT) 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', []), ('127.0.0.1', TEST_PORT)) # Jon : HARDWF while self.var != 1: time.sleep(.6) self.assertEqual(self.var, 1) self.assertEqual(self.dbi.get_obs_status(1), 'UV') # Jon : HARDWF finally: s.shutdown() thd.join()
def test_to_from_pkt(self): pkt = ts.to_pkt('UV', 5, 'still', ['1', '2', '3']) # Jon : HARDWF task, obs, still, args = ts.from_pkt(pkt) self.assertEqual(task, 'UV') # Jon : HARDWF self.assertEqual(obs, 5) self.assertEqual(still, 'still') self.assertEqual(args, ['1', '2', '3'])
def test_to_pkt(self): pkt = ts.to_pkt('UV', 5, 'still', ['1', '2', '3']) # Jon : HARDWF self.assertEqual(len(pkt), 7 * ts.PKT_LINE_LEN) self.assertEqual(pkt[:ts.PKT_LINE_LEN], ts.pad('7'))