From 4f95c68ef540ff32a30f11b57b6d4f0d06409029 Mon Sep 17 00:00:00 2001 From: Vadim Fedorov Date: Fri, 28 Mar 2025 12:03:30 +0100 Subject: [PATCH] Update tests --- tests/test_filters/test_callback_data.py | 15 ++++++++++----- 1 file changed, 10 insertions(+), 5 deletions(-) diff --git a/tests/test_filters/test_callback_data.py b/tests/test_filters/test_callback_data.py index 1bc50cae..e0e9b1f3 100644 --- a/tests/test_filters/test_callback_data.py +++ b/tests/test_filters/test_callback_data.py @@ -28,13 +28,18 @@ class MyCallback(CallbackData, prefix="test"): class TestCallbackData: - def test_init_subclass_prefix_required(self): - assert MyCallback.__prefix__ == "test" + def test_init_subclass_prefix_optional(self): + # Case 1: Explicitly provided prefix + class ExplicitCallbackData(CallbackData, prefix="explicit"): + pass - with pytest.raises(ValueError, match="prefix required.+"): + assert ExplicitCallbackData.__prefix__ == "explicit" - class MyInvalidCallback(CallbackData): - pass + # Case 2: No prefix provided; should default to class name + class DefaultCallbackData(CallbackData): + pass + + assert DefaultCallbackData.__prefix__ == "DefaultCallbackData" def test_init_subclass_sep_validation(self): assert MyCallback.__separator__ == ":"