diff --git a/tests/test_aliases.py b/tests/test_aliases.py index 8e529a2..a82fa22 100644 --- a/tests/test_aliases.py +++ b/tests/test_aliases.py @@ -13,7 +13,6 @@ from announce_server.decorator import ( @pytest.mark.parametrize( "decorator_alias", [ - announce_server, register_block, register_server, ], @@ -35,3 +34,28 @@ def test_alias_calls_register_service(decorator_alias): decorator(MagicMock()) mock_register_service.assert_called_once_with(*test_args, **test_kwargs) + + +@pytest.mark.parametrize( + "decorator_alias", + [ + announce_server, + ], +) +def test_deprecated_alias_calls_register_service(decorator_alias): + test_args = (None, None) + test_kwargs = { + "name": "test_server", + "ip": "127.0.0.1", + "port": 8000, + "host_ip": "127.0.0.1", + "host_port": 5000, + } + with patch("announce_server.decorator.register_service") as mock_register_service: + mock_register_service.return_value = MagicMock() + + with pytest.warns(DeprecationWarning): + decorator = decorator_alias(*test_args, **test_kwargs) + + decorator(MagicMock()) + mock_register_service.assert_called_once_with(*test_args, **test_kwargs)