From cb938486194c19857f88d143039f7680389ae9b5 Mon Sep 17 00:00:00 2001
From: Ji Hwan <jkim@polygon.technology>
Date: Tue, 29 Oct 2024 13:45:34 +0900
Subject: [PATCH] test: add test for policyMapping

Signed-off-by: Ji Hwan <jkim@polygon.technology>
---
 zk/txpool/policy_test.go | 45 ++++++++++++++++++++++++++++++++++++++++
 1 file changed, 45 insertions(+)

diff --git a/zk/txpool/policy_test.go b/zk/txpool/policy_test.go
index 82456b66700..66fae9cc9c0 100644
--- a/zk/txpool/policy_test.go
+++ b/zk/txpool/policy_test.go
@@ -189,6 +189,51 @@ func TestRemovePolicy(t *testing.T) {
 	})
 }
 
+func TestPolicyMapping(t *testing.T) {
+	// All policies
+	var policiesAll []byte
+	var pListAll []Policy
+	policiesAll = append(policiesAll, SendTx.ToByte())
+	policiesAll = append(policiesAll, Deploy.ToByte())
+	pListAll = append(pListAll, SendTx)
+	pListAll = append(pListAll, Deploy)
+
+	// Only sendTx policy
+	var policiesSendTx []byte
+	var pListSendTx []Policy
+	policiesSendTx = append(policiesSendTx, SendTx.ToByte())
+	pListSendTx = append(pListSendTx, SendTx)
+
+	// Only deploy policy
+	var policiesDeploy []byte
+	var pListDeploy []Policy
+	policiesDeploy = append(policiesDeploy, Deploy.ToByte())
+	pListDeploy = append(pListDeploy, Deploy)
+
+	// No policy
+	var policiesNone []byte
+	var pListNone []Policy
+
+	var tests = []struct {
+		policies []byte
+		pList    []Policy
+		want     string
+	}{
+		{policiesAll, pListAll, "\tsendTx: true\n\tdeploy: true"},
+		{policiesSendTx, pListSendTx, "\tsendTx: true"},
+		{policiesDeploy, pListDeploy, "\tdeploy: true"},
+		{policiesNone, pListNone, ""},
+	}
+	for _, tt := range tests {
+		t.Run("PolicyMapping", func(t *testing.T) {
+			ans := policyMapping(tt.policies, tt.pList)
+			if ans != tt.want {
+				t.Errorf("got %v, want %v", ans, tt.want)
+			}
+		})
+	}
+}
+
 func TestAddPolicy(t *testing.T) {
 	t.Parallel()