def test_files_sizes(self, test_case):
"""
Ensures that the files created are the correct size.
"""
test_dir = file_utilities.create_test_directory(test_case)
try:
for i in test_case:
if isinstance(i, tuple):
sys.stdout.write("checking size of %s\n" % (i[0], ))
file_size = os.stat(os.path.join(test_dir, i[0])).st_size
assert file_size == i[1]
finally:
shutil.rmtree(test_dir)
def test_files_sizes(self, test_case):
"""
Ensures that the files created are the correct size.
"""
test_dir = file_utilities.create_test_directory(test_case)
try:
for i in test_case:
if isinstance(i, tuple):
sys.stdout.write("checking size of %s\n" % (i[0], ))
file_size = os.stat(
os.path.join(test_dir, i[0])).st_size
assert file_size == i[1]
finally:
shutil.rmtree(test_dir)
def test_existence(self, test_case):
"""
Ensure that all the files that were supposed to get created are there,
and ensure that none were created that weren't supposed to be.
"""
test_dir = file_utilities.create_test_directory(test_case)
try:
real_contents = file_utilities.get_files(test_dir)
expected_contents = \
[(i[0] if isinstance(i, tuple) else i) for i in test_case]
sys.stdout.write("real_contents = %s\n" % (real_contents, ))
sys.stdout.write(
"expected_contents = %s\n" % (expected_contents, ))
assert set(real_contents) == set(expected_contents)
finally:
shutil.rmtree(test_dir)
def zip_tree(self, request, test_case):
try:
# We must create these variables up front for our finally
# statement to work correctly.
test_dir = zip_file = unzip_dir = None
# Create some files and directories to zip up
test_dir = file_utilities.create_test_directory(test_case)
# Get a temporary file that will become our zip file
zip_file_handle = tempfile.NamedTemporaryFile(delete = False)
zip_file_handle.close()
zip_file = zip_file_handle.name
# Zip up our directory tree
zipdir.zip_directory(test_dir, zip_file)
# Unzip our directory tree
unzip_dir = tempfile.mkdtemp()
with closing(zipfile.ZipFile(zip_file, "r")) as f:
f.extractall(unzip_dir)
except:
if test_dir is not None:
shutil.rmtree(test_dir)
if zip_file is not None:
os.remove(zip_file)
if unzip_dir is not None:
shutil.rmtree(unzip_dir)
raise
def cleanup():
shutil.rmtree(test_dir)
os.remove(zip_file)
shutil.rmtree(unzip_dir)
request.addfinalizer(cleanup)
return test_dir, zip_file, unzip_dir
