mirror of
https://github.com/Cronocide/wifipumpkin3.git
synced 2025-01-23 03:39:18 +00:00
34 lines
1.1 KiB
Python
34 lines
1.1 KiB
Python
|
import unittest
|
||
|
|
from wifipumpkin3.core.common.platforms import Linux
|
||
|
|
import wifipumpkin3.core.utility.constants as C
|
||
|
|
from wifipumpkin3.core.utility.collection import SettingsINI
|
||
|
|
import tempfile
|
||
|
|
import requests
|
||
|
|
import os
|
||
|
|
from os import path
|
||
|
|
from zipfile import ZipFile
|
||
|
|
|
||
|
|
|
||
|
|
class TestDownloadCaptiveFlaskTemplates(unittest.TestCase):
|
||
|
|
def setUp(self):
|
||
|
|
url = "https://github.com/mh4x0f/extra-captiveflask/archive/master.zip"
|
||
|
|
save_path = tempfile.gettempdir() + "/master.zip"
|
||
|
|
chunk_size = 128
|
||
|
|
r = requests.get(url, stream=True)
|
||
|
|
with open(save_path, "wb") as fd:
|
||
|
|
for chunk in r.iter_content(chunk_size=chunk_size):
|
||
|
|
fd.write(chunk)
|
||
|
|
self.assertTrue(path.isfile(save_path))
|
||
|
|
|
||
|
|
def test_unzip_file(self):
|
||
|
|
path_to_zip_file = tempfile.gettempdir() + "/master.zip"
|
||
|
|
with ZipFile(path_to_zip_file, "r") as zip_ref:
|
||
|
|
zip_ref.extractall(tempfile.gettempdir())
|
||
|
|
extracted_filepath = tempfile.gettempdir() + "/extra-captiveflask-master"
|
||
|
|
|
||
|
|
self.assertTrue(path.isdir(extracted_filepath))
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
unittest.main()
|