From 8a2a5fecf56a9ddd219639e0d1673948fa54302d Mon Sep 17 00:00:00 2001
From: Michael Pilosov <consistentbayes@gmail.com>
Date: Sun, 19 Mar 2023 17:56:04 -0600
Subject: [PATCH] improve tests

---
 tests/test_aliases.py | 26 +++++++++++++++++++++++++-
 1 file changed, 25 insertions(+), 1 deletion(-)

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)