use of org.batfish.datamodel.routing_policy.expr.ExplicitAs 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)));
}
use of org.batfish.datamodel.routing_policy.expr.ExplicitAs 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)));
}
Aggregations