@@ -107,8 +107,8 @@ async def test_seed_and_list(bundle_path):
107107 assert seed_data ["errors" ] == []
108108
109109 list_result = await session .call_tool ("list_contacts" , {})
110- contacts = json .loads (list_result .content [0 ].text )
111- assert len ( contacts ) == 2
110+ result_data = json .loads (list_result .content [0 ].text )
111+ assert result_data [ "count" ] == 2
112112
113113
114114@pytest .mark .asyncio
@@ -134,15 +134,15 @@ async def test_full_crud_cycle(bundle_path):
134134 assert contact_id .startswith ("ct_" )
135135
136136 # Get
137- get_result = await session .call_tool ("get_contact" , {"entity_id " : contact_id })
137+ get_result = await session .call_tool ("get_contact" , {"contact_id " : contact_id })
138138 fetched = json .loads (get_result .content [0 ].text )
139139 assert fetched ["first_name" ] == "Bob"
140140 assert fetched ["email" ] == "bob@test.com"
141141
142142 # Update
143143 update_result = await session .call_tool (
144144 "update_contact" ,
145- {"entity_id " : contact_id , "data" : {"lead_score" : 85 }},
145+ {"contact_id " : contact_id , "data" : {"lead_score" : 85 }},
146146 )
147147 updated = json .loads (update_result .content [0 ].text )
148148 assert updated ["lead_score" ] == 85
@@ -153,19 +153,16 @@ async def test_full_crud_cycle(bundle_path):
153153 "search_contacts" ,
154154 {"query" : "Bob" },
155155 )
156- results = json .loads (search_result .content [0 ].text )
157- assert len ( results ) == 1
158- assert results [0 ]["lead_score" ] == 85
156+ search_data = json .loads (search_result .content [0 ].text )
157+ assert search_data [ "count" ] == 1
158+ assert search_data [ "entities" ] [0 ]["lead_score" ] == 85
159159
160160 # Delete
161- delete_result = await session .call_tool ("delete_contact" , {"entity_id " : contact_id })
161+ delete_result = await session .call_tool ("delete_contact" , {"contact_id " : contact_id })
162162 deleted = json .loads (delete_result .content [0 ].text )
163163 assert deleted ["status" ] == "deleted"
164164
165165 # Verify gone from list
166166 list_result = await session .call_tool ("list_contacts" , {})
167- if list_result .content :
168- active = json .loads (list_result .content [0 ].text )
169- else :
170- active = []
171- assert all (c ["id" ] != contact_id for c in active )
167+ list_data = json .loads (list_result .content [0 ].text )
168+ assert all (c ["id" ] != contact_id for c in list_data ["entities" ])
0 commit comments