python unittest howto

Question

I`d like to know how I could unit-test the following module.

def download_distribution(url, tempdir):
    \"\"\" Method which downloads the distribution from         


        

Answer 1

To mock urllopen you can pre fetch some examples that you can then use in your unittests. Here's an example to get you started:

def urlopen(url):
    urlclean = url[:url.find('?')] # ignore GET parameters
    files = {
        'http://example.com/foo.xml': 'foo.xml',
        'http://example.com/bar.xml': 'bar.xml',
    }
    return file(files[urlclean])
yourmodule.urllib.urlopen = urlopen