blob: 89d4156743a7d605f329e3b16646511135dd8300 [file] [log] [blame]
#!/usr/bin/python
# Copyright (c) 2014 The Native Client Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
"""Fake downloader which can be used in place of http_download."""
import os
import shutil
import sys
import urllib
class FakeDownloader(object):
"""Testing replacement for http_download that copies files."""
def __init__(self, copy_func=None):
self._urls = {}
if copy_func is None:
self._copy_func = shutil.copyfile
else:
self._copy_func = copy_func
self._download_count = 0
def StoreURL(self, url, filepath):
"""Stores a known url into the downloader to download."""
self._urls[url] = filepath
def Download(self, url, target, username=None, verbose=True, password=None,
logger=None):
if url not in self._urls:
raise urllib.HTTPError(url, 404,
'Fake Downloader retrieved invalid URL',
[], None)
self._copy_func(self._urls[url], target)
self._download_count += 1
def GetDownloadCount(self):
return self._download_count