mirror of
https://github.com/TeaOSLab/EdgeNode.git
synced 2025-11-06 10:00:25 +08:00
42 lines
914 B
Go
42 lines
914 B
Go
|
package iplibrary
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
|
||
|
|
"github.com/TeaOSLab/EdgeCommon/pkg/serverconfigs/firewallconfigs"
|
||
|
|
"testing"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestHTTPAPIAction_AddItem(t *testing.T) {
|
||
|
|
action := NewHTTPAPIAction()
|
||
|
|
action.config = &firewallconfigs.FirewallActionHTTPAPIConfig{
|
||
|
|
URL: "http://127.0.0.1:2345/post",
|
||
|
|
TimeoutSeconds: 0,
|
||
|
|
}
|
||
|
|
err := action.AddItem(IPListTypeBlack, &pb.IPItem{
|
||
|
|
Type: "ipv4",
|
||
|
|
Id: 1,
|
||
|
|
IpFrom: "192.168.1.100",
|
||
|
|
})
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
t.Log("ok")
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestHTTPAPIAction_DeleteItem(t *testing.T) {
|
||
|
|
action := NewHTTPAPIAction()
|
||
|
|
action.config = &firewallconfigs.FirewallActionHTTPAPIConfig{
|
||
|
|
URL: "http://127.0.0.1:2345/post",
|
||
|
|
TimeoutSeconds: 0,
|
||
|
|
}
|
||
|
|
err := action.DeleteItem(IPListTypeBlack, &pb.IPItem{
|
||
|
|
Type: "ipv4",
|
||
|
|
Id: 1,
|
||
|
|
IpFrom: "192.168.1.100",
|
||
|
|
})
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
t.Log("ok")
|
||
|
|
}
|