Viewing file:  test_file_zip.py (5.83 KB) -rw-r--r--
Select action/file-type:
 (+) |  (+) |  (+) | Code (+) | Session (+) |  (+) | SDB (+) |  (+) |  (+) |  (+) |  (+) |  (+) |
from testfixtures import log_capture
from tests.base_test import BaseTest
from tests import config
from weevely.core.sessions import SessionURL
from weevely.core import modules
from weevely import utils
from weevely.core import messages
import subprocess
import os
def setUpModule():
subprocess.check_output("""
BASE_FOLDER="{config.base_folder}/test_file_zip/"
rm -rf "$BASE_FOLDER"
mkdir -p "$BASE_FOLDER/dir1/dir2/dir3/dir4"
echo -n 1 > "$BASE_FOLDER/dir1/f1"
echo -n 1 > "$BASE_FOLDER/dir1/dir2/f2"
echo -n 1 > "$BASE_FOLDER/dir1/dir2/dir3/f3"
echo -n 1 > "$BASE_FOLDER/dir1/dir2/dir3/dir4/f4"
cd "$BASE_FOLDER" && zip -r "test_0.zip" "dir1/"
echo -n 1 > "$BASE_FOLDER/f5"
rm -rf "$BASE_FOLDER/dir1"
chown www-data: -R "$BASE_FOLDER/"
""".format(
config = config
), shell=True)
class FileZip(BaseTest):
folders_rel = [
'test_file_zip/dir1',
'test_file_zip/dir1/dir2',
'test_file_zip/dir1/dir2/dir3',
'test_file_zip/dir1/dir2/dir3/dir4',
]
folders_abs = [
os.path.join(config.base_folder, f)
for f in folders_rel
]
files_rel = [
'test_file_zip/dir1/f1',
'test_file_zip/dir1/dir2/f2',
'test_file_zip/dir1/dir2/dir3/f3',
'test_file_zip/dir1/dir2/dir3/dir4/f4',
]
files_abs = [
os.path.join(config.base_folder, f)
for f in files_rel
]
zips_rel = [
'test_file_zip/test_0.zip'
]
zips_abs = [
os.path.join(config.base_folder, f)
for f in zips_rel
]
other_file_rel = 'test_file_zip/f5'
other_file_abs = os.path.join(config.base_folder, other_file_rel)
def setUp(self):
self.session = SessionURL(
self.url,
self.password,
volatile = True
)
modules.load_modules(self.session)
self.run_argv = modules.loaded['file_zip'].run_argv
def test_compress_decompress(self):
# Uncompress test.zip
self.assertTrue(self.run_argv(["--decompress", self.zips_rel[0], 'test_file_zip/' ]));
for file in self.files_abs:
self.assertEqual(subprocess.check_output("cat %s" % file, shell=True), b'1')
for folder in self.folders_abs:
subprocess.check_call('stat -c %%a "%s"' % folder, shell=True)
# Compress it again giving szipting folder
self.assertTrue(self.run_argv(['test_file_zip/test_1.zip', self.folders_rel[0]]));
self.zips_rel.append('test_file_zip/test_1.zip')
self.zips_abs.append(os.path.join(config.base_folder, self.zips_rel[-1]))
# Uncompress the new archive and recheck
self.assertTrue(self.run_argv(["--decompress", 'test_file_zip/test_1.zip', 'test_file_zip/']));
for file in self.files_abs:
self.assertEqual(subprocess.check_output("cat %s" % file, shell=True), b'1')
for folder in self.folders_abs:
subprocess.check_call('stat -c %%a "%s"' % folder, shell=True)
def test_compress_multiple(self):
# Uncompress test.zip
self.assertTrue(self.run_argv(["--decompress", self.zips_rel[0], 'test_file_zip/' ]));
for file in self.files_abs:
self.assertEqual(subprocess.check_output("cat %s" % file, shell=True), b'1')
for folder in self.folders_abs:
subprocess.check_call('stat -c %%a "%s"' % folder, shell=True)
# Create a new zip adding also other_file
self.assertTrue(self.run_argv(['test_file_zip/test_2.zip', self.folders_rel[0], self.other_file_rel]));
self.zips_rel.append('test_file_zip/test_2.zip')
self.zips_abs.append(os.path.join(config.base_folder, self.zips_rel[-1]))
# Remove all the files
subprocess.check_output("rm -rf %s" % self.folders_abs[0], shell=True)
subprocess.check_output("rm %s" % self.other_file_abs, shell=True)
# Uncompress the new archive and recheck
self.assertTrue(self.run_argv(["--decompress", 'test_file_zip/test_2.zip', 'test_file_zip/']));
for file in self.files_abs:
self.assertEqual(subprocess.check_output("cat %s" % file, shell=True), b'1')
for folder in self.folders_abs:
subprocess.check_call('stat -c %%a "%s"' % folder, shell=True)
# TODO: here skips the final check since f5 is misplaced
# on the archive
# Archive: /var/www/html/test_file_zip/test_2.zip
# testing: dir1/ OK
# testing: dir1/dir2/ OK
# testing: dir1/dir2/dir3/ OK
# testing: dir1/dir2/dir3/dir4/ OK
# testing: dir1/dir2/dir3/dir4/f4 OK
# testing: dir1/dir2/dir3/f3 OK
# testing: dir1/dir2/f2 OK
# testing: dir1/f1 OK
# testing: test_file_zip/f5 OK
#self.assertEqual(subprocess.check_output("cat %s" % self.other_file_abs, shell=True),'1')
@log_capture()
def test_already_exists(self, log_captured):
# Create a new zip with other_file, with the name test_0.zip
self.assertIsNone(self.run_argv(['test_file_zip/test_0.zip', self.other_file_rel]));
self.assertEqual(log_captured.records[-1].msg,
"File 'test_file_zip/test_0.zip' already exists, skipping compressing")
@log_capture()
def test_unexistant_decompress(self, log_captured):
self.assertIsNone(self.run_argv(["--decompress", 'bogus', '.']));
self.assertEqual(log_captured.records[-1].msg,
"Skipping file 'bogus', check existance and permission")
@log_capture()
def test_unexistant_compress(self, log_captured):
self.assertIsNone(self.run_argv(['bogus.zip', 'bogus']));
self.assertEqual(log_captured.records[-1].msg,
"File 'bogus.zip' not created, check existance and permission")
|