Fix tests

This commit is contained in:
2021-10-21 19:52:42 -04:00
parent 5e00ddccdb
commit ed9d148411
6 changed files with 32 additions and 9 deletions

View File

@@ -2,12 +2,16 @@ import os
from unittest import TestCase
from hexlib.web import download_file
import warnings
class TestDownloadFile(TestCase):
def setUp(self) -> None:
warnings.filterwarnings(action="ignore", category=ResourceWarning)
def test_download_file(self):
download_file("http://ovh.net/files/10Mb.dat", "/tmp/10Mb.dat")
download_file("https://github.com/simon987/hexlib/raw/master/10MB.bin", "/tmp/10Mb.dat")
self.assertTrue(os.path.exists("/tmp/10Mb.dat"))
os.remove("/tmp/10Mb.dat")
@@ -22,8 +26,8 @@ class TestDownloadFile(TestCase):
self.assertEqual(len(exceptions), 3)
def test_download_file_meta(self):
download_file("http://ovh.net/files/10Mb.dat", "/tmp/10Mb.dat", save_meta=True)
download_file("https://github.com/simon987/hexlib/raw/master/10MB.bin", "/tmp/10Mb.dat", save_meta=True)
self.assertTrue(os.path.exists("/tmp/10Mb.dat"))
self.assertTrue(os.path.exists("/tmp/10Mb.dat.meta"))
os.remove("/tmp/10Mb.dat")
# os.remove("/tmp/10Mb.dat.meta")
os.remove("/tmp/10Mb.dat.meta")