pkt/server: ListPolicy() and ListStatement() support filtering like ListDefinedSet() and ListPolicyAassignment()

Signed-off-by: FUJITA Tomonori <fujita.tomonori@lab.ntt.co.jp>

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")}