Examples with Environment org.batfish.datamodel.routing_policy.Environment used on opensource projects

Example 1 with Environment

use of org.batfish.datamodel.routing_policy.Environment in project batfish by batfish.

the class PrependAsPathTest method testPrepend.

@Test
public void testPrepend() {
    List<AsExpr> prepend = Lists.newArrayList(new ExplicitAs(1), new ExplicitAs(2));
    PrependAsPath operation = new PrependAsPath(new LiteralAsList(prepend));
    BgpRoute.Builder builder = new BgpRoute.Builder();
    builder.setAsPath(mkAsPath(3, 4));
    Environment env = newTestEnvironment(builder);
    operation.execute(env);
    assertThat(builder.getAsPath(), equalTo(mkAsPath(1, 2, 3, 4)));
}

Example 2 with Environment

use of org.batfish.datamodel.routing_policy.Environment in project batfish by batfish.

the class PrependAsPathTest method testPrependWithIntermediateAttributes.

@Test
public void testPrependWithIntermediateAttributes() {
    List<AsExpr> prepend = Lists.newArrayList(new ExplicitAs(1), new ExplicitAs(2));
    PrependAsPath operation = new PrependAsPath(new LiteralAsList(prepend));
    BgpRoute.Builder outputRoute = new BgpRoute.Builder();
    outputRoute.setAsPath(mkAsPath(3, 4));
    BgpRoute.Builder intermediateAttributes = new BgpRoute.Builder();
    intermediateAttributes.setAsPath(mkAsPath(5, 6));
    Environment env = newTestEnvironment(outputRoute);
    env.setIntermediateBgpAttributes(intermediateAttributes);
    env.setWriteToIntermediateBgpAttributes(true);
    operation.execute(env);
    assertThat(outputRoute.getAsPath(), equalTo(mkAsPath(1, 2, 3, 4)));
    assertThat(intermediateAttributes.getAsPath(), equalTo(mkAsPath(1, 2, 5, 6)));
}