simple test

Author: nat3058

livekit-api/tests/test_webhook.py

@@ -81,51 +81,14 @@ def test_invalid_body():
     with pytest.raises(Exception): # Using a broad Exception for existing test
         receiver.receive(body, jwt)
 
-
-def test_mismatched_api_key_secret():
-    """
-    Test that receiving a webhook with a token signed by a different API key/secret
-    raises an error.
-    """
-    TEST_API_KEY_BAD = "badkey"
-    TEST_API_SECRET_BAD = "badsecret"
-
-    token_verifier = TokenVerifier(TEST_API_KEY, TEST_API_SECRET)
-    receiver = WebhookReceiver(token_verifier)
-
-    # Token signed with incorrect credentials
-    token = AccessToken(TEST_API_KEY_BAD, TEST_API_SECRET_BAD)
-    hash64 = base64.b64encode(hashlib.sha256(TEST_EVENT.encode()).digest()).decode()
-    token.claims.sha256 = hash64
-    jwt = token.to_jwt()
-
-    # The LiveKit API internally catches jwt.exceptions.InvalidSignatureError
-    # and re-raises it as a LiveKitError with a message starting "could not verify token: "
-    # followed by the original JWT error message.
-    expected_match = r"could not verify token: Signature verification failed"
-    with pytest.raises(Exception, match=expected_match):
-        receiver.receive(TEST_EVENT, jwt)
-
-
-def test_expired_token():
+def test_malformed_jwt():
     """
-    Test that receiving a webhook with an expired token raises an ExpiredSignatureError.
+    Test that receiving a webhook with a malformed JWT string raises an error.
     """
     token_verifier = TokenVerifier(TEST_API_KEY, TEST_API_SECRET)
     receiver = WebhookReceiver(token_verifier)
 
-    token = AccessToken(TEST_API_KEY, TEST_API_SECRET)
-    hash64 = base64.b64encode(hashlib.sha256(TEST_EVENT.encode()).digest()).decode()
-    token.claims.sha256 = hash64
-
-    # Set the token's expiration to a time in the past
-    # Using datetime.now(UTC) to address the DeprecationWarning
-    token.claims.exp = datetime.now(UTC) - timedelta(seconds=60) # 1 minute ago
-
-    jwt = token.to_jwt()
+    malformed_jwt = "this.is.not.a.valid.jwt" # A clearly malformed string
 
-    # Similar to mismatched key, LiveKit API wraps the ExpiredSignatureError
-    # The message will start with "could not verify token: "
-    expected_match = r"could not verify token: Signature has expired"
-    with pytest.raises(Exception, match=expected_match):
-        receiver.receive(TEST_EVENT, jwt)
+    with pytest.raises(Exception):
+        receiver.receive(TEST_EVENT, malformed_jwt)