コード例 #1
0
ファイル: task_server_test.py プロジェクト: nkern/capo
 def test_shutdown(self):
     s = ts.TaskServer(self.dbi)
     t = threading.Thread(target=s.start)
     t.start()
     s.shutdown()
     t.join()
     self.assertFalse(s.is_running)
コード例 #2
0
    def test_many_stills(self):
        dbi = FakeDataBaseInterface()
        try:
            os.mkdir('still0')
            os.mkdir('still1')
        except (OSError):
            pass
        ts0 = ts.TaskServer(dbi, data_dir='still0', port=4441)
        ts1 = ts.TaskServer(dbi, data_dir='still1', port=4442)
        ts0_thd = threading.Thread(target=ts0.start)
        ts1_thd = threading.Thread(target=ts1.start)
        tc = {
            0: ts.TaskClient(dbi, 'localhost', port=4441),
            1: ts.TaskClient(dbi, 'localhost', port=4442)
        }
        s = ts.Scheduler(tc, actions_per_still=2, blocksize=2)
        s_thd = threading.Thread(target=s.start,
                                 args=(dbi, ),
                                 kwargs={
                                     'ActionClass': ts.Action,
                                     'action_args': (tc, 30),
                                     'sleeptime': 0
                                 })
        ts0_thd.start()
        ts1_thd.start()
        s_thd.start()
        try:

            def all_done():
                for f in dbi.files:
                    if dbi.get_obs_status(f) != 'COMPLETE': return False
                return True

            while not all_done():
                time.sleep(.1)
        finally:
            s.quit()
            ts0.shutdown()
            ts1.shutdown()
            s_thd.join()
            ts0_thd.join()
            ts1_thd.join()
        for i in dbi.files:
            self.assertEqual(dbi.get_obs_status(i), 'COMPLETE')
コード例 #3
0
ファイル: task_server_test.py プロジェクト: nkern/capo
 def test_tx(self):
     self.pkt = ''
     class SleepHandler(ts.TaskHandler):
         def handle(me):
             self.pkt = me.get_pkt()
     s = ts.TaskServer(self.dbi, handler=SleepHandler)
     thd = threading.Thread(target=s.start)
     thd.start()
     try:
         tc = ts.TaskClient(self.dbi, 'localhost')
         tc.tx('UV', 1)
     finally:
         s.shutdown()
         thd.join()
     self.assertEqual(self.pkt, ('UV',1,'localhost',['test.uv', 'localhost:./test.uv']))
コード例 #4
0
ファイル: task_server_test.py プロジェクト: nkern/capo
 def test_basics(self):
     s = ts.TaskServer(self.dbi)
     t = SleepTask('UV',1,'still',[],self.dbi)
     s.append_task(t)
     self.assertEqual(len(s.active_tasks), 1)
     t.run()
     s.kill(t.process.pid)
     while t.poll() is None: time.sleep(.01)
     self.assertEqual(t.poll(), -9)
     thd = threading.Thread(target=s.finalize_tasks, args=(.1,))
     s.is_running = True
     thd.start()
     s.is_running = False
     thd.join()
     self.assertEqual(len(s.active_tasks),0)
コード例 #5
0
ファイル: task_server_test.py プロジェクト: nkern/capo
 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()
コード例 #6
0
ファイル: task_server_test.py プロジェクト: nkern/capo
 def test_send_task(self):
     self.var = 0
     class SleepHandler(ts.TaskHandler):
         def handle(me):
             self.var += 1
             t = SleepTask('UV',1,'still',[],self.dbi)
             t.run()
             me.server.append_task(t)
     s = ts.TaskServer(self.dbi, handler=SleepHandler)
     thd = threading.Thread(target=s.start)
     thd.start()
     try:
         self.assertEqual(len(s.active_tasks), 0)
         self.assertEqual(self.var, 0)
         sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
         sock.sendto('test', ('localhost', ts.STILL_PORT))
         while self.var != 1: time.sleep(.1)
         self.assertEqual(self.var, 1)
         self.assertEqual(len(s.active_tasks), 1)
     finally:
         s.shutdown()
         thd.join()
コード例 #7
0
 def setUp(self):
     self.dbi = FakeDataBaseInterface()
     self.ts = ts.TaskServer(self.dbi)
     self.ts_thread = threading.Thread(target=self.ts.start)
     self.ts_thread.start()