improve tests

This commit is contained in:
Michael Pilosov 2023-03-19 17:56:04 -06:00
parent 8c414fd87d
commit 8e14fc19e6

View File

@ -13,7 +13,6 @@ from announce_server.decorator import (
@pytest.mark.parametrize( @pytest.mark.parametrize(
"decorator_alias", "decorator_alias",
[ [
announce_server,
register_block, register_block,
register_server, register_server,
], ],
@ -35,3 +34,28 @@ def test_alias_calls_register_service(decorator_alias):
decorator(MagicMock()) decorator(MagicMock())
mock_register_service.assert_called_once_with(*test_args, **test_kwargs) 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)