diff --git a/rule_test.go b/rule_test.go index 4420e5b..ebcd2e0 100644 --- a/rule_test.go +++ b/rule_test.go @@ -73,7 +73,6 @@ func TestRuleAddDel(t *testing.T) { func TestRuleListFiltered(t *testing.T) { skipUnlessRoot(t) - defer setUpNetlinkTest(t)() t.Run("IPv4", testRuleListFilteredIPv4) t.Run("IPv6", testRuleListFilteredIPv6) @@ -95,8 +94,6 @@ func testRuleListFilteredIPv6(t *testing.T) { } func runRuleListFiltered(t *testing.T, family int, srcNet, dstNet *net.IPNet) { - defaultRules, _ := RuleList(family) - tests := []struct { name string ruleFilter *Rule @@ -112,7 +109,11 @@ func runRuleListFiltered(t *testing.T, family int, srcNet, dstNet *net.IPNet) { preRun: func() *Rule { return nil }, postRun: func(r *Rule) {}, setupWant: func(_ *Rule) ([]Rule, bool) { - return defaultRules, false + rules, err := RuleList(family) + if err != nil { + t.Fatalf("Failed to list rules: %v", err) + } + return rules, false }, }, { @@ -583,6 +584,7 @@ func runRuleListFiltered(t *testing.T, family int, srcNet, dstNet *net.IPNet) { } for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { + defer setUpNetlinkTest(t)() rule := tt.preRun() wantRules, wantErr := tt.setupWant(rule)