public BatfishJob(Settings settings) { _settings = settings; _logger = new BatfishLogger(_settings.getLogLevel(), _settings.getTimestamp()); } }
public static void main(String[] args) { checkArgument(args.length == 2, "Expected arguments: <input_dir> <output_dir>"); Path inputPath = Paths.get(args[0]); Path outputPath = Paths.get(args[1]); Settings settings = new Settings(new String[] {"-storagebase", "/"}); settings.setLogger(new BatfishLogger(BatfishLogger.LEVELSTR_WARN, false, System.out)); new Batfish(settings, null, null, null, null, null, null, null, null) .flatten(inputPath, outputPath); } }
@Before public void setup() { _logger = new BatfishLogger(BatfishLogger.LEVELSTR_OUTPUT, false); }
@Before public void before() throws IOException { _containerDir = _folder.newFolder("container").toPath(); _logger = new BatfishLogger(BatfishLogger.LEVELSTR_DEBUG, false); _storage = new FileBasedStorage(_containerDir.getParent(), _logger, (m, n) -> new AtomicInteger()); }
public static void main(String[] args) { mainInit(args); _logger = new BatfishLogger(_settings.getLogLevel(), false, _settings.getLogFile(), false, true); mainRun(new BindPortFutures()); }
new BatfishLogger(_settings.getLogLevel(), false, _settings.getLogFile(), false, false); break; case gendatamodel: _logger = new BatfishLogger(_settings.getLogLevel(), false, _settings.getLogFile(), false, false); break; case genquestions: new BatfishLogger(_settings.getLogLevel(), false, _settings.getLogFile(), false, false); break; case interactive: OutputStream os = new WriterOutputStream(pWriter, StandardCharsets.UTF_8); PrintStream ps = new PrintStream(os, true); _logger = new BatfishLogger(_settings.getLogLevel(), false, ps); } catch (Exception e) { System.err.printf("Could not initialize client: %s\n", e.getMessage());
public static void main(String[] args) { mainInit(args); _mainLogger = new BatfishLogger( _mainSettings.getLogLevel(), _mainSettings.getTimestamp(), _mainSettings.getLogFile(), _mainSettings.getLogTee(), true); mainRun(); }
private void checkProcessCommandErrorMessage( Command command, String[] parameters, String expected) throws Exception { Client client = new Client(new String[] {"-runmode", "gendatamodel"}); File tempFile = _folder.newFile("writer"); FileWriter writer = new FileWriter(tempFile); client._logger = new BatfishLogger("output", false); String[] args = ArrayUtils.addAll(new String[] {command.commandName()}, parameters); assertFalse(client.processCommand(args, writer)); assertThat(client.getLogger().getHistory().toString(500), equalTo(expected)); writer.close(); }
private void testProcessCommandWithValidInput( Command command, String[] parameters, String expected) throws Exception { Client client = new Client(new String[] {"-runmode", "gendatamodel"}); File tempFile = _folder.newFile("writer"); FileWriter writer = new FileWriter(tempFile); String[] args = ArrayUtils.addAll(new String[] {command.commandName()}, parameters); client._logger = new BatfishLogger("output", false); assertTrue(client.processCommand(args, writer)); assertThat(client.getLogger().getHistory().toString(500), equalTo(expected)); writer.close(); }
@Before public void setup() { BatfishLogger logger = new BatfishLogger("debug", false); Main.mainInit( new String[] { "-containerslocation", _networksFolder.getRoot().toString(), "-templatedirs", _questionsTemplatesFolder.getRoot().toString() }); Main.setLogger(logger); Main.initAuthorizer(); }
@Test public void testDefaultCase() throws Exception { Client client = new Client(new String[] {"-runmode", "gendatamodel"}); File tempFile = _folder.newFile("writer"); FileWriter writer = new FileWriter(tempFile); client._logger = new BatfishLogger("output", false); String[] args = new String[] {"non-exist command"}; String expected = "Command failed: Not a valid command: \"non-exist command\"\n"; assertFalse(client.processCommand(args, writer)); assertThat(client.getLogger().getHistory().toString(500), equalTo(expected)); }
new BatfishLogger( settings.getLogLevel(), settings.getTimestamp(),
@Test public void testEmptyQuestionTemplateDir() throws Exception { JSONObject testQuestion = new JSONObject(); testQuestion.put( "instance", new JSONObject() .put("instanceName", "testQuestion") .put("description", "test question description")); Path questionJsonPath = _folder.newFile("testquestion.json").toPath(); CommonUtil.writeFile(questionJsonPath, testQuestion.toString()); Main.mainInit(new String[0]); Main.setLogger(new BatfishLogger("debug", false)); Main.getSettings() .setQuestionTemplateDirs(ImmutableList.of(_folder.getRoot().toPath(), Paths.get(""))); Map<String, String> questionTemplates = getQuestionTemplates(); assertThat(questionTemplates, notNullValue()); assertThat(questionTemplates.keySet(), hasSize(1)); // Both templates should be present assertThat( questionTemplates, IsMapContaining.hasEntry( "testquestion", "{\"instance\":{\"instanceName\":\"testQuestion\",\"description\":\"test question description\"}}")); } }
public static void initWorkManager(IdManager idManager, StorageProvider storage) { BatfishLogger logger = new BatfishLogger("debug", false); Main.mainInit(new String[] {}); Main.setLogger(logger); Main.initAuthorizer(); Main.setWorkMgr(new WorkMgr(Main.getSettings(), logger, idManager, storage)); }
@Test public void testLoadQuestionsNames() throws Exception { Client client = new Client(new String[] {"-runmode", "gendatamodel", "-prettyanswers", "false"}); JSONObject testQuestion = new JSONObject(); testQuestion.put( "instance", new JSONObject() .put("instanceName", "testQuestionName") .put("description", "test question description")); Path questionJsonPath = _folder.newFile("testquestion.json").toPath(); CommonUtil.writeFile(questionJsonPath, testQuestion.toString()); client._logger = new BatfishLogger("output", false); client.processCommand( new String[] {LOAD_QUESTIONS.commandName(), questionJsonPath.getParent().toString()}, null); // Reading the answer written by load-questions Answer answerLoadQuestions = _mapper.readValue( client.getLogger().getHistory().toString(BatfishLogger.LEVEL_OUTPUT), Answer.class); LoadQuestionAnswerElement ae = (LoadQuestionAnswerElement) answerLoadQuestions.getAnswerElements().get(0); // Checking that question name in answer element matches instanceName in file assertEquals("testQuestionName", ae.getAdded().first()); }
@Before public void init() throws Exception { Main.mainInit(new String[0]); Main.setLogger(new BatfishLogger("debug", false)); _workQueueMgr = new WorkQueueMgr(Type.memory, Main.getLogger()); WorkMgrTestUtils.initWorkManager(_folder); Main.getWorkMgr().initNetwork(NETWORK, null); _networkId = Main.getWorkMgr().getIdManager().getNetworkId(NETWORK); }
public static void initWorkManager(TemporaryFolder folder) throws Exception { BatfishLogger logger = new BatfishLogger("debug", false); Main.mainInit(new String[] {"-containerslocation", folder.getRoot().toString()}); Main.setLogger(logger); Main.initAuthorizer(); Main.setWorkMgr( new WorkMgr( Main.getSettings(), logger, new FileBasedIdManager(Main.getSettings().getContainersLocation()), new FileBasedStorage(Main.getSettings().getContainersLocation(), logger))); }
private void initNetwork() { Settings settings = new Settings(new String[] {}); BatfishLogger logger = new BatfishLogger("debug", false); Main.mainInit( new String[] { "-containerslocation", _networksFolder.getRoot().toString(), "-templatedirs", _questionsTemplatesFolder.getRoot().toString() }); Main.setLogger(logger); Main.initAuthorizer(); WorkMgr manager = new WorkMgr( settings, logger, new FileBasedIdManager(Main.getSettings().getContainersLocation()), new FileBasedStorage(Main.getSettings().getContainersLocation(), logger)); Main.setWorkMgr(manager); manager.initNetwork(_networkName, null); manager.getIdManager().assignNetwork(_networkName, _networkId); _service = new WorkMgrService(); }