From df4796da340e5718ec7904073d95693653209a8b Mon Sep 17 00:00:00 2001
From: Daniel Lavrushin <lavrushin@gmail.com>
Date: Sun, 16 Aug 2026 03:12:04 +0200
Subject: [PATCH] fix(tests): enhance stubBinaries function to properly manage
 binary cache cleanup

---
 src/tables/mssclamp_test.go | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/src/tables/mssclamp_test.go b/src/tables/mssclamp_test.go
index dafb7865..8281d105 100644
--- a/src/tables/mssclamp_test.go
+++ b/src/tables/mssclamp_test.go
@@ -19,8 +19,24 @@ func mssClampSet(id string, size int, ips, macs []string) *config.SetConfig {
 	return &set
 }
 
+func stubBinaries(t *testing.T, names ...string) {
+	t.Helper()
+	for _, name := range names {
+		prev, had := hasBinaryCache.Load(name)
+		t.Cleanup(func() {
+			if had {
+				hasBinaryCache.Store(name, prev)
+			} else {
+				hasBinaryCache.Delete(name)
+			}
+		})
+		hasBinaryCache.Store(name, true)
+	}
+}
+
 func mssSpecs(t *testing.T, cfg *config.Config, ipt string) []string {
 	t.Helper()
+	stubBinaries(t, backendIPTables, backendIP6Tables, "ipset")
 	manager := NewIPTablesManager(cfg, false)
 	_, rules := manager.buildMSSManifest("PREROUTING")
 	var out []string