_configurations.remove(router); topology.prune(null, toRemove, null);
private Topology computeLayer3Topology(NetworkSnapshot networkSnapshot) { try (ActiveSpan span = GlobalTracer.get().buildSpan("TopologyProviderImpl::computeLayer3Topology").startActive()) { assert span != null; // avoid unused warning Map<String, Configuration> configurations = _batfish.loadConfigurations(networkSnapshot); Topology topology = getRawLayer3Topology(networkSnapshot); topology.prune( _batfish.getEdgeBlacklist(networkSnapshot), _batfish.getNodeBlacklist(networkSnapshot), _batfish.getInterfaceBlacklist(networkSnapshot)); topology.pruneFailedIpsecSessionEdges( IpsecUtil.initIpsecTopology(configurations), configurations); return topology; } }
@Test public void testPruneNode2() { Topology topo = new Topology(_bothEdges); topo.prune(null, ImmutableSet.of("n2"), null); assertThat(topo.getEdges(), equalTo(ImmutableSet.of())); assertThat(topo.getInterfaceEdges(), equalTo(ImmutableMap.of())); assertThat(topo.getNodeEdges(), equalTo(ImmutableMap.of())); assertThat(topo.getNeighbors(_nip1), equalTo(ImmutableSet.of())); assertThat(topo.getNeighbors(_nip3), equalTo(ImmutableSet.of())); }
prune(failedIpsecEdges, ImmutableSet.of(), ImmutableSet.of());
@Test public void testPruneInterface2() { Topology topo = new Topology(_bothEdges); topo.prune(null, null, ImmutableSet.of(_nip2)); assertThat(topo.getEdges(), equalTo(ImmutableSet.of())); assertThat(topo.getInterfaceEdges(), equalTo(ImmutableMap.of())); assertThat(topo.getNodeEdges(), equalTo(ImmutableMap.of())); assertThat(topo.getNeighbors(_nip1), equalTo(ImmutableSet.of())); assertThat(topo.getNeighbors(_nip3), equalTo(ImmutableSet.of())); }
@Test public void testPruneInterface1() { Topology topo = new Topology(_bothEdges); topo.prune(null, null, ImmutableSet.of(_nip1)); assertThat(topo.getEdges(), equalTo(_edge2to3Set)); assertThat( topo.getInterfaceEdges(), equalTo(ImmutableMap.of(_nip2, _edge2to3Set, _nip3, _edge2to3Set))); assertThat( topo.getNodeEdges(), equalTo(ImmutableMap.of("n2", _edge2to3Set, "n3", _edge2to3Set))); assertThat(topo.getNeighbors(_nip2), equalTo(ImmutableSet.of(_nip3))); assertThat(topo.getNeighbors(_nip3), equalTo(ImmutableSet.of())); }
@Test public void testPruneNode3() { Topology topo = new Topology(_bothEdges); topo.prune(null, ImmutableSet.of("n3"), null); assertThat(topo.getEdges(), equalTo(_edge1to2Set)); assertThat( topo.getInterfaceEdges(), equalTo(ImmutableMap.of(_nip1, _edge1to2Set, _nip2, _edge1to2Set))); assertThat( topo.getNodeEdges(), equalTo(ImmutableMap.of("n1", _edge1to2Set, "n2", _edge1to2Set))); assertThat(topo.getNeighbors(_nip1), equalTo(ImmutableSet.of(_nip2))); assertThat(topo.getNeighbors(_nip2), equalTo(ImmutableSet.of())); }
@Test public void testPruneInterface3() { Topology topo = new Topology(_bothEdges); topo.prune(null, null, ImmutableSet.of(_nip3)); assertThat(topo.getEdges(), equalTo(_edge1to2Set)); assertThat( topo.getInterfaceEdges(), equalTo(ImmutableMap.of(_nip1, _edge1to2Set, _nip2, _edge1to2Set))); assertThat( topo.getNodeEdges(), equalTo(ImmutableMap.of("n1", _edge1to2Set, "n2", _edge1to2Set))); assertThat(topo.getNeighbors(_nip1), equalTo(ImmutableSet.of(_nip2))); assertThat(topo.getNeighbors(_nip2), equalTo(ImmutableSet.of())); } }
@Test public void testPruneNode1() { Topology topo = new Topology(_bothEdges); topo.prune(null, ImmutableSet.of("n1"), null); assertThat(topo.getEdges(), equalTo(_edge2to3Set)); assertThat( topo.getInterfaceEdges(), equalTo(ImmutableMap.of(_nip2, _edge2to3Set, _nip3, _edge2to3Set))); assertThat( topo.getNodeEdges(), equalTo(ImmutableMap.of("n2", _edge2to3Set, "n3", _edge2to3Set))); assertThat(topo.getNeighbors(_nip2), equalTo(ImmutableSet.of(_nip3))); assertThat(topo.getNeighbors(_nip3), equalTo(ImmutableSet.of())); }
@Test public void testPruneEdge() { Topology topo = new Topology(_bothEdges); topo.prune(_edge1to2Set, null, null); assertThat(topo.getEdges(), equalTo(_edge2to3Set)); assertThat( topo.getInterfaceEdges(), equalTo(ImmutableMap.of(_nip2, _edge2to3Set, _nip3, _edge2to3Set))); assertThat( topo.getNodeEdges(), equalTo(ImmutableMap.of("n2", _edge2to3Set, "n3", _edge2to3Set))); assertThat(topo.getNeighbors(_nip1), equalTo(ImmutableSet.of())); assertThat(topo.getNeighbors(_nip2), equalTo(ImmutableSet.of(_nip3))); assertThat(topo.getNeighbors(_nip3), equalTo(ImmutableSet.of())); }