Package

tests/test_publish_package.py

@@ -1,27 +1,19 @@
-import hashlib
 import pathlib
-import subprocess
-import sys
-import tarfile
 import tempfile
 import unittest
 import unittest.mock
 
-import package.scripts.publish_latest_generic_package
+import modulus.artifacts
+import modulus.cli
+import modulus.config
+import modulus.python
 
 class PackagePublishingTests(unittest.TestCase):
   def test_configured_packager_matches_publisher(self):
     root = pathlib.Path(__file__).resolve().parents[1]
-    runtime = root / 'build/build_common/modulus'
-    if not (runtime / 'modulus/cli.py').is_file():
-      self.skipTest('Configure first to check the exported Modulus packager')
-    with unittest.mock.patch.object(sys, 'path', [str(runtime), *sys.path]):
-      import modulus.cli
-      import modulus.config
     with tempfile.TemporaryDirectory() as temporary:
       directory = pathlib.Path(temporary)
       context = modulus.cli.build_context(directory, modulus.config.load_repo_config(root))
-      name, registry, targets = package.scripts.publish_latest_generic_package.read_config(root / 'modulus.json')
       def fake_build(context, command, **kwargs):
         if command[:2] == ['cmake', '--install']:
           stage = pathlib.Path(command[2]).parent / 'stage'
@@ -29,54 +21,9 @@ class PackagePublishingTests(unittest.TestCase):
         return 0
       with unittest.mock.patch.object(modulus.cli, 'in_docker', return_value=True), \
         unittest.mock.patch.object(modulus.cli, 'run', side_effect=fake_build), \
-        unittest.mock.patch.object(modulus.cli, 'run_capture', return_value=subprocess.CompletedProcess([], 0, stdout='fixture\n', stderr='')), \
-        unittest.mock.patch.object(modulus.cli, 'upload_generic_artifact') as upload:
+        unittest.mock.patch.object(modulus.python, 'interpreter', return_value=pathlib.Path('/usr/bin/python3')), \
+        unittest.mock.patch.object(modulus.artifacts, 'main') as publish:
         self.assertEqual(modulus.cli.package_artifacts(context, ['--no-remote']), 0)
-        upload.assert_not_called()
-      try:
-        artifacts = package.scripts.publish_latest_generic_package.discover_artifacts(directory / 'build/packages', name, targets)
-      except SystemExit as error:
-        self.fail(str(error))
-      for archive, checksum in zip(artifacts[::2], artifacts[1::2]):
-        self.assertEqual(checksum.read_text(), f'{hashlib.sha256(archive.read_bytes()).hexdigest()}  {archive.name}\n')
-        with tarfile.open(archive) as contents:
-          self.assertIn('manifest.json', contents.getnames())
-  def test_only_configured_target_archives_and_checksums_are_uploaded(self):
-    with tempfile.TemporaryDirectory() as temporary:
-      root = pathlib.Path(temporary)
-      archive = root / 'product-vamp-arm_64_gcc12_release.tar.gz'
-      checksum = archive.with_name(archive.name + '.sha256')
-      for path in (archive, checksum, root / 'product-vamp-obsolete-arm_64_gcc12_release.tar.gz'):
-        path.write_bytes(b'fixture')
-      self.assertEqual(package.scripts.publish_latest_generic_package.discover_artifacts(root, 'product-vamp', ['arm_64_gcc12_release']), [archive, checksum])
-      checksum.unlink()
-      with self.assertRaisesRegex(SystemExit, 'missing artifact files'):
-        package.scripts.publish_latest_generic_package.discover_artifacts(root, 'product-vamp', ['arm_64_gcc12_release'])
-      archive.unlink()
-      with self.assertRaisesRegex(SystemExit, 'missing artifact files'):
-        package.scripts.publish_latest_generic_package.discover_artifacts(root, 'product-vamp', ['arm_64_gcc12_release'])
-  def test_mismatched_filenames_report_actual_artifacts_without_uploading(self):
-    with tempfile.TemporaryDirectory() as temporary, unittest.mock.patch.object(package.scripts.publish_latest_generic_package, 'request') as request:
-      archive = pathlib.Path(temporary) / 'product-vamp-abc123-arm_64_gcc12_release.tar.gz'
-      archive.write_bytes(b'fixture')
-      with self.assertRaises(SystemExit) as error:
-        package.scripts.publish_latest_generic_package.main(['--dist-dir', temporary, '--user', 'user', '--token', 'test-token'])
-      self.assertIn(f'Found: {archive.name}', str(error.exception))
-      self.assertIn('Publish the current Modulus tooling', str(error.exception))
-      request.assert_not_called()
-  def test_missing_artifacts_do_not_touch_the_registry(self):
-    with tempfile.TemporaryDirectory() as temporary, unittest.mock.patch.object(package.scripts.publish_latest_generic_package, 'request') as request:
-      with self.assertRaisesRegex(SystemExit, 'missing artifact files'):
-        package.scripts.publish_latest_generic_package.main(['--dist-dir', temporary, '--user', 'user', '--token', 'test-token'])
-      request.assert_not_called()
-  def test_publication_replaces_the_single_registry_slot(self):
-    archive = pathlib.Path('product-vamp-arm_64_gcc12_release.tar.gz')
-    with unittest.mock.patch.object(package.scripts.publish_latest_generic_package, 'discover_artifacts', return_value=[archive]), \
-      unittest.mock.patch.object(package.scripts.publish_latest_generic_package, 'list_versions', return_value=['latest', 'obsolete']), \
-      unittest.mock.patch.object(package.scripts.publish_latest_generic_package, 'delete_version') as delete, \
-      unittest.mock.patch.object(package.scripts.publish_latest_generic_package, 'upload_file') as upload:
-      self.assertEqual(package.scripts.publish_latest_generic_package.main(['--user', 'user', '--token', 'test-token']), 0)
-      self.assertEqual([call.args[-2] for call in delete.call_args_list], ['latest', 'obsolete'])
-      self.assertEqual(upload.call_args.args[2], archive)
-      url = package.scripts.publish_latest_generic_package.file_upload_url('https://example.test/generic', 'product-vamp', archive)
-      self.assertEqual(url, 'https://example.test/generic/product-vamp/latest/product-vamp-arm_64_gcc12_release.tar.gz')
+        publish.assert_not_called()
+      artifacts = modulus.artifacts.discover_artifacts(directory / 'build/packages', context.package['name'], context.artifact['targets'])
+      self.assertEqual([p.name for p in artifacts], ['product_vamp_arm_64_gcc12_release.tar.gz', 'product_vamp_arm_64_gcc12_release.tar.gz.sha256'])