def test_save_into_load_from_fd(self): other = copy.deepcopy(self.dict_file) fnm = create_temp_file() with open(fnm, "w") as fd: self.dict_file.save_into_fd(fd) with open(fnm, "r") as fd: other.load_from_fd(fd, erase_first=True) self.assertTrue(self.dict_file.is_equal(other)) os.remove(fnm)
def test_save_into_load_from_fd(self): other = copy.deepcopy(self.dict_file) fnm = create_temp_file() fd = open(fnm, "w") self.dict_file.save_into_fd(fd) fd.close() fd = open(fnm, "r") other.load_from_fd(fd, erase_first=True) fd.close() self.assertTrue(self.dict_file.is_equal(other)) os.remove(fnm)
def setUp(self): self.extract_dir = tempfile.mkdtemp() self.working_dir = tempfile.mkdtemp() self.number_of_files = 5 self.files = [] self.strings = { "string1": "Why did the chicken cross the road?", "string2": "To get to the other side." } files_created = 0 while not (files_created == self.number_of_files): path = create_temp_file(file_dir=self.working_dir) md5sum = extract_md5(path) self.files.append({"path": path, "md5sum": md5sum}) files_created += 1
def extract_archive_blob(self, blob): # handle the tarball temp_path = create_temp_file(file_dir=self.working_dir, write_path_to_file=False) temp_file = open(temp_path, 'w') temp_file.write(blob) temp_file.seek(0) temp_file.close() shutil.move(temp_path, "%s.tar.gz" % temp_path) tarball = GlideinTar() self.assertTrue(tarball.is_tarfile("%s.tar.gz" % temp_path), "Blob tarball fails tarball.is_tarfile test") self.extract_archive_file("%s.tar.gz" % temp_path)
def create_files(self, number_of_files, suffix=""): """ Create temporary files using the tempfile module. The absolute path toThe file extension to place on the temporary file the file is written to the file for content. @type number_of_files: int @param number_of_files: The number of temporary files to create @type file_suffix: string @param file_suffix: The file extension to place on the temporary file """ files_created = 0 while not (files_created == number_of_files): path = create_temp_file(file_suffix=suffix, file_dir=self.cleanup_dir) files_created += 1
def extract_archive_blob(self, blob): # handle the tarball temp_path = create_temp_file(file_dir=self.working_dir, write_path_to_file=False) # TODO #23166: Use context managers[with statement] when python 3 # once we get rid of SL6 and tarballs temp_file = open(temp_path, 'w') temp_file.write(blob) temp_file.seek(0) temp_file.close() shutil.move(temp_path, "%s.tar.gz" % temp_path) tarball = GlideinTar() self.assertTrue(tarball.is_tarfile("%s.tar.gz" % temp_path), "Blob tarball fails tarball.is_tarfile test") self.extract_archive_file("%s.tar.gz" % temp_path)
def create_files(self, number_of_files, suffix=""): """ Create temporary files using the tempfile module. The absolute path toThe file extension to place on the temporary file the file is written to the file for content. @type number_of_files: int @param number_of_files: The number of temporary files to create @type file_suffix: string @param file_suffix: The file extension to place on the temporary file """ files_created = 0 while not files_created == number_of_files: path = create_temp_file( file_suffix=suffix, file_dir=self.cleanup_dir) files_created += 1 self.assertTrue(len(path) > 0)
def test_save(self): fnm = create_temp_file() os.remove(fnm) self.dict_file.save(fname=fnm, save_only_if_changed=False) self.assertTrue(os.path.exists(fnm)) os.remove(fnm)
def test_save(self): fnm = create_temp_file() os.remove(fnm) self.dict_file.save(fname=fnm, save_only_if_changed=False) self.assertTrue(os.path.exists(fnm)) os.remove(fnm)
import select import os import unittest2 as unittest from glideinwms.unittests.unittest_utils import runTest from glideinwms.unittests.unittest_utils import FakeLogger from glideinwms.unittests.unittest_utils import create_temp_file from glideinwms.lib.fork import ForkResultError from glideinwms.lib.fork import fork_in_bg from glideinwms.lib.fork import fetch_fork_result from glideinwms.lib.fork import fetch_fork_result_list from glideinwms.lib.fork import fetch_ready_fork_result_list from glideinwms.lib.fork import wait_for_pids from glideinwms.lib.fork import ForkManager import glideinwms.lib.logSupport LOG_FILE = create_temp_file() def global_log_setup(): fd = open(LOG_FILE, 'w', 0) glideinwms.lib.logSupport.log = FakeLogger(fd) def global_log_cleanup(): if os.path.exists(LOG_FILE): os.remove(LOG_FILE) def sleep_fn(sleep_tm=None): if not sleep_tm: sleep_tm = 1
import xmlrunner import platform import unittest2 as unittest from glideinwms.unittests.unittest_utils import FakeLogger from glideinwms.unittests.unittest_utils import create_temp_file from glideinwms.lib.fork import ForkResultError from glideinwms.lib.fork import fork_in_bg from glideinwms.lib.fork import fetch_fork_result from glideinwms.lib.fork import fetch_fork_result_list from glideinwms.lib.fork import fetch_ready_fork_result_list from glideinwms.lib.fork import wait_for_pids from glideinwms.lib.fork import ForkManager import glideinwms.lib.logSupport LOG_FILE = create_temp_file() def global_log_setup(): fd = open(LOG_FILE, 'w', 0) glideinwms.lib.logSupport.log = FakeLogger(fd) def global_log_cleanup(): if os.path.exists(LOG_FILE): os.remove(LOG_FILE) def sleep_fn(sleep_tm=None): if not sleep_tm: sleep_tm = 1