test_Task.py

Go to the documentation of this file.

#!/usr/bin/env/python
# -*- Python -*-
 
#
# \file Task.py
# \brief Task class
# \date $Date: $
# \author Shinji Kurihara
#
# Copyright (C) 2007
#     Task-intelligence Research Group,
#     Intelligent Systems Research Institute,
#     National Institute of
#         Advanced Industrial Science and Technology (AIST), Japan
#     All rights reserved.
#
 
import sys,time
sys.path.insert(1,"../")
 
import unittest
 
from Task import *
 
cnt = 0
 
class Mock(Task):
        def __init__(self):
                global cnt
                cnt += 1
                self.cnt = cnt
                Task.__init__(self)
                self.flag = False
 
        def svc(self):
                print("cnt: ", end=' ')
                while self.flag:
                        print(self.cnt, " ", end=' ')
                        time.sleep(0.1)
                return
 
        def activate(self):
                self.flag = True
                Task.activate(self)
                return
 
        def finalize(self):
                self.flag = False
 
class Mock2(Task):
        def __init__(self,func):
                global cnt
                cnt += 1
                self.cnt = cnt
                Task.__init__(self)
                self.flag = False
                self.func = func
 
        def svc(self):
                time.sleep(1)
                self.func()
                return
 
        def activate(self):
                self.flag = True
                Task.activate(self)
                return
 
        def finalize(self):
                self.flag = False
                
 
class TestTask(unittest.TestCase):
        def setUp(self):
                self.mock = Mock()
                return
 
        def test_activate(self):
                self.mock.activate()
                mock2= Mock()
                mock2.activate()
                time.sleep(2)
                mock2.finalize()
                self.mock.finalize()
                return
 
        def test_wait(self):
                self.mock.activate()
                time.sleep(0.5)
                mock2= Mock2(self.mock.finalize)
                mock2.activate()
                self.mock.wait()
                mock2.finalize()
                return
    
 
        def test_reset(self):
                self.mock.reset()
                return
 
 
        def test_finalize(self):
                self.mock.finalize()
                return
    
 
        def test_svn_run(self):
                self.mock.svc_run()
                return
 
 
 
if __name__ == '__main__':
        unittest.main()