summaryrefslogtreecommitdiffhomepage
path: root/table
diff options
context:
space:
mode:
Diffstat (limited to 'table')
-rw-r--r--table/destination_test.go10
-rw-r--r--table/path.go11
-rw-r--r--table/path_test.go21
-rw-r--r--table/table.go4
-rw-r--r--table/table_test.go24
5 files changed, 31 insertions, 39 deletions
diff --git a/table/destination_test.go b/table/destination_test.go
index 7584ae47..29534c04 100644
--- a/table/destination_test.go
+++ b/table/destination_test.go
@@ -27,14 +27,14 @@ func TestDestinationNewIPv4(t *testing.T) {
peerD := DestCreatePeer()
msgD := DestCreateMSG(peerD)
pathD := DestCreatePath(msgD)
- ipv4d := NewIPv4Destination(pathD[0].getNlri())
+ ipv4d := NewIPv4Destination(pathD[0].GetNlri())
assert.NotNil(t, ipv4d)
}
func TestDestinationNewIPv6(t *testing.T) {
peerD := DestCreatePeer()
msgD := DestCreateMSG(peerD)
pathD := DestCreatePath(msgD)
- ipv6d := NewIPv6Destination(pathD[0].getNlri())
+ ipv6d := NewIPv6Destination(pathD[0].GetNlri())
assert.NotNil(t, ipv6d)
}
@@ -82,7 +82,7 @@ func TestDestinationSetBestPath(t *testing.T) {
peerD := DestCreatePeer()
msgD := DestCreateMSG(peerD)
pathD := DestCreatePath(msgD)
- ipv4d := NewIPv4Destination(pathD[0].getNlri())
+ ipv4d := NewIPv4Destination(pathD[0].GetNlri())
ipv4d.setBestPath(pathD[0])
r_pathD := ipv4d.getBestPath()
assert.Equal(t, r_pathD, pathD[0])
@@ -91,7 +91,7 @@ func TestDestinationGetBestPath(t *testing.T) {
peerD := DestCreatePeer()
msgD := DestCreateMSG(peerD)
pathD := DestCreatePath(msgD)
- ipv4d := NewIPv4Destination(pathD[0].getNlri())
+ ipv4d := NewIPv4Destination(pathD[0].GetNlri())
ipv4d.setBestPath(pathD[0])
r_pathD := ipv4d.getBestPath()
assert.Equal(t, r_pathD, pathD[0])
@@ -100,7 +100,7 @@ func TestDestinationCalculate(t *testing.T) {
peerD := DestCreatePeer()
msgD := DestCreateMSG(peerD)
pathD := DestCreatePath(msgD)
- ipv4d := NewIPv4Destination(pathD[0].getNlri())
+ ipv4d := NewIPv4Destination(pathD[0].GetNlri())
//best path selection
ipv4d.addNewPath(pathD[0])
ipv4d.addNewPath(pathD[1])
diff --git a/table/path.go b/table/path.go
index 649ab10a..444fe838 100644
--- a/table/path.go
+++ b/table/path.go
@@ -37,8 +37,7 @@ type Path interface {
getSourceVerNum() int
setWithdraw(withdraw bool)
isWithdraw() bool
- setNlri(nlri bgp.AddrPrefixInterface)
- getNlri() bgp.AddrPrefixInterface
+ GetNlri() bgp.AddrPrefixInterface
getPrefix() net.IP
setMedSetByTargetNeighbor(medSetByTargetNeighbor bool)
getMedSetByTargetNeighbor() bool
@@ -125,11 +124,7 @@ func (pd *PathDefault) isWithdraw() bool {
return pd.withdraw
}
-func (pd *PathDefault) setNlri(nlri bgp.AddrPrefixInterface) {
- pd.nlri = nlri
-}
-
-func (pd *PathDefault) getNlri() bgp.AddrPrefixInterface {
+func (pd *PathDefault) GetNlri() bgp.AddrPrefixInterface {
return pd.nlri
}
@@ -170,7 +165,7 @@ func (pi *PathDefault) clone(forWithdrawal bool) Path {
if !forWithdrawal {
pathAttrs = pi.getPathAttributeMap()
}
- def := NewPathDefault(pi.getSource(), pi.getNlri(), pi.getSourceVerNum(),
+ def := NewPathDefault(pi.getSource(), pi.GetNlri(), pi.getSourceVerNum(),
pi.getNexthop(), forWithdrawal, pathAttrs, pi.getMedSetByTargetNeighbor())
switch pi.ROUTE_FAMILY {
case RF_IPv4_UC:
diff --git a/table/path_test.go b/table/path_test.go
index 3de2561e..e5a2397f 100644
--- a/table/path_test.go
+++ b/table/path_test.go
@@ -13,7 +13,7 @@ func TestPathNewIPv4(t *testing.T) {
peerP := PathCreatePeer()
msgP := PathCreateMSG(peerP)
pathP := PathCreatePath(msgP)
- ipv4p := NewIPv4Path(pathP[0].getSource(), pathP[0].getNlri(), pathP[0].getSourceVerNum(), pathP[0].getNexthop(),
+ ipv4p := NewIPv4Path(pathP[0].getSource(), pathP[0].GetNlri(), pathP[0].getSourceVerNum(), pathP[0].getNexthop(),
true, pathP[0].getPathAttributeMap(), pathP[0].getMedSetByTargetNeighbor())
assert.NotNil(t, ipv4p)
}
@@ -21,7 +21,7 @@ func TestPathNewIPv6(t *testing.T) {
peerP := PathCreatePeer()
msgP := PathCreateMSG(peerP)
pathP := PathCreatePath(msgP)
- ipv6p := NewIPv6Path(pathP[0].getSource(), pathP[0].getNlri(), pathP[0].getSourceVerNum(), pathP[0].getNexthop(),
+ ipv6p := NewIPv6Path(pathP[0].getSource(), pathP[0].GetNlri(), pathP[0].getSourceVerNum(), pathP[0].getNexthop(),
true, pathP[0].getPathAttributeMap(), pathP[0].getMedSetByTargetNeighbor())
assert.NotNil(t, ipv6p)
}
@@ -122,20 +122,16 @@ func TestPathGetWithdaw(t *testing.T) {
r_wd := pd.isWithdraw()
assert.Equal(t, r_wd, wd)
}
-func TestPathSetNlri(t *testing.T) {
- pd := &PathDefault{}
- nlri := bgp.NewNLRInfo(24, "13.2.3.1")
- pd.setNlri(nlri)
- r_nlri := pd.getNlri()
- assert.Equal(t, r_nlri, nlri)
-}
+
func TestPathGetNlri(t *testing.T) {
- pd := &PathDefault{}
nlri := bgp.NewNLRInfo(24, "13.2.3.2")
- pd.setNlri(nlri)
- r_nlri := pd.getNlri()
+ pd := &PathDefault{
+ nlri: nlri,
+ }
+ r_nlri := pd.GetNlri()
assert.Equal(t, r_nlri, nlri)
}
+
func TestPathSetMedSetByTargetNeighbor(t *testing.T) {
pd := &PathDefault{}
msbt := true
@@ -143,6 +139,7 @@ func TestPathSetMedSetByTargetNeighbor(t *testing.T) {
r_msbt := pd.getMedSetByTargetNeighbor()
assert.Equal(t, r_msbt, msbt)
}
+
func TestPathGetMedSetByTargetNeighbor(t *testing.T) {
pd := &PathDefault{}
msbt := true
diff --git a/table/table.go b/table/table.go
index 9b1aaa08..85110f7b 100644
--- a/table/table.go
+++ b/table/table.go
@@ -62,8 +62,8 @@ func insert(table Table, path Path) Destination {
var dest Destination
table.validatePath(path)
- table.validateNlri(path.getNlri())
- dest = getOrCreateDest(table, path.getNlri())
+ table.validateNlri(path.GetNlri())
+ dest = getOrCreateDest(table, path.GetNlri())
if path.isWithdraw() {
// withdraw insert
diff --git a/table/table_test.go b/table/table_test.go
index 63ae026f..47213a82 100644
--- a/table/table_test.go
+++ b/table/table_test.go
@@ -41,13 +41,13 @@ func TestTableDeleteDestByNlri(t *testing.T) {
pathT := TableCreatePath(msgT)
ipv4t := NewIPv4Table(0)
for _, path := range pathT {
- tableKey := ipv4t.tableKey(path.getNlri())
- dest := ipv4t.createDest(path.getNlri())
+ tableKey := ipv4t.tableKey(path.GetNlri())
+ dest := ipv4t.createDest(path.GetNlri())
ipv4t.setDestination(tableKey.String(), dest)
}
- tableKey := ipv4t.tableKey(pathT[0].getNlri())
+ tableKey := ipv4t.tableKey(pathT[0].GetNlri())
gdest := ipv4t.getDestination(tableKey.String())
- rdest := deleteDestByNlri(ipv4t, pathT[0].getNlri())
+ rdest := deleteDestByNlri(ipv4t, pathT[0].GetNlri())
assert.Equal(t, rdest, gdest)
}
@@ -57,12 +57,12 @@ func TestTableDeleteDest(t *testing.T) {
pathT := TableCreatePath(msgT)
ipv4t := NewIPv4Table(0)
for _, path := range pathT {
- tableKey := ipv4t.tableKey(path.getNlri())
- dest := ipv4t.createDest(path.getNlri())
+ tableKey := ipv4t.tableKey(path.GetNlri())
+ dest := ipv4t.createDest(path.GetNlri())
ipv4t.setDestination(tableKey.String(), dest)
}
- tableKey := ipv4t.tableKey(pathT[0].getNlri())
- dest := ipv4t.createDest(pathT[0].getNlri())
+ tableKey := ipv4t.tableKey(pathT[0].GetNlri())
+ dest := ipv4t.createDest(pathT[0].GetNlri())
ipv4t.setDestination(tableKey.String(), dest)
deleteDest(ipv4t, dest)
gdest := ipv4t.getDestination(tableKey.String())
@@ -82,8 +82,8 @@ func TestTableSetDestinations(t *testing.T) {
ipv4t := NewIPv4Table(0)
destinations := make(map[string]Destination)
for _, path := range pathT {
- tableKey := ipv4t.tableKey(path.getNlri())
- dest := ipv4t.createDest(path.getNlri())
+ tableKey := ipv4t.tableKey(path.GetNlri())
+ dest := ipv4t.createDest(path.GetNlri())
destinations[tableKey.String()] = dest
}
ipv4t.setDestinations(destinations)
@@ -97,8 +97,8 @@ func TestTableGetDestinations(t *testing.T) {
ipv4t := NewIPv4Table(0)
destinations := make(map[string]Destination)
for _, path := range pathT {
- tableKey := ipv4t.tableKey(path.getNlri())
- dest := ipv4t.createDest(path.getNlri())
+ tableKey := ipv4t.tableKey(path.GetNlri())
+ dest := ipv4t.createDest(path.GetNlri())
destinations[tableKey.String()] = dest
}
ipv4t.setDestinations(destinations)