def setUpClass(cls): # Create zipfile for tests to read tmpdir = test_utils.mkdtemp() zipfile_name = os.path.join(tmpdir, '_support_test_sample.par') z = zipfile.ZipFile(zipfile_name, 'w') entry_name = '_support_test_helper_script.py' entry_data = b'print("Hello world")' z.writestr(entry_name, entry_data) z.close() cls.zipfile_name = zipfile_name cls.entry_name = entry_name cls.entry_data = entry_data # Create mock loader object class MockLoader(object): pass mock_loader = MockLoader() mock_loader.archive = zipfile_name mock_loader.prefix = '' main = sys.modules.get('__main__') old_loader = getattr(main, '__loader__', None) main.__loader__ = mock_loader cls.mock_loader = mock_loader cls.old_loader = old_loader
def setUpClass(cls): # Create zipfile for tests to read tmpdir = test_utils.mkdtemp() zipfile_name = os.path.join(tmpdir, '_support_test_sample.par') z = zipfile.ZipFile(zipfile_name, 'w') entry_name = '_support_test_helper_script.py' entry_data = b'print("Hello world")' z.writestr(entry_name, entry_data) z.close() cls.zipfile_name = zipfile_name cls.entry_name = entry_name cls.entry_data = entry_data # Create mock loader object class MockLoader(object): pass mock_loader = MockLoader() mock_loader.archive = zipfile_name mock_loader.prefix = '' main = sys.modules.get('__main__') old_loader = getattr(main, '__loader__', None) main.__loader__ = mock_loader cls.mock_loader = mock_loader cls.old_loader = old_loader
def test_remove_if_present(self): tmpdir = test_utils.mkdtemp() filename = os.path.join(tmpdir, 'afile') with open(filename, 'wb') as f: f.write(b'dontcare') # File exists self.assertTrue(os.path.exists(filename)) python_archive.remove_if_present(filename) self.assertFalse(os.path.exists(filename)) # File doesn't exist python_archive.remove_if_present(filename) self.assertFalse(os.path.exists(filename))
def test_remove_if_present(self): tmpdir = test_utils.mkdtemp() filename = os.path.join(tmpdir, 'afile') with open(filename, 'wb') as f: f.write(b'dontcare') # File exists self.assertTrue(os.path.exists(filename)) python_archive.remove_if_present(filename) self.assertFalse(os.path.exists(filename)) # File doesn't exist python_archive.remove_if_present(filename) self.assertFalse(os.path.exists(filename))
def _write_and_check(self, resource, name, expected_content): # Write zipfile tmpdir = test_utils.mkdtemp() zipfile_name = os.path.join(tmpdir, 'baz.zip') z = zipfile.ZipFile(zipfile_name, 'w') resource.store(z) z.close() # Read and validate zipfile z = zipfile.ZipFile(zipfile_name, 'r') self.assertEqual(z.namelist(), [name]) with z.open(name) as infile: actual_content = infile.read() self.assertEqual(expected_content, actual_content) z.close()
def _write_and_check(self, resource, name, expected_content): # Write zipfile tmpdir = test_utils.mkdtemp() zipfile_name = os.path.join(tmpdir, 'baz.zip') z = zipfile.ZipFile(zipfile_name, 'w') resource.store(z) z.close() # Read and validate zipfile z = zipfile.ZipFile(zipfile_name, 'r') self.assertEqual(z.namelist(), [name]) with z.open(name) as infile: actual_content = infile.read() self.assertEqual(expected_content, actual_content) z.close()
def setUp(self): # Setup directory structure and files self.tmpdir = test_utils.mkdtemp() self.input_dir = os.path.join(self.tmpdir, 'input') if not os.path.exists(self.input_dir): os.makedirs(self.input_dir) self.manifest_filename = os.path.join(self.input_dir, 'manifest') self.main_file = test_utils.temp_file(b'print("Hello World!")', suffix='.py') manifest_content = '%s %s\n' % ( os.path.basename(self.main_file.name), self.main_file.name) self.manifest_file = test_utils.temp_file( manifest_content.encode('utf8')) self.output_dir = os.path.join(self.tmpdir, 'output') if not os.path.exists(self.output_dir): os.makedirs(self.output_dir) self.output_filename = os.path.join(self.output_dir, 'output.par') self.interpreter = '/usr/bin/python2' self.import_roots = []
def setUp(self): # Setup directory structure and files self.tmpdir = test_utils.mkdtemp() self.input_dir = os.path.join(self.tmpdir, 'input') if not os.path.exists(self.input_dir): os.makedirs(self.input_dir) self.manifest_filename = os.path.join(self.input_dir, 'manifest') self.main_file = test_utils.temp_file(b'print("Hello World!")', suffix='.py') manifest_content = '%s %s\n' % (os.path.basename( self.main_file.name), self.main_file.name) self.manifest_file = test_utils.temp_file( manifest_content.encode('utf8')) self.output_dir = os.path.join(self.tmpdir, 'output') if not os.path.exists(self.output_dir): os.makedirs(self.output_dir) self.output_filename = os.path.join(self.output_dir, 'output.par') self.interpreter = '/usr/bin/python2' self.import_roots = []
def setUp(self): # Setup directory structure and files self.tmpdir = test_utils.mkdtemp() self.input_dir = os.path.join(self.tmpdir, 'input') if not os.path.exists(self.input_dir): os.makedirs(self.input_dir) self.manifest_filename = os.path.join(self.input_dir, 'manifest') self.main_file = test_utils.temp_file(b'print("Hello World!")', suffix='.py') manifest_content = '%s %s\n' % (os.path.basename( self.main_file.name), self.main_file.name) self.manifest_file = test_utils.temp_file( manifest_content.encode('utf8')) self.output_dir = os.path.join(self.tmpdir, 'output') if not os.path.exists(self.output_dir): os.makedirs(self.output_dir) self.output_filename = os.path.join(self.output_dir, 'output.par') self.interpreter = sys.executable self.import_roots = [] self.date_time_tuple = (1980, 1, 1, 0, 0, 0) self.timestamp = 315532800 self.zip_safe = True
def setUp(self): # Setup directory structure and files self.tmpdir = test_utils.mkdtemp() self.input_dir = os.path.join(self.tmpdir, 'input') if not os.path.exists(self.input_dir): os.makedirs(self.input_dir) self.manifest_filename = os.path.join(self.input_dir, 'manifest') self.main_file = test_utils.temp_file(b'print("Hello World!")', suffix='.py') manifest_content = '%s %s\n' % ( os.path.basename(self.main_file.name), self.main_file.name) self.manifest_file = test_utils.temp_file( manifest_content.encode('utf8')) self.output_dir = os.path.join(self.tmpdir, 'output') if not os.path.exists(self.output_dir): os.makedirs(self.output_dir) self.output_filename = os.path.join(self.output_dir, 'output.par') self.interpreter = sys.executable self.import_roots = [] self.date_time_tuple = (1980, 1, 1, 0, 0, 0) self.timestamp = 315532800 self.zip_safe = True