TRKJ/EdgeNode
2
0
Fork 0
mirror of https://github.com/TeaOSLab/EdgeNode.git synced 2025-11-11 05:00:25 +08:00
Code Issues Packages Projects Releases Wiki Activity

增加IP动作

Browse Source
This commit is contained in:
刘祥超
2021-02-06 17:34:33 +08:00
parent 5d58321167
commit 3f469e9284
22 changed files with 1406 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

79
internal/iplibrary/action_firewalld_test.go Normal file
View File

@@ -0,0 +1,79 @@
package iplibrary
import (
"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
"github.com/TeaOSLab/EdgeCommon/pkg/serverconfigs/firewallconfigs"
"testing"
"time"
)
func TestFirewalldAction_AddItem(t *testing.T) {
{
action := NewFirewalldAction()
action.config = &firewallconfigs.FirewallActionFirewalldConfig{
Path: "/usr/bin/firewalld",
}
err := action.AddItem(IPListTypeWhite, &pb.IPItem{
Type: "ipv4",
Id: 1,
IpFrom: "192.168.1.100",
ExpiredAt: time.Now().Unix() + 30,
})
if err != nil {
t.Fatal(err)
}
t.Log("ok")
}
{
action := NewFirewalldAction()
action.config = &firewallconfigs.FirewallActionFirewalldConfig{
Path: "/usr/bin/firewalld",
}
err := action.AddItem(IPListTypeBlack, &pb.IPItem{
Type: "ipv4",
Id: 1,
IpFrom: "192.168.1.101",
ExpiredAt: time.Now().Unix() + 30,
})
if err != nil {
t.Fatal(err)
}
t.Log("ok")
}
}
func TestFirewalldAction_DeleteItem(t *testing.T) {
action := NewFirewalldAction()
action.config = &firewallconfigs.FirewallActionFirewalldConfig{
Path: "/usr/bin/firewalld",
}
err := action.DeleteItem(IPListTypeWhite, &pb.IPItem{
Type: "ipv4",
Id: 1,
IpFrom: "192.168.1.100",
ExpiredAt: time.Now().Unix() + 30,
})
if err != nil {
t.Fatal(err)
}
t.Log("ok")
}
func TestFirewalldAction_MultipleItem(t *testing.T) {
action := NewFirewalldAction()
action.config = &firewallconfigs.FirewallActionFirewalldConfig{
Path: "/usr/bin/firewalld",
}
err := action.AddItem(IPListTypeBlack, &pb.IPItem{
Type: "ipv4",
Id: 1,
IpFrom: "192.168.1.30",
IpTo: "192.168.1.200",
ExpiredAt: time.Now().Unix() + 30,
})
if err != nil {
t.Fatal(err)
}
t.Log("ok")
}