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