From 6380f6beaabeb3b9e85751005ff54131407b43f3 Mon Sep 17 00:00:00 2001 From: GoEdgeLab Date: Thu, 7 Mar 2024 19:31:35 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D=E9=83=A8=E5=88=86=E6=B5=8B?= =?UTF-8?q?=E8=AF=95=E7=94=A8=E4=BE=8B?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- internal/firewalls/nftables/conn_test.go | 24 ++++++++++++++++++------ 1 file changed, 18 insertions(+), 6 deletions(-) diff --git a/internal/firewalls/nftables/conn_test.go b/internal/firewalls/nftables/conn_test.go index 8840210..44d2ac8 100644 --- a/internal/firewalls/nftables/conn_test.go +++ b/internal/firewalls/nftables/conn_test.go @@ -7,7 +7,6 @@ package nftables_test import ( "github.com/TeaOSLab/EdgeNode/internal/firewalls/nftables" executils "github.com/TeaOSLab/EdgeNode/internal/utils/exec" - "os/exec" "testing" ) @@ -21,7 +20,10 @@ func TestConn_Test(t *testing.T) { } func TestConn_GetTable_NotFound(t *testing.T) { - var conn = nftables.NewConn() + conn, err := nftables.NewConn() + if err != nil { + t.Fatal(err) + } table, err := conn.GetTable("a", nftables.TableFamilyIPv4) if err != nil { @@ -36,7 +38,10 @@ func TestConn_GetTable_NotFound(t *testing.T) { } func TestConn_GetTable(t *testing.T) { - var conn = nftables.NewConn() + conn, err := nftables.NewConn() + if err != nil { + t.Fatal(err) + } table, err := conn.GetTable("myFilter", nftables.TableFamilyIPv4) if err != nil { @@ -51,7 +56,10 @@ func TestConn_GetTable(t *testing.T) { } func TestConn_AddTable(t *testing.T) { - var conn = nftables.NewConn() + conn, err := nftables.NewConn() + if err != nil { + t.Fatal(err) + } { table, err := conn.AddIPv4Table("test_ipv4") @@ -70,8 +78,12 @@ func TestConn_AddTable(t *testing.T) { } func TestConn_DeleteTable(t *testing.T) { - var conn = nftables.NewConn() - err := conn.DeleteTable("test_ipv4", nftables.TableFamilyIPv4) + conn, err := nftables.NewConn() + if err != nil { + t.Fatal(err) + } + + err = conn.DeleteTable("test_ipv4", nftables.TableFamilyIPv4) if err != nil { t.Fatal(err) }