Ensure optional addon values are set

matrix-org · Jan 19, 2024 · a051c51 · a051c51
1 parent b6b812b
commit a051c51
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/tests/mitmproxy_addons/status_code.py b/tests/mitmproxy_addons/status_code.py
@@ -42,6 +42,10 @@ def configure(self, updates):
             self.reset()
             return
         self.config = ctx.options.statuscode
+        # ensure optional fields are set
+        self.config.setdefault("count", 0)
+        self.config.setdefault("block_request", False)
+
         new_filter = self.config.get('filter', None)
         print(f"statuscode will return HTTP {self.config['return_status']} filter={new_filter} count={self.config['count']}")
         if new_filter: