@VisibleForTesting static InitializationMetadata getInitializationMetadata(String network, String snapshot) throws IOException { // already resolved return SnapshotMetadataMgr.getInitializationMetadata( new NetworkId(network), new SnapshotId(snapshot)); } }
@Override public @Nonnull NetworkId getNetworkId(String network) { if (!hasNetworkId(network)) { throw new IllegalArgumentException( String.format("No ID assigned to non-existent network %s", network)); } return new NetworkId(CommonUtil.readFile(getNetworkIdPath(network))); }
private static void updateInitializationStatus( String network, String snapshot, ProcessingStatus status, String errMessage) throws IOException { // already resolved SnapshotMetadataMgr.updateInitializationStatus( new NetworkId(network), new SnapshotId(snapshot), status, errMessage); }
public NetworkSnapshot networkSnapshot() { return new NetworkSnapshot( new NetworkId(UUID.randomUUID().toString()), new SnapshotId(UUID.randomUUID().toString())); }
@Override public NetworkSnapshot getNetworkSnapshot() { return new NetworkSnapshot(new NetworkId("a"), new SnapshotId("b")); } })
@Override public NetworkSnapshot getNetworkSnapshot() { return new NetworkSnapshot(new NetworkId("a"), new SnapshotId("b")); } })
@Test public void testLoadSnapshotInputObjectFile() throws IOException { NetworkId network = new NetworkId("network"); SnapshotId snapshot = new SnapshotId("snapshot"); String testSting = "what is life"; FileUtils.copyInputStreamToFile( new ByteArrayInputStream(testSting.getBytes()), _storage.getSnapshotInputObjectPath(network, snapshot, "test").toFile()); try (InputStream inputStream = _storage.loadSnapshotInputObject(network, snapshot, "test")) { assertThat(IOUtils.toString(inputStream, StandardCharsets.UTF_8.name()), equalTo(testSting)); } }
@Test public void testCheckNetworkExistsFalse() { NetworkId network = new NetworkId("network"); assertThat(_storage.checkNetworkExists(network), equalTo(false)); }
@Test public void roundTripConfigurationsSucceeds() { NetworkId network = new NetworkId("network"); SnapshotId snapshot = new SnapshotId("snapshot"); Map<String, Configuration> configs = new HashMap<>(); configs.put("node1", new Configuration("node1", ConfigurationFormat.CISCO_IOS)); _storage.storeConfigurations( configs, new ConvertConfigurationAnswerElement(), network, snapshot); Map<String, Configuration> deserialized = _storage.loadConfigurations(network, snapshot); assertThat(deserialized, not(nullValue())); assertThat(deserialized.keySet(), equalTo(Sets.newHashSet("node1"))); }
private QueuedWork resolvedQueuedWork(WorkItem workItem, WorkDetails workDetails) { WorkItem resolvedWorkItem = WorkMgr.resolveIds(workItem); return new QueuedWork( WorkMgr.resolveIds(workItem), resolveIds(new NetworkId(resolvedWorkItem.getContainerName()), workDetails)); }
@Test public void testCheckNetworkExistsTrue() { NetworkId network = new NetworkId("network"); _storage.getDirectoryProvider().getNetworkDir(network).toFile().mkdirs(); assertThat(_storage.checkNetworkExists(network), equalTo(true)); }
@Test public void testLoadCompletionMetadataMissing() throws IOException { NetworkId networkId = new NetworkId("network"); SnapshotId snapshotId = new SnapshotId("snapshot"); // if CompletionMetadata file is missing, should return a CompletionMetadata object with all // fields empty assertThat( _storage.loadCompletionMetadata(networkId, snapshotId), equalTo(CompletionMetadata.EMPTY)); } }
@Test public void testLoadMissingMajorIssueConfig() { IssueSettingsId majorIssue = new IssueSettingsId("majorIssue"); NetworkId network = new NetworkId("network"); assertThat(_storage.loadMajorIssueConfig(network, majorIssue), nullValue()); }
@Test public void testLoadWorkLog() throws IOException { // setup: pretend a worker logger has written a file NetworkId network = new NetworkId("network"); SnapshotId snapshot = new SnapshotId("snapshot"); Path dir = _storage.getDirectoryProvider().getSnapshotOutputDir(network, snapshot); final boolean mkdirs = dir.toFile().mkdirs(); assertThat(mkdirs, equalTo(true)); CommonUtil.writeFile(dir.resolve("workid.log"), "testoutput"); // Test: read log using storage API assertThat(_storage.loadWorkLog(network, snapshot, "workid"), equalTo("testoutput")); }
@Test public void loadMissingConfigurationsReturnsNull() { assertThat( _storage.loadConfigurations(new NetworkId("nonexistent"), new SnapshotId("nonexistent")), nullValue()); }
@Test public void testLoadQuestionSettingsMissing() throws IOException { NetworkId network = new NetworkId("network"); QuestionSettingsId questionSettingsId = new QuestionSettingsId("q1"); assertThat(_storage.loadQuestionSettings(network, questionSettingsId), nullValue()); }
@Test public void testLoadWorkLogMissing() throws IOException { // setup: pretend a worker logger has written a file NetworkId network = new NetworkId("network"); SnapshotId snapshot = new SnapshotId("snapshot"); _thrown.expect(FileNotFoundException.class); assertThat(_storage.loadWorkLog(network, snapshot, "workid"), equalTo("testoutput")); }
@Test public void testStoreQuestionSettingsThenLoad() throws IOException { NetworkId network = new NetworkId("network"); QuestionSettingsId questionSettingsId = new QuestionSettingsId("q1"); String settings = "{}"; _storage.storeQuestionSettings(settings, network, questionSettingsId); assertThat(_storage.loadQuestionSettings(network, questionSettingsId), equalTo(settings)); }
@Test public void testMajorIssueConfigRoundTrip() throws IOException { String majorIssue = "majorIssue"; IssueSettingsId issueSettingsId = new IssueSettingsId("issueSettingsId"); NetworkId network = new NetworkId("network"); MinorIssueConfig minorIssueConfig = new MinorIssueConfig("minorIssue", 100, "www.google.com"); MajorIssueConfig majorIssueConfig = new MajorIssueConfig(majorIssue, ImmutableList.of(minorIssueConfig)); _storage.storeMajorIssueConfig(network, issueSettingsId, majorIssueConfig); assertThat(_storage.loadMajorIssueConfig(network, issueSettingsId), equalTo(majorIssueConfig)); }
@Test public void getCompletedWorkBadFilter() throws Exception { // Make sure we get no results or error filtering on a bogus snapshot List<QueuedWork> works0 = _workQueueMgr.getCompletedWork(new NetworkId("bogus"), new SnapshotId("bogus")); assertThat(works0, iterableWithSize(0)); }