2
0

test_bundle.py 1.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051
  1. # test_bundle.py -- tests for bundle
  2. # Copyright (C) 2020 Jelmer Vernooij <jelmer@jelmer.uk>
  3. #
  4. # Dulwich is dual-licensed under the Apache License, Version 2.0 and the GNU
  5. # General Public License as public by the Free Software Foundation; version 2.0
  6. # or (at your option) any later version. You can redistribute it and/or
  7. # modify it under the terms of either of these two licenses.
  8. #
  9. # Unless required by applicable law or agreed to in writing, software
  10. # distributed under the License is distributed on an "AS IS" BASIS,
  11. # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12. # See the License for the specific language governing permissions and
  13. # limitations under the License.
  14. #
  15. # You should have received a copy of the licenses; if not, see
  16. # <http://www.gnu.org/licenses/> for a copy of the GNU General Public License
  17. # and <http://www.apache.org/licenses/LICENSE-2.0> for a copy of the Apache
  18. # License, Version 2.0.
  19. #
  20. """Tests for bundle support."""
  21. import os
  22. import tempfile
  23. from io import BytesIO
  24. from dulwich.bundle import Bundle, read_bundle, write_bundle
  25. from dulwich.pack import PackData, write_pack_objects
  26. from . import TestCase
  27. class BundleTests(TestCase):
  28. def test_roundtrip_bundle(self):
  29. origbundle = Bundle()
  30. origbundle.version = 3
  31. origbundle.capabilities = {"foo": None}
  32. origbundle.references = {b"refs/heads/master": b"ab" * 20}
  33. origbundle.prerequisites = [(b"cc" * 20, "comment")]
  34. b = BytesIO()
  35. write_pack_objects(b.write, [])
  36. b.seek(0)
  37. origbundle.pack_data = PackData.from_file(b)
  38. with tempfile.TemporaryDirectory() as td:
  39. with open(os.path.join(td, "foo"), "wb") as f:
  40. write_bundle(f, origbundle)
  41. with open(os.path.join(td, "foo"), "rb") as f:
  42. newbundle = read_bundle(f)
  43. self.assertEqual(origbundle, newbundle)