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')