khodowany
36 lines
1.6 KiB
Python
36 lines
1.6 KiB
Python
import hashlib
|
|
import importlib.util
|
|
from pathlib import Path
|
|
import sys
|
|
import types
|
|
import unittest
|
|
from unittest.mock import Mock, patch
|
|
|
|
class BuildBootstrapTests(unittest.TestCase):
|
|
def test_requirements_fingerprint_invalidates_cached_build_image(self):
|
|
root = Path(__file__).resolve().parents[1]
|
|
spec = importlib.util.spec_from_file_location('vamp_bootstrap', root / 'modulus.py')
|
|
bootstrap = importlib.util.module_from_spec(spec)
|
|
spec.loader.exec_module(bootstrap)
|
|
digest = hashlib.sha256((root / 'generator/requirements.txt').read_bytes()).hexdigest()
|
|
self.assertEqual(bootstrap.build_requirements_digest(), digest)
|
|
self.assertIn(f'build-requirements="{digest}"', bootstrap.product_build_dockerfile())
|
|
self.assertIn('COPY generator/requirements.txt /tmp/vamp-build-requirements.txt', bootstrap.product_build_dockerfile())
|
|
package = types.ModuleType('modulus')
|
|
cli = types.ModuleType('modulus.cli')
|
|
package.cli = cli
|
|
docker_run = Mock()
|
|
cli.docker_run = docker_run
|
|
cli.main = Mock(return_value=0)
|
|
for command in ('clangd', 'format'):
|
|
with patch.dict(sys.modules, {'modulus': package, 'modulus.cli': cli}), \
|
|
patch.object(bootstrap, 'ensure_runtime') as prepare, \
|
|
patch.object(bootstrap, 'runtime_ready', return_value=True), \
|
|
patch.object(sys, 'path', list(sys.path)):
|
|
self.assertEqual(bootstrap.delegate(command, []), 0)
|
|
prepare.assert_not_called()
|
|
docker_run.assert_not_called()
|
|
cli.main.assert_called_once_with(command)
|
|
self.assertIs(cli.docker_run, docker_run)
|
|
cli.main.reset_mock()
|