Add type hints to synapse/tests/rest/admin (#11590)

This commit is contained in:
Dirk Klimpel 2021-12-16 20:59:56 +01:00 committed by GitHub
parent 1847d027e6
commit 8428ef66c7
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
7 changed files with 74 additions and 57 deletions

View file

@ -16,11 +16,14 @@ from typing import List, Optional
from parameterized import parameterized
from twisted.test.proto_helpers import MemoryReactor
import synapse.rest.admin
from synapse.api.errors import Codes
from synapse.rest.client import login
from synapse.server import HomeServer
from synapse.types import JsonDict
from synapse.util import Clock
from tests import unittest
@ -31,7 +34,7 @@ class FederationTestCase(unittest.HomeserverTestCase):
login.register_servlets,
]
def prepare(self, reactor, clock, hs: HomeServer):
def prepare(self, reactor: MemoryReactor, clock: Clock, hs: HomeServer) -> None:
self.store = hs.get_datastore()
self.register_user("admin", "pass", admin=True)
self.admin_user_tok = self.login("admin", "pass")
@ -44,7 +47,7 @@ class FederationTestCase(unittest.HomeserverTestCase):
("/_synapse/admin/v1/federation/destinations/dummy",),
]
)
def test_requester_is_no_admin(self, url: str):
def test_requester_is_no_admin(self, url: str) -> None:
"""
If the user is not a server admin, an error 403 is returned.
"""
@ -62,7 +65,7 @@ class FederationTestCase(unittest.HomeserverTestCase):
self.assertEqual(HTTPStatus.FORBIDDEN, channel.code, msg=channel.json_body)
self.assertEqual(Codes.FORBIDDEN, channel.json_body["errcode"])
def test_invalid_parameter(self):
def test_invalid_parameter(self) -> None:
"""
If parameters are invalid, an error is returned.
"""
@ -117,7 +120,7 @@ class FederationTestCase(unittest.HomeserverTestCase):
self.assertEqual(HTTPStatus.NOT_FOUND, channel.code, msg=channel.json_body)
self.assertEqual(Codes.NOT_FOUND, channel.json_body["errcode"])
def test_limit(self):
def test_limit(self) -> None:
"""
Testing list of destinations with limit
"""
@ -137,7 +140,7 @@ class FederationTestCase(unittest.HomeserverTestCase):
self.assertEqual(channel.json_body["next_token"], "5")
self._check_fields(channel.json_body["destinations"])
def test_from(self):
def test_from(self) -> None:
"""
Testing list of destinations with a defined starting point (from)
"""
@ -157,7 +160,7 @@ class FederationTestCase(unittest.HomeserverTestCase):
self.assertNotIn("next_token", channel.json_body)
self._check_fields(channel.json_body["destinations"])
def test_limit_and_from(self):
def test_limit_and_from(self) -> None:
"""
Testing list of destinations with a defined starting point and limit
"""
@ -177,7 +180,7 @@ class FederationTestCase(unittest.HomeserverTestCase):
self.assertEqual(len(channel.json_body["destinations"]), 10)
self._check_fields(channel.json_body["destinations"])
def test_next_token(self):
def test_next_token(self) -> None:
"""
Testing that `next_token` appears at the right place
"""
@ -238,7 +241,7 @@ class FederationTestCase(unittest.HomeserverTestCase):
self.assertEqual(len(channel.json_body["destinations"]), 1)
self.assertNotIn("next_token", channel.json_body)
def test_list_all_destinations(self):
def test_list_all_destinations(self) -> None:
"""
List all destinations.
"""
@ -259,7 +262,7 @@ class FederationTestCase(unittest.HomeserverTestCase):
# Check that all fields are available
self._check_fields(channel.json_body["destinations"])
def test_order_by(self):
def test_order_by(self) -> None:
"""
Testing order list with parameter `order_by`
"""
@ -268,7 +271,7 @@ class FederationTestCase(unittest.HomeserverTestCase):
expected_destination_list: List[str],
order_by: Optional[str],
dir: Optional[str] = None,
):
) -> None:
"""Request the list of destinations in a certain order.
Assert that order is what we expect
@ -358,13 +361,13 @@ class FederationTestCase(unittest.HomeserverTestCase):
[dest[0][0], dest[2][0], dest[1][0]], "last_successful_stream_ordering", "b"
)
def test_search_term(self):
def test_search_term(self) -> None:
"""Test that searching for a destination works correctly"""
def _search_test(
expected_destination: Optional[str],
search_term: str,
):
) -> None:
"""Search for a destination and check that the returned destinationis a match
Args:
@ -410,7 +413,7 @@ class FederationTestCase(unittest.HomeserverTestCase):
_search_test(None, "foo")
_search_test(None, "bar")
def test_get_single_destination(self):
def test_get_single_destination(self) -> None:
"""
Get one specific destinations.
"""
@ -429,7 +432,7 @@ class FederationTestCase(unittest.HomeserverTestCase):
# convert channel.json_body into a List
self._check_fields([channel.json_body])
def _create_destinations(self, number_destinations: int):
def _create_destinations(self, number_destinations: int) -> None:
"""Create a number of destinations
Args:
@ -442,7 +445,7 @@ class FederationTestCase(unittest.HomeserverTestCase):
self.store.set_destination_last_successful_stream_ordering(dest, 100)
)
def _check_fields(self, content: List[JsonDict]):
def _check_fields(self, content: List[JsonDict]) -> None:
"""Checks that the expected destination attributes are present in content
Args: