@@ -76,23 +76,11 @@ def test_get_aligned_run_info_success(self, mock_get):
7676 self .assertEqual (len (res ), 4 )
7777 self .assertEqual (next_date , '2025-10-03T01:00:00Z' )
7878
79- @patch ('process_test_history.TestHistoryEntry' )
80- def test_should_process_run_true (self , mock_entry ):
81- mock_entry .query .return_value .get .return_value = None
82- run = {'id' : '1' }
83- self .assertTrue (process_test_history .should_process_run (run ))
84-
85- @patch ('process_test_history.TestHistoryEntry' )
86- def test_should_process_run_false (self , mock_entry ):
87- mock_entry .query .return_value .get .return_value = MagicMock ()
88- run = {'id' : '1' }
89- self .assertFalse (process_test_history .should_process_run (run ))
90-
91- @patch ('process_test_history.should_process_run' , return_value = True )
79+
9280 @patch ('process_test_history.ndb' )
9381 @patch ('process_test_history.storage.Client' )
9482 @patch ('process_test_history.requests.get' )
95- def test_process_single_run (self , mock_get , mock_storage , mock_ndb , mock_should_process ):
83+ def test_process_single_run (self , mock_get , mock_storage , mock_ndb ):
9684 # Mock GCS
9785 mock_bucket = mock_storage .return_value .bucket .return_value
9886 mock_blob = mock_bucket .blob .return_value
@@ -208,11 +196,10 @@ def test_build_new_test_history_entry_uses_deterministic_key(self):
208196 self .assertEqual (res .key .id (), expected_key )
209197
210198
211- @patch ('process_test_history.should_process_run' , return_value = True )
212199 @patch ('process_test_history.ndb' )
213200 @patch ('process_test_history.storage.Client' )
214201 @patch ('process_test_history.requests.get' )
215- def test_process_single_run_batching (self , mock_get , mock_storage , mock_ndb , mock_should_process ):
202+ def test_process_single_run_batching (self , mock_get , mock_storage , mock_ndb ):
216203 # Mock GCS
217204 mock_bucket = mock_storage .return_value .bucket .return_value
218205 mock_blob = mock_bucket .blob .return_value
@@ -257,15 +244,13 @@ def fake_put_multi(ents):
257244
258245 @patch ('process_test_history.ThreadPoolExecutor' )
259246 @patch ('process_test_history.process_single_run' )
260- @patch ('process_test_history.should_process_run' )
261- def test_process_runs_parallel (self , mock_should_process , mock_process_single , mock_executor ):
247+ def test_process_runs_parallel (self , mock_process_single , mock_executor ):
262248 mock_exec_instance = mock_executor .return_value .__enter__ .return_value
263249
264250 runs = [
265251 {'id' : 1 , 'browser_name' : 'chrome' , 'time_start' : '2025-10-03T00:00:00Z' },
266252 {'id' : 2 , 'browser_name' : 'firefox' , 'time_start' : '2025-10-03T00:00:00Z' },
267253 ]
268- mock_should_process .return_value = True
269254
270255 mock_future = MagicMock ()
271256 mock_exec_instance .submit .return_value = mock_future
0 commit comments