Added export_repository support and corresponding tests for hg client.

Signed-off-by: Leander Stephen D'Souza <leanderdsouza1234@gmail.com>

Author: leander-dsouza

test/test_hg.py

@@ -1,6 +1,7 @@
 """Integration tests for HgClient checkout functionality"""
 
 import os
+import subprocess
 import tempfile
 import unittest
 from shutil import which
@@ -61,5 +62,62 @@ def test_checkout_existing_directory(self):
         )
 
 
+@unittest.skipIf(not hg, '`hg` was not found')
+class TestExportRepository(unittest.TestCase):
+    """Integration tests for HgClient _export_repository functionality."""
+
+    def setUp(self):
+        """Set up test fixtures for each test"""
+        self.test_dir = tempfile.mkdtemp(prefix='hg_test_')
+        self.repo_path = os.path.join(self.test_dir, 'hello')
+        self.export_base_path = os.path.join(self.test_dir, 'exported_repo')
+        self.repo_url = 'https://www.mercurial-scm.org/repo/hello'
+
+        self.hg_client = HgClient(self.repo_path)
+
+        self._ensure_repo_cloned()
+
+    def tearDown(self):
+        """Clean up test fixtures"""
+        if os.path.exists(self.test_dir):
+            rmtree(self.test_dir)
+
+    def _ensure_repo_cloned(self):
+        """Ensure the test repository is cloned locally"""
+        if not os.path.exists(self.repo_path):
+            try:
+                subprocess.run(
+                    ['hg', 'clone', '--rev', 'tip', self.repo_url, self.repo_path],
+                    check=True,
+                    capture_output=True,
+                    text=True,
+                )
+            except subprocess.CalledProcessError as e:
+                self.fail(f'Failed to clone repository: {e.stderr}')
+
+    def test_export_repository_specific_revision(self):
+        """Test exporting a specific revision"""
+        result = self.hg_client._export_repository('1', self.export_base_path)
+        self.assertTrue(result, "Export should succeed for revision '1'")
+
+        # Verify files were created correctly
+        tar_gz_path = self.export_base_path + '.tar.gz'
+        self.assertTrue(os.path.exists(tar_gz_path))
+
+    def test_export_repository_invalid_revision(self):
+        """Test exporting with an invalid revision"""
+        invalid_revision = 'nonexistent123456789'
+
+        result = self.hg_client._export_repository(
+            invalid_revision, self.export_base_path
+        )
+        self.assertFalse(result, 'Export should fail for invalid revision')
+
+        tar_gz_path = self.export_base_path + '.tar.gz'
+        tar_path = self.export_base_path + '.tar'
+        self.assertFalse(os.path.exists(tar_gz_path))
+        self.assertFalse(os.path.exists(tar_path))
+
+
 if __name__ == '__main__':
     unittest.main()

vcs2l/clients/hg.py

@@ -1,3 +1,4 @@
+import gzip
 import os
 from shutil import which
 from threading import Lock
@@ -339,6 +340,36 @@ def _check_color(self, cmd):
         if HgClient._config_color:
             cmd[1:1] = '--color', 'always'
 
+    def _export_repository(self, version, basepath):
+        """Export the hg repository at a given version to a tar.gz file."""
+        self._check_executable()
+
+        cmd = [
+            HgClient._executable,
+            'archive',
+            '-t',
+            'tar',
+            '-r',
+            version,
+            basepath + '.tar',
+        ]
+        result = self._run_command(cmd)
+        if result['returncode']:
+            print('Failed to export hg repository: %s' % result['output'])
+            return False
+
+        try:
+            with open(basepath + '.tar', 'rb') as tar_file:
+                with gzip.open(basepath + '.tar.gz', 'wb') as gzip_file:
+                    gzip_file.writelines(tar_file)
+        finally:
+            try:
+                os.remove(basepath + '.tar')
+            except OSError:
+                print('Could not remove intermediate tar file %s.tar' % basepath)
+
+        return True
+
     def checkout(self, url, version='', verbose=False, shallow=False, timeout=None):
         """Checkout the repository from the given URL."""
         if url is None or url.strip() == '':