class TestStatusAlgorithm(unittest.TestCase): def setUp(self): handler_mock = HandlerMock test_policy_file = open('test_policy_file', 'w') test_policy_file.close() time_stamp = os.path.getmtime('test_policy_file') szig = SZIG("", handler_mock) szig.handler.data["info"]["policy"]["file"] = 'test_policy_file' szig.handler.data["info"]["policy"]["file_stamp"] = time_stamp szig.handler.data["info"]["policy"]["reload_stamp"] = time_stamp self.algorithm = StatusAlgorithm() self.algorithm.setInstance(Instance(name='testinstance', process_num=0)) self.algorithm.szig = szig self.algorithm.pidfiledir = self.test_dir = './var/run/zorp/' if not os.path.exists(self.test_dir): os.makedirs(self.test_dir) self.test_pid_file = 'zorp-testinstance#0.pid' pid_file = open(self.test_dir + self.test_pid_file, 'w') pid_file.write('123456') pid_file.close() def __del__(self): os.remove("test_policy_file") os.remove(self.test_dir + self.test_pid_file) os.removedirs(self.test_dir) def test_status(self): status = self.algorithm.status() self.assertEquals(status.pid, 123456) self.assertEquals( status.threads, self.algorithm.szig.handler.data['stats']['threads_running']) self.assertEquals( status.policy_file, self.algorithm.szig.handler.data['info']['policy']['file']) self.assertEquals( status.reload_timestamp, self.algorithm.szig.handler.data['info']['policy']['reload_stamp']) self.assertEquals( status.timestamp_szig, self.algorithm.szig.handler.data['info']['policy']['file_stamp'])
def setUp(self): handler_mock = HandlerMock test_policy_file = open('test_policy_file', 'w') test_policy_file.close() time_stamp = os.path.getmtime('test_policy_file') szig = SZIG("", handler_mock) szig.handler.data["info"]["policy"]["file"] = 'test_policy_file' szig.handler.data["info"]["policy"]["file_stamp"] = time_stamp szig.handler.data["info"]["policy"]["reload_stamp"] = time_stamp self.algorithm = StatusAlgorithm() self.algorithm.setInstance(Instance(name='testinstance', process_num=0)) self.algorithm.szig = szig self.algorithm.pidfiledir = self.test_dir = './var/run/zorp/' if not os.path.exists(self.test_dir): os.makedirs(self.test_dir) self.test_pid_file = 'zorp-testinstance#0.pid' pid_file = open(self.test_dir + self.test_pid_file, 'w') pid_file.write('123456') pid_file.close()
class TestStatusAlgorithm(unittest.TestCase): def setUp(self): handler_mock = HandlerMock test_policy_file = open('test_policy_file', 'w') test_policy_file.close() time_stamp = os.path.getmtime('test_policy_file') szig = SZIG("", handler_mock) szig.handler.data["info"]["policy"]["file"] = 'test_policy_file' szig.handler.data["info"]["policy"]["file_stamp"] = time_stamp szig.handler.data["info"]["policy"]["reload_stamp"] = time_stamp self.algorithm = StatusAlgorithm() self.algorithm.setInstance(Instance(name='testinstance', process_num=0)) self.algorithm.szig = szig self.algorithm.prefix = "." self.test_dir = './var/run/zorp/' if not os.path.exists(self.test_dir): os.makedirs(self.test_dir) self.test_pid_file = 'zorp-testinstance#0.pid' pid_file = open(self.test_dir + self.test_pid_file, 'w') pid_file.write('123456') pid_file.close() def __del__(self): os.remove("test_policy_file") os.remove(self.test_dir + self.test_pid_file) os.removedirs(self.test_dir) def test_status(self): status = self.algorithm.status() self.assertEquals(status.pid, 123456) self.assertEquals(status.threads, self.algorithm.szig.handler.data['stats']['threads_running']) self.assertEquals(status.policy_file, self.algorithm.szig.handler.data['info']['policy']['file']) self.assertEquals(status.reload_timestamp, self.algorithm.szig.handler.data['info']['policy']['reload_stamp']) self.assertEquals(status.timestamp_szig, self.algorithm.szig.handler.data['info']['policy']['file_stamp'])
def status(): return ZorpHandler.callAlgorithmToAllInstances(StatusAlgorithm())