mirror of
https://github.com/TeaOSLab/EdgeNode.git
synced 2025-11-05 01:20:26 +08:00
47 lines
964 B
Go
47 lines
964 B
Go
|
package iplibrary
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
|
||
|
|
"github.com/TeaOSLab/EdgeCommon/pkg/serverconfigs/firewallconfigs"
|
||
|
|
"testing"
|
||
|
|
"time"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestScriptAction_AddItem(t *testing.T) {
|
||
|
|
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) {
|
||
|
|
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")
|
||
|
|
}
|