55from typing import Any , Dict , List , cast
66
77
8- def normalize_list (obj : Any ) -> List [Any ]:
9- if obj is None :
10- return []
11- if isinstance (obj , list ):
12- return obj
13- return [obj ]
14-
15-
168class TestServerSxPB (unittest .TestCase ):
179 def test_sxpb_cases (self ):
1810 sxpb_path = os .path .join (os .path .dirname (__file__ ), "server_test_cases.sxpb" )
@@ -32,8 +24,7 @@ def test_sxpb_cases(self):
3224 # Setup phase
3325 setup_block = case .get ("setup" , {})
3426 if setup_block :
35- setup_requests = normalize_list (setup_block .get ("request" ))
36- for req in setup_requests :
27+ for req in cast (List [Dict [str , Any ]], setup_block .get ("requests" , [])):
3728 self ._run_request (impl , req , now_ms )
3829
3930 # Request phase
@@ -79,7 +70,7 @@ def _run_request(self, impl: RendezqueueImpl, req: Dict[str, Any], now_ms: float
7970 msg ["b64" ] = req ["b64" ]
8071
8172 if "values" in req :
82- msg ["values" ] = normalize_list ( req ["values" ])
73+ msg ["values" ] = req ["values" ]
8374
8475 return impl .tryswap (msg , now_ms = now_ms )
8576
@@ -109,9 +100,8 @@ def _verify_response(self, actual, expected, case_name):
109100 pass
110101
111102 if "values" in expected :
112- expected_values = normalize_list (expected ["values" ])
113103 self .assertEqual (
114- actual .values , expected_values , f"Case { case_name } : Values mismatch"
104+ actual .values , expected [ "values" ] , f"Case { case_name } : Values mismatch"
115105 )
116106 else :
117107 # Expect no values
0 commit comments