1.4.5.2

EdgeNode/internal/iplibrary/action_script_test.go

@@ -0,0 +1,55 @@
+package iplibrary
+
+import (
+	"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
+	"github.com/TeaOSLab/EdgeCommon/pkg/serverconfigs/firewallconfigs"
+	"github.com/TeaOSLab/EdgeNode/internal/utils/testutils"
+	"testing"
+	"time"
+)
+
+func TestScriptAction_AddItem(t *testing.T) {
+	if !testutils.IsSingleTesting() {
+		return
+	}
+
+	action := NewScriptAction()
+	action.config = &firewallconfigs.FirewallActionScriptConfig{
+		Path: "/tmp/ip-item.sh",
+		Cwd:  "",
+		Args: nil,
+	}
+	err := action.AddItem(IPListTypeBlack, &pb.IPItem{
+		Type:      "ipv4",
+		Id:        1,
+		IpFrom:    "192.168.1.100",
+		ExpiredAt: time.Now().Unix(),
+	})
+	if err != nil {
+		t.Fatal(err)
+	}
+	t.Log("ok")
+}
+
+func TestScriptAction_DeleteItem(t *testing.T) {
+	if !testutils.IsSingleTesting() {
+		return
+	}
+
+	action := NewScriptAction()
+	action.config = &firewallconfigs.FirewallActionScriptConfig{
+		Path: "/tmp/ip-item.sh",
+		Cwd:  "",
+		Args: nil,
+	}
+	err := action.DeleteItem(IPListTypeBlack, &pb.IPItem{
+		Type:      "ipv4",
+		Id:        1,
+		IpFrom:    "192.168.1.100",
+		ExpiredAt: time.Now().Unix(),
+	})
+	if err != nil {
+		t.Fatal(err)
+	}
+	t.Log("ok")
+}