Create download.py

VenkateshDoijode · web-flow · commit 6b1808091cdd · 2018-06-20T14:42:54.000+05:30
diff --git a/Python_basics/SeleniumWebDriver_Advanced/download.py b/Python_basics/SeleniumWebDriver_Advanced/download.py
@@ -0,0 +1,42 @@
+import os
+import time
+import shutil
+import tempfile
+import unittest
+from selenium import webdriver
+
+
+class Download(unittest.TestCase):
+
+    def setUp(self):
+        self.download_dir = tempfile.mkdtemp()
+
+        profile = webdriver.FirefoxProfile()
+        profile.set_preference("browser.download.dir", self.download_dir)
+        profile.set_preference("browser.download.folderList", 2)
+        profile.set_preference(
+            "browser.helperApps.neverAsk.saveToDisk",
+            "images/jpeg, application/pdf, application/octet-stream")
+        profile.set_preference("pdfjs.disabled", True)
+        self.driver = webdriver.Firefox(firefox_profile=profile)
+
+    def tearDown(self):
+        self.driver.quit()
+        shutil.rmtree(self.download_dir)
+
+    def test_example_1(self):
+        driver = self.driver
+        driver.get('http://the-internet.herokuapp.com/download')
+        download_link = driver.find_element_by_css_selector('.example a')
+        download_link.click()
+
+        time.sleep(1.0)  # necessary for slow download speeds
+
+        files = os.listdir(self.download_dir)
+        files = [os.path.join(self.download_dir, f)
+                 for f in files]  # add directory to each filename
+        assert len(files) > 0, "no files were downloaded"
+        assert os.path.getsize(files[0]) > 0, "downloaded file was empty"
+
+if __name__ == "__main__":
+    unittest.main()
