upstream/mercurial-mirror Files · contrib/python-zstandard/tests/test_roundtrip.py

bundle1: fix bundle1-denied reporting for push over ssh...

bundle1: fix bundle1-denied reporting for push over ssh Changeset introduced a config option to have the server deny push using bundle1. The original protocol has not really be design to allow such kind of error reporting so some hack was used. It turned the hack only works on HTTP and that ssh wire peer hangs forever when the same hack is used. After further digging, there is no way to report the error in a unified way. Using 'ooberror' freeze ssh and raising 'Abort' makes HTTP return a HTTP500 without further details. So with sadness we implement a version that dispatch according to the protocol used. We also add a test for pushing over ssh to make sure we won't regress in the future. That test show that the hint is missing, this is another bug fixed in the next changeset.

Gregory Szorc - - Load All Authors

File last commit:

r30895:c32454d6 default


                r30909:d554e624

stable

Download file

             test_roundtrip.py
        
                    64 lines
            
             | 2.1 KiB
            
                | text/x-python
            
             |
                PythonLexer
            
             / contrib / python-zstandard / tests / test_roundtrip.py
          
                    History
                
                 |
                  Source
                 | Raw
                 |Copy content
                 |Copy permalink

        Gregory Szorc
    
zstd: vendor python-zstandard 0.5.0...

              r30435
            
      import io

      try:

          import unittest2 as unittest

      except ImportError:

          import unittest

      try:

          import hypothesis

          import hypothesis.strategies as strategies

      except ImportError:

          raise unittest.SkipTest('hypothesis not available')

      import zstd

      compression_levels = strategies.integers(min_value=1, max_value=22)

      class TestRoundTrip(unittest.TestCase):

          @hypothesis.given(strategies.binary(), compression_levels)

          def test_compress_write_to(self, data, level):

              """Random data from compress() roundtrips via write_to."""

              cctx = zstd.ZstdCompressor(level=level)

              compressed = cctx.compress(data)

              buffer = io.BytesIO()

              dctx = zstd.ZstdDecompressor()

              with dctx.write_to(buffer) as decompressor:

                  decompressor.write(compressed)

              self.assertEqual(buffer.getvalue(), data)

          @hypothesis.given(strategies.binary(), compression_levels)

          def test_compressor_write_to_decompressor_write_to(self, data, level):

              """Random data from compressor write_to roundtrips via write_to."""

              compress_buffer = io.BytesIO()

              decompressed_buffer = io.BytesIO()

              cctx = zstd.ZstdCompressor(level=level)

              with cctx.write_to(compress_buffer) as compressor:

                  compressor.write(data)

              dctx = zstd.ZstdDecompressor()

              with dctx.write_to(decompressed_buffer) as decompressor:

                  decompressor.write(compress_buffer.getvalue())

              self.assertEqual(decompressed_buffer.getvalue(), data)

          @hypothesis.given(strategies.binary(average_size=1048576))

          @hypothesis.settings(perform_health_check=False)

          def test_compressor_write_to_decompressor_write_to_larger(self, data):

              compress_buffer = io.BytesIO()

              decompressed_buffer = io.BytesIO()

              cctx = zstd.ZstdCompressor(level=5)

              with cctx.write_to(compress_buffer) as compressor:

                  compressor.write(data)

              dctx = zstd.ZstdDecompressor()

              with dctx.write_to(decompressed_buffer) as decompressor:

                  decompressor.write(compress_buffer.getvalue())

              self.assertEqual(decompressed_buffer.getvalue(), data)

	Site-wide shortcuts
/	Use quick search box
g h	Goto home page
g g	Goto my private gists page
g G	Goto my public gists page
g 0-9	Goto bookmarked items from 0-9
n r	New repository page
n g	New gist page

	Repositories
g s	Goto summary page
g c	Goto changelog page
g f	Goto files page
g F	Goto files page with file search activated
g p	Goto pull requests page
g o	Goto repository settings
g O	Goto repository access permissions settings
t s	Toggle sidebar on some pages

Gregory Szorc zstd: vendor python-zstandard 0.5.0...	r30435	import io

		try:
		import unittest2 as unittest
		except ImportError:
		import unittest

		try:
		import hypothesis
		import hypothesis.strategies as strategies
		except ImportError:
		raise unittest.SkipTest('hypothesis not available')

		import zstd


		compression_levels = strategies.integers(min_value=1, max_value=22)


		class TestRoundTrip(unittest.TestCase):
		@hypothesis.given(strategies.binary(), compression_levels)
		def test_compress_write_to(self, data, level):
		"""Random data from compress() roundtrips via write_to."""
		cctx = zstd.ZstdCompressor(level=level)
		compressed = cctx.compress(data)

		buffer = io.BytesIO()
		dctx = zstd.ZstdDecompressor()
		with dctx.write_to(buffer) as decompressor:
		decompressor.write(compressed)

		self.assertEqual(buffer.getvalue(), data)

		@hypothesis.given(strategies.binary(), compression_levels)
		def test_compressor_write_to_decompressor_write_to(self, data, level):
		"""Random data from compressor write_to roundtrips via write_to."""
		compress_buffer = io.BytesIO()
		decompressed_buffer = io.BytesIO()

		cctx = zstd.ZstdCompressor(level=level)
		with cctx.write_to(compress_buffer) as compressor:
		compressor.write(data)

		dctx = zstd.ZstdDecompressor()
		with dctx.write_to(decompressed_buffer) as decompressor:
		decompressor.write(compress_buffer.getvalue())

		self.assertEqual(decompressed_buffer.getvalue(), data)

		@hypothesis.given(strategies.binary(average_size=1048576))
		@hypothesis.settings(perform_health_check=False)
		def test_compressor_write_to_decompressor_write_to_larger(self, data):
		compress_buffer = io.BytesIO()
		decompressed_buffer = io.BytesIO()

		cctx = zstd.ZstdCompressor(level=5)
		with cctx.write_to(compress_buffer) as compressor:
		compressor.write(data)

		dctx = zstd.ZstdDecompressor()
		with dctx.write_to(decompressed_buffer) as decompressor:
		decompressor.write(compress_buffer.getvalue())

		self.assertEqual(decompressed_buffer.getvalue(), data)