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 / 'tools/requirements.txt').read_bytes()).hexdigest()
        self.assertEqual(bootstrap.build_requirements_digest(), digest)
        self.assertIn(f'build-requirements="{digest}"', 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)
        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('clangd', []), 0)
            prepare.assert_not_called()
            docker_run.assert_not_called()
            cli.main.assert_called_once_with('clangd')
            self.assertIs(cli.docker_run, docker_run)