def test_will_download_file(self): test_url = 'http://www.google.com/images/logos/ps_logo2.png' tempdir = tempfile.mkdtemp() download_url_to(test_url, tempdir) self.assertTrue(os.path.isfile(os.path.join( tempdir, 'ps_logo2.png')))
def get_selenium(url=JAR_DOWNLOAD_URL): """ If needed, will grab the Selenium 2 jar file from the Internet and save it locally. It then updates the config file with the new location of the jar. """ try: path = config.get('selenium', 'jar_filename') if os.path.isfile(path): # We've already fetched Selenium, and the config has been updated return except ConfigParser.Error: # Selenium has not been fetched and the config has not been updated pass path = download_url_to(url, tempfile.mkdtemp()) try: config.add_section('selenium') except ConfigParser.DuplicateSectionError: # We already have the section, that's ok pass config.set('selenium', 'jar_filename', path) save_config(config)