diff options
Diffstat (limited to 'internal/pkg/table/policy_test.go')
| -rw-r--r-- | internal/pkg/table/policy_test.go | 17 |
1 files changed, 17 insertions, 0 deletions
diff --git a/internal/pkg/table/policy_test.go b/internal/pkg/table/policy_test.go index 7f1a1dd9..31889980 100644 --- a/internal/pkg/table/policy_test.go +++ b/internal/pkg/table/policy_test.go @@ -32,6 +32,23 @@ import ( "github.com/stretchr/testify/require" ) +func TestGetStatement(t *testing.T) { + r := NewRoutingPolicy() + r.statementMap["statement1"] = &Statement{Name: "statement1"} + r.statementMap["statement2"] = &Statement{Name: "statement2"} + assert.Equal(t, len(r.GetStatement("")), 2) + assert.Equal(t, len(r.GetStatement("statement1")), 1) + assert.Equal(t, len(r.GetStatement("unknown")), 0) +} + +func TestGetPolicy(t *testing.T) { + r := NewRoutingPolicy() + r.policyMap["p1"] = &Policy{Name: "p1"} + r.policyMap["p2"] = &Policy{Name: "p2"} + assert.Equal(t, len(r.GetPolicy("")), 2) + assert.Equal(t, len(r.GetPolicy("p1")), 1) + assert.Equal(t, len(r.GetPolicy("unknown")), 0) +} func TestPrefixCalcurateNoRange(t *testing.T) { // create path peer := &PeerInfo{AS: 65001, Address: net.ParseIP("10.0.0.1")} |