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_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'))
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'))
