FilesExpand file tree

main

/

test_request_name_param.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

37 lines (23 loc) · 1.22 KB

main

/

test_request_name_param.py

Copy path

Top

File metadata and controls

• Code
• Blame

37 lines (23 loc) · 1.22 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

"""`Request.name_param`: the wire-params key a request type declares for `Mcp-Name` emission."""

from typing import Literal

import mcp_types as types

from mcp_types import CallToolRequest, PingRequest, Request

class _VendorParams(types.RequestParams):

task_id: str

class _VendorRequest(Request[_VendorParams, Literal["vendor/tasks/get"]]):

method: Literal["vendor/tasks/get"] = "vendor/tasks/get"

name_param = "taskId"

def test_request_base_declares_no_name_param() -> None:

assert Request.name_param is None

def test_core_request_types_inherit_none() -> None:

assert CallToolRequest.name_param is None

assert PingRequest.name_param is None

def test_subclass_overrides_by_bare_assignment() -> None:

"""Subclasses set `name_param` by bare assignment; the override is class-local."""

assert _VendorRequest.name_param == "taskId"

assert Request.name_param is None

def test_name_param_is_not_a_pydantic_field() -> None:

request = _VendorRequest(params=_VendorParams(task_id="t-1"))

assert "name_param" not in _VendorRequest.model_fields

dumped = request.model_dump(by_alias=True, mode="json", exclude_none=True)

assert dumped == {"method": "vendor/tasks/get", "params": {"taskId": "t-1"}}