diff options
Diffstat (limited to 'table/destination_test.go')
| -rw-r--r-- | table/destination_test.go | 57 |
1 files changed, 57 insertions, 0 deletions
diff --git a/table/destination_test.go b/table/destination_test.go index 1169e9e0..32316f5f 100644 --- a/table/destination_test.go +++ b/table/destination_test.go @@ -160,6 +160,63 @@ func TestCalculate2(t *testing.T) { assert.Equal(t, len(d.knownPathList), 3) } +func TestImplicitWithdrawCalculate(t *testing.T) { + origin := bgp.NewPathAttributeOrigin(0) + aspathParam := []bgp.AsPathParamInterface{bgp.NewAs4PathParam(2, []uint32{65001})} + aspath := bgp.NewPathAttributeAsPath(aspathParam) + nexthop := bgp.NewPathAttributeNextHop("10.0.0.1") + med := bgp.NewPathAttributeMultiExitDisc(0) + pathAttributes := []bgp.PathAttributeInterface{origin, aspath, nexthop, med} + nlri := bgp.NewIPAddrPrefix(24, "10.10.0.101") + updateMsg := bgp.NewBGPUpdateMessage(nil, pathAttributes, []*bgp.IPAddrPrefix{nlri}) + peer1 := &PeerInfo{AS: 1, Address: net.IP{1, 1, 1, 1}} + path1 := ProcessMessage(updateMsg, peer1, time.Now())[0] + path1.Filter("1", POLICY_DIRECTION_IMPORT) + + // suppose peer2 has import policy to prepend as-path + action := &AsPathPrependAction{ + asn: 100, + repeat: 1, + } + + path2 := action.Apply(path1.Clone(false)) + path1.Filter("2", POLICY_DIRECTION_IMPORT) + path2.Filter("1", POLICY_DIRECTION_IMPORT) + path2.Filter("3", POLICY_DIRECTION_IMPORT) + + d := NewDestination(nlri) + d.addNewPath(path1) + d.addNewPath(path2) + + d.Calculate(nil) + + assert.Equal(t, len(d.GetKnownPathList("1")), 0) // peer "1" is the originator + assert.Equal(t, len(d.GetKnownPathList("2")), 1) + assert.Equal(t, d.GetKnownPathList("2")[0].GetAsString(), "100 65001") // peer "2" has modified path {100, 65001} + assert.Equal(t, len(d.GetKnownPathList("3")), 1) + assert.Equal(t, d.GetKnownPathList("3")[0].GetAsString(), "65001") // peer "3" has original path {65001} + assert.Equal(t, len(d.knownPathList), 2) + + // say, we removed peer2's import policy and + // peer1 advertised new path with the same prefix + aspathParam = []bgp.AsPathParamInterface{bgp.NewAs4PathParam(2, []uint32{65001, 65002})} + aspath = bgp.NewPathAttributeAsPath(aspathParam) + pathAttributes = []bgp.PathAttributeInterface{origin, aspath, nexthop, med} + updateMsg = bgp.NewBGPUpdateMessage(nil, pathAttributes, []*bgp.IPAddrPrefix{nlri}) + path3 := ProcessMessage(updateMsg, peer1, time.Now())[0] + path3.Filter("1", POLICY_DIRECTION_IMPORT) + + d.addNewPath(path3) + d.Calculate(nil) + + assert.Equal(t, len(d.GetKnownPathList("1")), 0) // peer "1" is the originator + assert.Equal(t, len(d.GetKnownPathList("2")), 1) + assert.Equal(t, d.GetKnownPathList("2")[0].GetAsString(), "65001 65002") // peer "2" has new original path {65001, 65002} + assert.Equal(t, len(d.GetKnownPathList("3")), 1) + assert.Equal(t, d.GetKnownPathList("3")[0].GetAsString(), "65001 65002") // peer "3" has new original path {65001, 65002} + assert.Equal(t, len(d.knownPathList), 1) +} + func DestCreatePeer() []*PeerInfo { peerD1 := &PeerInfo{AS: 65000} peerD2 := &PeerInfo{AS: 65001} |