@@ -117,7 +117,7 @@ def test_download_new_update_buffer(self):
117117
118118 # Assertions to verify the behavior
119119 self .requests_mock .get .assert_called_with (
120- "some.url.com" , timeout = 10 , stream = True
120+ "some.url.com" , headers = utils . _BROWSER_HEADERS , timeout = 60 , stream = True
121121 )
122122 buffer .write .assert_has_calls (
123123 [call (b"a" ), call (b"b" ), call (b"c" ), call (b"d" )]
@@ -153,7 +153,7 @@ def test_download_new_update_file(self):
153153
154154 # Assertions to verify the behavior
155155 self .requests_mock .get .assert_called_with (
156- "some.url.com" , timeout = 10 , stream = True
156+ "some.url.com" , headers = utils . _BROWSER_HEADERS , timeout = 60 , stream = True
157157 )
158158 self .open_mock .assert_called_with (output_file + ".part" , "wb" )
159159 handle = self .open_mock ()
@@ -255,7 +255,7 @@ def test_progress_mechanism_enabled_file(self):
255255 tqdm_mock .assert_called_once ()
256256 self .bar_mock .update .assert_has_calls ([call (1 ), call (1 ), call (1 ), call (1 )])
257257 self .requests_mock .get .assert_called_with (
258- "some.url.com" , timeout = 10 , stream = True
258+ "some.url.com" , headers = utils . _BROWSER_HEADERS , timeout = 60 , stream = True
259259 )
260260 self .open_mock .assert_called_with (output_file + ".part" , "wb" )
261261
@@ -285,7 +285,7 @@ def test_progress_mechanism_disabled_file(self):
285285 tqdm_mock .assert_not_called ()
286286 self .bar_mock .update .assert_not_called ()
287287 self .requests_mock .get .assert_called_with (
288- "some.url.com" , timeout = 10 , stream = True
288+ "some.url.com" , headers = utils . _BROWSER_HEADERS , timeout = 60 , stream = True
289289 )
290290 self .open_mock .assert_called_with (output_file + ".part" , "wb" )
291291
0 commit comments