diff --git a/tests/helpers/stubs.py b/tests/helpers/stubs.py
index f414fba74..adaebd168 100644
--- a/tests/helpers/stubs.py
+++ b/tests/helpers/stubs.py
@@ -385,8 +385,14 @@ class KeyConfigStub:
 
     """Stub for the key-config object."""
 
-    def get_bindings_for(self, _section):
-        return {}
+    def __init__(self):
+        self.bindings = {}
+
+    def get_bindings_for(self, section):
+        return self.bindings.get(section)
+
+    def set_bindings_for(self, section, bindings):
+        self.bindings[section] = bindings
 
 
 class HostBlockerStub: