Skip to content

Commit 90f7993

Browse files
abelsm2claude
authored andcommitted
test: add tests for falsy parameter handling
🤖 Generated with [Claude Code](https://claude.ai/code) Co-Authored-By: Claude <[email protected]>
1 parent c09548e commit 90f7993

File tree

1 file changed

+60
-0
lines changed

1 file changed

+60
-0
lines changed

tests.py

Lines changed: 60 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -490,6 +490,38 @@ def test_result_search_series_raises_without_required_parameters(
490490
mock_get_resource.assert_not_called()
491491
mock_get_chunks.assert_not_called()
492492

493+
@patch.object(irDataClient, "_get_chunks")
494+
@patch.object(irDataClient, "_get_resource")
495+
def test_result_search_series_passes_false_params(
496+
self, mock_get_resource, mock_get_chunks
497+
):
498+
client = irDataClient(username="test_user", password="test_password")
499+
mock_get_resource.return_value = {"data": {"chunk_info": "chunk_data"}}
500+
mock_get_chunks.return_value = ["result1", "result2"]
501+
502+
season_year = 2021
503+
season_quarter = 2
504+
race_week_num = 0
505+
official_only = False
506+
response = client.result_search_series(
507+
season_year=season_year,
508+
season_quarter=season_quarter,
509+
race_week_num=race_week_num,
510+
official_only=official_only
511+
)
512+
513+
mock_get_resource.assert_called_once_with(
514+
"/data/results/search_series",
515+
payload={
516+
"season_year": 2021,
517+
"season_quarter": 2,
518+
"race_week_num": 0,
519+
"official_only": False,
520+
},
521+
)
522+
mock_get_chunks.assert_called_once_with("chunk_data")
523+
self.assertEqual(response, ["result1", "result2"])
524+
493525
@patch.object(irDataClient, "_get_resource")
494526
def test_member_with_required_parameters(self, mock_get_resource):
495527
client = irDataClient(username="test_user", password="test_password")
@@ -1143,6 +1175,34 @@ def test_result_search_hosted(self, mock_get_chunks, mock_get_resource):
11431175
mock_get_chunks.assert_called_once()
11441176
self.assertEqual(result, mock_get_chunks.return_value)
11451177

1178+
@patch.object(irDataClient, "_get_resource")
1179+
@patch.object(irDataClient, "_get_chunks")
1180+
def test_result_search_hosted_passes_false_params(self, mock_get_chunks, mock_get_resource):
1181+
mock_get_chunks.return_value = [{"session": "hosted_session"}]
1182+
start_range_begin = "2022-01-01T00:00:00Z"
1183+
start_range_end = "2022-01-31T23:59:59Z"
1184+
expected_payload = {
1185+
"start_range_begin": start_range_begin,
1186+
"start_range_end": start_range_end,
1187+
"cust_id": 111111,
1188+
"car_id": 0,
1189+
"category_ids": [0],
1190+
}
1191+
1192+
result = self.client.result_search_hosted(
1193+
cust_id=111111,
1194+
start_range_begin="2022-01-01T00:00:00Z",
1195+
start_range_end="2022-01-31T23:59:59Z",
1196+
car_id=0,
1197+
category_ids=[0],
1198+
)
1199+
1200+
mock_get_resource.assert_called_once_with(
1201+
"/data/results/search_hosted", payload=expected_payload
1202+
)
1203+
mock_get_chunks.assert_called_once()
1204+
self.assertEqual(result, mock_get_chunks.return_value)
1205+
11461206
@patch.object(irDataClient, "_get_resource")
11471207
def test_result_season_results(self, mock_get_resource):
11481208
mock_get_resource.return_value = [{"result": "season_result"}]

0 commit comments

Comments
 (0)