fixing tests

Author: maeste

tests/server/test_integration.py

@@ -29,9 +29,11 @@
     Artifact,
     DataPart,
     InternalError,
+    InvalidParamsError,
     InvalidRequestError,
     JSONParseError,
     Message,
+    MethodNotFoundError,
     Part,
     PushNotificationConfig,
     Role,
@@ -976,7 +978,7 @@ def test_unknown_method(client: TestClient):
     data = response.json()
     assert 'error' in data
     # This should produce an UnsupportedOperationError error code
-    assert data['error']['code'] == InvalidRequestError().code
+    assert data['error']['code'] == MethodNotFoundError().code
 
 
 def test_validation_error(client: TestClient):
@@ -987,7 +989,7 @@ def test_validation_error(client: TestClient):
         json={
             'jsonrpc': '2.0',
             'id': '123',
-            'method': 'messages/send',
+            'method': 'message/send',
             'params': {
                 'message': {
                     # Missing required fields
@@ -999,7 +1001,7 @@ def test_validation_error(client: TestClient):
     assert response.status_code == 200
     data = response.json()
     assert 'error' in data
-    assert data['error']['code'] == InvalidRequestError().code
+    assert data['error']['code'] == InvalidParamsError().code
 
 
 def test_unhandled_exception(client: TestClient, handler: mock.AsyncMock):