summarylogtreecommitdiffstats
path: root/replace_wget.patch
blob: edf55ced45f64bd73f0a9ce8cf05cdf8387f0764 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
--- tests/test_pytest_localftpserver.py
+++ tests/test_pytest_localftpserver.py
@@ -12,7 +12,7 @@  Replace import
 import os
 
 import pytest
-import wget
+import urllib.request
 
 from pytest_localftpserver.plugin import PytestLocalFTPServer
 from pytest_localftpserver.servers import USE_PROCESS
@@ -156,14 +156,11 @@  Replace download code
         Contains urls to check
     """
     # checking files by url
-    download_dir = tmpdir.mkdir("download_url")
     for url in url_iterable:
         _, filename = os.path.split(os.path.relpath(url, base_url))
-        download_file = download_dir.join(filename)
-        wget.download(url, str(download_file))
-        with open(str(download_file)) as f:
-            assert f.read() == filename
-    download_dir.remove()
+        response = urllib.request.urlopen(url)
+        contents = response.read().decode("utf8")
+        assert contents == filename
 
 
 def check_get_file_contents(