test: Fix some more untyped tests (#4373)

Makefile

@@ -36,7 +36,7 @@ api-tests:
 	SNUBA_SETTINGS=test pytest -vv tests/*_api.py
 
 backend-typing:
-	mypy snuba tests scripts --strict --config-file mypy.ini --exclude 'tests/datasets|tests/query|tests/state|tests/snapshots|tests/clickhouse|tests/test_split.py'
+	mypy snuba tests scripts --strict --config-file mypy.ini --exclude 'tests/datasets|tests/query|tests/snapshots|tests/clickhouse|tests/test_split.py'
 
 install-python-dependencies:
 	pip uninstall -qqy uwsgi  # pip doesn't do well with swapping drop-ins

tests/state/test_cache.py

@@ -16,7 +16,10 @@
 from snuba.state.cache.abstract import Cache, ExecutionError, ExecutionTimeoutError
 from snuba.state.cache.redis.backend import RedisCache
 from snuba.utils.codecs import ExceptionAwareCodec
-from snuba.utils.serializable_exception import SerializableException
+from snuba.utils.serializable_exception import (
+    SerializableException,
+    SerializableExceptionDict,
+)
 from tests.assertions import assert_changes, assert_does_not_change
 
 redis_client = get_redis_client(RedisClientKey.CACHE)
@@ -44,7 +47,7 @@ def encode(self, value: bytes) -> bytes:
 
     def decode(self, value: bytes) -> bytes:
         try:
-            ret = rapidjson.loads(value)
+            ret: SerializableExceptionDict = rapidjson.loads(value)
             if not isinstance(ret, dict):
                 return value
             if ret.get("__type__", "NOP") == "SerializableException":
@@ -55,7 +58,7 @@ def decode(self, value: bytes) -> bytes:
         return value
 
     def encode_exception(self, value: SerializableException) -> bytes:
-        return rapidjson.dumps(value.to_dict()).encode("utf-8")
+        return bytes(rapidjson.dumps(value.to_dict()).encode("utf-8"))
 
 
 @pytest.fixture
@@ -260,10 +263,10 @@ def test_notify_queue_ttl() -> None:
     num_waiters = 9
 
     class DelayedRedisClient:
-        def __init__(self, redis_client):
+        def __init__(self, redis_client: RedisClientType) -> None:
             self._client = redis_client
 
-        def __getattr__(self, attr: str):
+        def __getattr__(self, attr: str) -> Any:
             # simulate the queue pop taking longer than expected.
             # the notification queue TTL is 60 seconds so running into a timeout
             # shouldn't happen (unless something has drastically changed in the TTL

tests/state/test_record.py

@@ -1,5 +1,5 @@
 from snuba.state import _kafka_producer
 
 
-def test_get_producer():
+def test_get_producer() -> None:
     assert _kafka_producer() is not None