def test_api_get_output_file(self, mock_url, mock_creds, mock_config,
mock_download):
"""Test get_output_file"""
_callback = mock.Mock()
_api = BatchAppsApi(mock_creds, mock_config)
mock_url.return_value = "https://test.com/{jobid}/{name}"
mock_download.return_value = {}
val = _api.get_output_file("c:\\dir", 500, False)
self.assertFalse(mock_download.called)
self.assertFalse(val.success)
val = _api.get_output_file("c:\\dir", 500, False, url="http://url")
mock_download.assert_called_with(mock_creds,
"http://url",
self.headers,
"c:\\dir",
500,
False,
f_name=None,
block_size=4096,
callback=None)
self.assertTrue(val.success)
val = _api.get_output_file("c:\\dir",
500,
False,
url="http://url",
job_id="test_id",
callback=_callback,
block=1)
mock_download.assert_called_with(mock_creds,
"http://url",
self.headers,
"c:\\dir",
500,
False,
f_name=None,
block_size=1,
callback=_callback)
self.assertTrue(val.success)
mock_url.reset()
mock_download.called = False
val = _api.get_output_file("c:\\dir", 500, False, job_id="test_id")
self.assertFalse(mock_url.called)
self.assertFalse(mock_download.called)
self.assertFalse(val.success)
val = _api.get_output_file("c:\\dir",
500,
False,
job_id="test_id",
fname="test.zip")
self.assertTrue(val.success)
def test_api_get_output_file(self, mock_url, mock_creds, mock_config, mock_download):
"""Test get_output_file"""
_callback = mock.Mock()
_api = BatchAppsApi(mock_creds, mock_config)
mock_url.return_value = "https://test.com/{jobid}/{name}"
mock_download.return_value = {}
val = _api.get_output_file("c:\\dir", 500, False)
self.assertFalse(mock_download.called)
self.assertFalse(val.success)
val = _api.get_output_file("c:\\dir", 500, False, url="http://url")
mock_download.assert_called_with(
mock_creds, "http://url", self.headers, "c:\\dir", 500, False, f_name=None, block_size=4096, callback=None
)
self.assertTrue(val.success)
val = _api.get_output_file(
"c:\\dir", 500, False, url="http://url", job_id="test_id", callback=_callback, block=1
)
mock_download.assert_called_with(
mock_creds, "http://url", self.headers, "c:\\dir", 500, False, f_name=None, block_size=1, callback=_callback
)
self.assertTrue(val.success)
mock_url.reset()
mock_download.called = False
val = _api.get_output_file("c:\\dir", 500, False, job_id="test_id")
self.assertFalse(mock_url.called)
self.assertFalse(mock_download.called)
self.assertFalse(val.success)
val = _api.get_output_file("c:\\dir", 500, False, job_id="test_id", fname="test.zip")
self.assertTrue(val.success)
