def test_api_get_output(self, mock_url, mock_creds, mock_config, mock_download):
"""Test get_output"""
_callback = mock.Mock()
_api = BatchAppsApi(mock_creds, mock_config)
mock_url.return_value = "https://test.com/{jobid}/{type}"
mock_download.return_value = {}
val = _api.get_output("c:\\dir", 500, "output.zip", False)
self.assertFalse(mock_download.called)
self.assertFalse(val.success)
val = _api.get_output("c:\\dir", 500, "output.zip", False, url="http://url")
mock_download.assert_called_with(
mock_creds,
"http://url",
self.headers,
"c:\\dir",
500,
False,
f_name="output.zip",
block_size=4096,
callback=None,
)
self.assertTrue(val.success)
val = _api.get_output(
"c:\\dir", 500, "output.zip", 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="output.zip",
block_size=1,
callback=_callback,
)
self.assertTrue(val.success)
val = _api.get_output("c:\\dir", 500, "output.zip", False, job_id="test_id")
mock_download.assert_called_with(
mock_creds,
"https://test.com/test_id/output",
self.headers,
"c:\\dir",
500,
False,
f_name="output.zip",
block_size=4096,
callback=None,
)
self.assertTrue(val.success)
val = _api.get_output("c:\\dir", 500, "output.zip", False, job_id="test_id", otype="test")
self.assertFalse(val.success)
def test_api_get_output(self, mock_url, mock_creds, mock_config,
mock_download):
"""Test get_output"""
_callback = mock.Mock()
_api = BatchAppsApi(mock_creds, mock_config)
mock_url.return_value = "https://test.com/{jobid}/{type}"
mock_download.return_value = {}
val = _api.get_output("c:\\dir", 500, "output.zip", False)
self.assertFalse(mock_download.called)
self.assertFalse(val.success)
val = _api.get_output("c:\\dir",
500,
"output.zip",
False,
url="http://url")
mock_download.assert_called_with(mock_creds,
"http://url",
self.headers,
"c:\\dir",
500,
False,
f_name="output.zip",
block_size=4096,
callback=None)
self.assertTrue(val.success)
val = _api.get_output("c:\\dir",
500,
"output.zip",
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="output.zip",
block_size=1,
callback=_callback)
self.assertTrue(val.success)
val = _api.get_output("c:\\dir",
500,
"output.zip",
False,
job_id="test_id")
mock_download.assert_called_with(mock_creds,
"https://test.com/test_id/output",
self.headers,
"c:\\dir",
500,
False,
f_name="output.zip",
block_size=4096,
callback=None)
self.assertTrue(val.success)
val = _api.get_output("c:\\dir",
500,
"output.zip",
False,
job_id="test_id",
otype="test")
self.assertFalse(val.success)
