def test_config_file_path_not_specified(self, mock_sys_error, mock_print_usage, mock_argparse): """Test main() with no config_file_path specified.""" expected_exit_code = 2 mock_argparse.return_value = NameSpace(endpoint='[::]:50051', services=['scanner'], forseti_db=None, config_file_path=None, log_level='info', enable_console_log=False) mock_print_usage.return_value = None with self.assertRaises(SystemExit) as e: server.main() self.assertEqual(expected_exit_code, e.exception.code) self.assertTrue(mock_sys_error.write.called)
def test_config_file_path_non_existent_file(self, mock_sys_error, mock_print_usage, mock_argparse): """Test main() with non-existent config file.""" expected_exit_code = 4 mock_argparse.return_value = NameSpace(endpoint='[::]:50051', services=['scanner'], forseti_db=None, config_file_path='/what/ever', log_level='info', enable_console_log=False) mock_print_usage.return_value = None with mock.patch.object(server.os.path, "isfile") as mock_isfile: mock_isfile.return_value = True with mock.patch.object(server.os, "access") as mock_access: mock_access.return_value = False with self.assertRaises(SystemExit) as e: server.main() self.assertEqual(expected_exit_code, e.exception.code) self.assertTrue(mock_sys_error.write.called)
def RunForsetiServer(): """Run Forseti API server.""" import google.cloud.forseti.services.server as forseti_server forseti_server.main()