优化nftables集合元素过期时间判断

2025-12-25 14:36:34 +08:00 · 2023-04-19 13:20:31 +08:00
parent 37ddff86f1
commit 7e8c09a684
2 changed files with 13 additions and 2 deletions
--- a/internal/firewalls/nftables/expration_test.go
+++ b/internal/firewalls/nftables/expration_test.go
@@ -17,6 +17,14 @@ func TestExpiration_Add(t *testing.T) {
 		expiration.Add([]byte{'a', 'b', 'c'}, time.Now())
 		t.Log(expiration.Contains([]byte{'a', 'b', 'c'}))
 	}
+	{
+		expiration.Add([]byte{'a', 'b', 'c'}, time.Now().Add(1*time.Second))
+		t.Log(expiration.Contains([]byte{'a', 'b', 'c'}))
+	}
+	{
+		expiration.Add([]byte{'a', 'b', 'c'}, time.Time{})
+		t.Log(expiration.Contains([]byte{'a', 'b', 'c'}))
+	}
 	{
 		expiration.Add([]byte{'a', 'b', 'c'}, time.Now().Add(-1*time.Second))
 		t.Log(expiration.Contains([]byte{'a', 'b', 'c'}))