EdgeNode/internal/iplibrary/action_script_test.go

package iplibrary

import (
	"testing"
	"time"

	"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
	"github.com/TeaOSLab/EdgeCommon/pkg/serverconfigs/firewallconfigs"
	"github.com/TeaOSLab/EdgeNode/internal/utils/testutils"
)

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