fix test files

Author: lukasIO

livekit-api/tests/test_access_token.py

@@ -9,7 +9,7 @@
 TEST_API_SECRET = "thiskeyistotallyunsafe"
 
 
-def test_verify_token():
+def test_verify_token() -> None:
     grants = VideoGrants(room_join=True, room="test_room")
     sip = SIPGrants(admin=True)
 
@@ -30,11 +30,12 @@ def test_verify_token():
     assert claims.metadata == "test_metadata"
     assert claims.video == grants
     assert claims.sip == sip
+    assert claims.attributes is not None
     assert claims.attributes["key1"] == "value1"
     assert claims.attributes["key2"] == "value2"
 
 
-def test_agent_config():
+def test_agent_config() -> None:
     token = (
         AccessToken(TEST_API_KEY, TEST_API_SECRET)
         .with_identity("test_identity")
@@ -50,6 +51,7 @@ def test_agent_config():
     token_verifier = TokenVerifier(TEST_API_KEY, TEST_API_SECRET)
     claims = token_verifier.verify(token)
     # Verify the decoded claims match
+    assert claims.room_config is not None
     assert claims.room_config.agents[0].agent_name == "test-agent"
 
     # Split token into header.payload.signature
@@ -74,7 +76,7 @@ def test_agent_config():
     assert payload_json["roomConfig"]["agents"][0]["agentName"] == "test-agent"
 
 
-def test_verify_token_invalid():
+def test_verify_token_invalid() -> None:
     token = AccessToken(TEST_API_KEY, TEST_API_SECRET).with_identity("test_identity").to_jwt()
 
     token_verifier = TokenVerifier(TEST_API_KEY, "invalid_secret")
@@ -86,7 +88,7 @@ def test_verify_token_invalid():
         token_verifier.verify(token)
 
 
-def test_verify_token_expired():
+def test_verify_token_expired() -> None:
     token = (
         AccessToken(TEST_API_KEY, TEST_API_SECRET)
         .with_identity("test_identity")

livekit-api/tests/test_webhook.py

@@ -45,7 +45,7 @@
 """
 
 
-def test_webhook_receiver():
+def test_webhook_receiver() -> None:
     token_verifier = TokenVerifier(TEST_API_KEY, TEST_API_SECRET)
     receiver = WebhookReceiver(token_verifier)
 
@@ -56,7 +56,7 @@ def test_webhook_receiver():
     receiver.receive(TEST_EVENT, jwt)
 
 
-def test_bad_hash():
+def test_bad_hash() -> None:
     token_verifier = TokenVerifier(TEST_API_KEY, TEST_API_SECRET)
     receiver = WebhookReceiver(token_verifier)
 
@@ -68,7 +68,7 @@ def test_bad_hash():
         receiver.receive(TEST_EVENT, jwt)
 
 
-def test_invalid_body():
+def test_invalid_body() -> None:
     token_verifier = TokenVerifier(TEST_API_KEY, TEST_API_SECRET)
     receiver = WebhookReceiver(token_verifier)
 

livekit-rtc/tests/test_apm.py

@@ -5,7 +5,7 @@
 from livekit.rtc import AudioProcessingModule, AudioFrame
 
 
-def test_audio_processing():
+def test_audio_processing() -> None:
     sample_rate = 48000
     num_channels = 1
     frames_per_chunk = sample_rate // 100

livekit-rtc/tests/test_resampler.py

@@ -1,10 +1,11 @@
-from livekit.rtc import AudioResampler, AudioResamplerQuality
+from livekit.rtc import AudioResampler, AudioResamplerQuality, AudioFrame
 import time
 import wave
 import os
+from typing import List
 
 
-def test_audio_resampler():
+def test_audio_resampler() -> None:
     current_dir = os.path.dirname(os.path.abspath(__file__))
     wav_file_path = os.path.join(current_dir, "test_audio.wav")
 
@@ -27,9 +28,9 @@ def test_audio_resampler():
     ]
 
     for quality in qualities:
-        total_time = 0
+        total_time = 0.0
         nb_runs = 20
-        output_frames = []
+        output_frames: List[AudioFrame] = []
         for i in range(nb_runs):
             output_frames = []
             resampler = AudioResampler(44100, 8000, quality=quality)