def test_download_asset_failure(self, mock_do_pivnet_download, mock_util):
mock_do_pivnet_download.return_value = "download failed", "", 1
out, err, exit_code = download_and_import.download_assets(
self.settings, '/home/ubuntu/tiles/')
self.assertEqual(mock_do_pivnet_download.call_count, 1)
self.assertEqual(out, "download failed")
self.assertEqual(exit_code, 1)
def test_download_asset_success(self, mock_do_pivnet_download, mock_util):
mock_do_pivnet_download.return_value = "", "", 0
out, err, exit_code = download_and_import.download_assets(
self.settings, '/home/ubuntu/tiles/')
self.assertEqual(mock_do_pivnet_download.call_count, 2)
self.assertEqual(exit_code, 0)
self.assertEqual(mock_do_pivnet_download.mock_calls[0][1][0],
'stemcells')
self.assertEqual(mock_do_pivnet_download.mock_calls[1][1][0], 'cf')