def test_should_load_file_if_config_file_present(self): with patch('data.options._parse_args') as args_mock: args_mock.return_value = {'config_file': 'dummy_file.ini'} with patch('data.options._parse_ini_file') as ini_mock: ini_mock.return_value = { 'failure': False, 'successful_read': True, 'server_configs': [{ 'host': '192.168.1.23', 'password': '******', 'user': '******' }, { 'host': '10.0.0.1', 'password': '******', 'user': '******' }] } config = get_config() expectations = { 'servers': [{ 'host': '192.168.1.23', 'password': '******', 'user': '******' }, { 'host': '10.0.0.1', 'password': '******', 'user': '******' }] } self.assertEquals(expectations, config)
def test_should_exit_if_cannot_parse_config_file(self): with patch('data.options._parse_args') as args_mock: args_mock.return_value = {'config_file': 'dummy_file.ini'} with patch('data.options._parse_ini_file') as ini_mock: ini_mock.return_value = {'failure': True, 'successful_read': False, 'server_configs': []} old_stderr = sys.stderr sys.stderr = StringIO() try: with self.assertRaises(SystemExit): config = get_config() self.assertEquals("Could not parse 'dummy_file.ini'\n", sys.stderr.getvalue()) finally: sys.stderr = old_stderr
def test_should_exit_if_cannot_parse_config_file(self): with patch('data.options._parse_args') as args_mock: args_mock.return_value = {'config_file': 'dummy_file.ini'} with patch('data.options._parse_ini_file') as ini_mock: ini_mock.return_value = { 'failure': True, 'successful_read': False, 'server_configs': [] } old_stderr = sys.stderr sys.stderr = StringIO() try: with self.assertRaises(SystemExit): config = get_config() self.assertEquals("Could not parse 'dummy_file.ini'\n", sys.stderr.getvalue()) finally: sys.stderr = old_stderr
def test_should_load_file_if_config_file_present(self): with patch('data.options._parse_args') as args_mock: args_mock.return_value = {'config_file': 'dummy_file.ini'} with patch('data.options._parse_ini_file') as ini_mock: ini_mock.return_value = {'failure': False, 'successful_read': True, 'server_configs': [{'host': '192.168.1.23', 'password': '******', 'user': '******'}, {'host': '10.0.0.1', 'password': '******', 'user': '******'}] } config = get_config() expectations = {'servers': [ {'host': '192.168.1.23', 'password': '******', 'user': '******'}, {'host': '10.0.0.1', 'password': '******', 'user': '******'}]} self.assertEquals(expectations, config)
def test_should_remove_config_file_parameter(self, _parse_args_mock): _parse_args_mock.return_value = {'config_file': ''} config = get_config() self.assertEquals({}, config)
def main(config, static_path): hostname = 'testing' monitor = WorkerMonitor() server_state = ServerState() if not config['test']: for server_config in config['servers']: add_server(monitor, server_config, server_state, config['stat_window']) print('Started gathering varnish data') else: from tests.data_generator import DummyDataGenerator dummydata = DummyDataGenerator(server_state) worker = Worker('Testing', dummydata) monitor.add_worker(worker) Worker('WorkerMonitor', monitor).start() http_server.start(server_state, static_path, config['port'], config['wsgi_server']) print('Simverest stopping...') monitor.stop() if __name__ == "__main__": static_path = os.path.join(os.path.dirname(__file__), 'web', 'static') try: main(get_config(), static_path) except: pass
health = VarnishHealth(host, user, password, hostname, server_state) monitor.add_worker(Worker(health_name, health)) def main(config, static_path): hostname = 'testing' monitor = WorkerMonitor() server_state = ServerState() if not config['test']: for server_config in config['servers']: add_server(monitor, server_config, server_state, config['stat_window']) print('Started gathering varnish data') else: from tests.data_generator import DummyDataGenerator dummydata = DummyDataGenerator(server_state) worker = Worker('Testing', dummydata) monitor.add_worker(worker) Worker('WorkerMonitor', monitor).start() http_server.start(server_state, static_path, config['port'], config['wsgi_server']) print('Simverest stopping...') monitor.stop() if __name__ == "__main__": static_path = os.path.join(os.path.dirname(__file__), 'web', 'static') try: main(get_config(), static_path) except: pass