def test_multiple(self):
     self.assertEqual(binary_replace(b'aaaacaaaa\x00', b'aaaa', b'bbbb'),
                      b'bbbbcbbbb\x00')
     self.assertEqual(binary_replace(b'aaaacaaaa\x00', b'aaaa', b'bbb'),
                      b'bbbcbbb\x00\x00\x00')
     self.assertRaises(_PaddingError, binary_replace, b'aaaacaaaa\x00',
                       b'aaaa', b'bbbbb')
Exemple #2
0
 def test_multiple(self):
     self.assertEqual(
         binary_replace(b'aaaacaaaa\x00', b'aaaa', b'bbbb'),
         b'bbbbcbbbb\x00')
     self.assertEqual(
         binary_replace(b'aaaacaaaa\x00', b'aaaa', b'bbb'),
         b'bbbcbbb\x00\x00\x00')
     self.assertRaises(_PaddingError, binary_replace,
                       b'aaaacaaaa\x00', b'aaaa', b'bbbbb')
Exemple #3
0
 def test_simple(self):
     for encoding in [
             'utf-8', 'utf-16-le', 'utf-16-be', 'utf-32-le', 'utf-32-be'
     ]:
         a = 'aaaaa'.encode(encoding)
         b = 'bbbb'.encode(encoding)
         data = 'xxxaaaaaxyz\0zz'.encode(encoding)
         result = 'xxxbbbbxyz\0\0zz'.encode(encoding)
         self.assertEqual(binary_replace(data, a, b, encoding=encoding),
                          result)
    def test_windows_entry_point(self):
        """
        This emulates pip-created entry point executables on windows.  For more info,
        refer to conda/install.py::replace_entry_point_shebang
        """
        tmp_dir = tempfile.mkdtemp()
        cwd = getcwd()
        chdir(tmp_dir)
        original_prefix = "C:\\BogusPrefix\\python.exe"
        try:
            url = 'https://s3.amazonaws.com/conda-dev/pyzzerw.pyz'
            download(url, 'pyzzerw.pyz')
            url = 'https://files.pythonhosted.org/packages/source/c/conda/conda-4.1.6.tar.gz'
            download(url, 'conda-4.1.6.tar.gz')
            subprocess.check_call(
                [
                    sys.executable,
                    'pyzzerw.pyz',
                    # output file
                    '-o',
                    'conda.exe',
                    # entry point
                    '-m',
                    'conda.cli.main:main',
                    # initial shebang
                    '-s',
                    '#! ' + original_prefix,
                    # launcher executable to use (32-bit text should be compatible)
                    '-l',
                    't32',
                    # source archive to turn into executable
                    'conda-4.1.6.tar.gz',
                ],
                cwd=tmp_dir)
            # this is the actual test: change the embedded prefix and make sure that the exe runs.
            data = open('conda.exe', 'rb').read()
            fixed_data = binary_replace(data, original_prefix, sys.executable)
            with open("conda.fixed.exe", 'wb') as f:
                f.write(fixed_data)
            # without a valid shebang in the exe, this should fail
            with pytest.raises(subprocess.CalledProcessError):
                subprocess.check_call(['conda.exe', '-h'])

            process = subprocess.Popen(['conda.fixed.exe', '-h'],
                                       stdout=subprocess.PIPE,
                                       stderr=subprocess.PIPE)
            output, error = process.communicate()
            output = output.decode('utf-8')
            error = error.decode('utf-8')
            assert ("conda is a tool for managing and deploying applications, "
                    "environments and packages.") in output
        except:
            raise
        finally:
            chdir(cwd)
Exemple #5
0
    def test_windows_entry_point(self):
        """
        This emulates pip-created entry point executables on windows.  For more info,
        refer to conda/install.py::replace_entry_point_shebang
        """
        tmp_dir = tempfile.mkdtemp()
        cwd = getcwd()
        chdir(tmp_dir)
        original_prefix = "C:\\BogusPrefix\\python.exe"
        try:
            url = 'https://s3.amazonaws.com/conda-dev/pyzzerw.pyz'
            download(url, 'pyzzerw.pyz')
            url = 'https://files.pythonhosted.org/packages/source/c/conda/conda-4.1.6.tar.gz'
            download(url, 'conda-4.1.6.tar.gz')
            subprocess.check_call([sys.executable, 'pyzzerw.pyz',
                                   # output file
                                   '-o', 'conda.exe',
                                   # entry point
                                   '-m', 'conda.cli.main:main',
                                   # initial shebang
                                   '-s', '#! ' + original_prefix,
                                   # launcher executable to use (32-bit text should be compatible)
                                   '-l', 't32',
                                   # source archive to turn into executable
                                   'conda-4.1.6.tar.gz',
                                   ],
                                  cwd=tmp_dir)
            # this is the actual test: change the embedded prefix and make sure that the exe runs.
            data = open('conda.exe', 'rb').read()
            fixed_data = binary_replace(data, original_prefix, sys.executable)
            with open("conda.fixed.exe", 'wb') as f:
                f.write(fixed_data)
            # without a valid shebang in the exe, this should fail
            with pytest.raises(subprocess.CalledProcessError):
                subprocess.check_call(['conda.exe', '-h'])

            process = subprocess.Popen(['conda.fixed.exe', '-h'],
                                       stdout=subprocess.PIPE,
                                       stderr=subprocess.PIPE)
            output, error = process.communicate()
            output = output.decode('utf-8')
            error = error.decode('utf-8')
            assert ("conda is a tool for managing and deploying applications, "
                    "environments and packages.") in output
        except:
            raise
        finally:
            chdir(cwd)
 def test_spaces(self):
     self.assertEqual(binary_replace(b' aaaa \x00', b'aaaa', b'bbbb'),
                      b' bbbb \x00')
 def test_two(self):
     self.assertEqual(
         binary_replace(b'aaaaa\x001234aaaaacc\x00\x00', b'aaaaa',
                        b'bbbbb'), b'bbbbb\x001234bbbbbcc\x00\x00')
 def test_no_extra(self):
     self.assertEqual(binary_replace(b'aaaaa\x00', b'aaaaa', b'bbbbb'),
                      b'bbbbb\x00')
 def test_shorter(self):
     self.assertEqual(
         binary_replace(b'xxxaaaaaxyz\x00zz', b'aaaaa', b'bbbb'),
         b'xxxbbbbxyz\x00\x00zz')
Exemple #10
0
 def test_spaces(self):
     self.assertEqual(
         binary_replace(b' aaaa \x00', b'aaaa', b'bbbb'),
         b' bbbb \x00')
Exemple #11
0
 def test_two(self):
     self.assertEqual(
         binary_replace(b'aaaaa\x001234aaaaacc\x00\x00', b'aaaaa',
                        b'bbbbb'),
         b'bbbbb\x001234bbbbbcc\x00\x00')
Exemple #12
0
 def test_no_extra(self):
     self.assertEqual(binary_replace(b'aaaaa\x00', b'aaaaa', b'bbbbb'),
                      b'bbbbb\x00')
Exemple #13
0
 def test_shorter(self):
     self.assertEqual(
         binary_replace(b'xxxaaaaaxyz\x00zz', b'aaaaa', b'bbbb'),
         b'xxxbbbbxyz\x00\x00zz')