From a50dcd870508fa6e2b6a701efd986e3ed2d45dc6 Mon Sep 17 00:00:00 2001 From: Jeremiah Lowin <153965+jlowin@users.noreply.github.com> Date: Wed, 18 Mar 2026 11:33:19 -0400 Subject: [PATCH] fix: filter warnings by message in KEY_PREFIX test (#3549) --- tests/utilities/test_components.py | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) diff --git a/tests/utilities/test_components.py b/tests/utilities/test_components.py index 39182d434..a42c9a1f7 100644 --- a/tests/utilities/test_components.py +++ b/tests/utilities/test_components.py @@ -237,8 +237,10 @@ class TestKeyPrefix: class NoPrefix(FastMCPComponent): pass - assert len(w) == 1 - assert "NoPrefix does not define KEY_PREFIX" in str(w[0].message) + key_prefix_warnings = [ + x for x in w if "does not define KEY_PREFIX" in str(x.message) + ] + assert len(key_prefix_warnings) == 1 def test_no_warning_when_key_prefix_defined(self): """Test that subclassing with KEY_PREFIX does not emit a warning.""" @@ -248,7 +250,10 @@ class TestKeyPrefix: class WithPrefix(FastMCPComponent): KEY_PREFIX = "custom" - assert len(w) == 0 + key_prefix_warnings = [ + x for x in w if "does not define KEY_PREFIX" in str(x.message) + ] + assert len(key_prefix_warnings) == 0 assert WithPrefix.make_key("test") == "custom:test"