import hashlib
import importlib.util
from pathlib import Path
import subprocess
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
        available, docker_run = Mock(return_value=True), Mock()
        cli.docker_image_available, cli.docker_run = available, docker_run
        cli.main = lambda command: cli.docker_image_available(types.SimpleNamespace(docker_image='test-image'))
        for label, expected in [('',False), ('outdated',False), (digest,True)]:
            result = subprocess.CompletedProcess([], 0, label+'\n', '')
            with self.subTest(label=label), patch.dict(sys.modules, {'modulus':package, 'modulus.cli':cli}), \
                 patch.object(bootstrap, 'ensure_runtime'), patch.object(bootstrap.subprocess, 'run', return_value=result), \
                 patch.object(sys, 'path', list(sys.path)):
                self.assertEqual(bootstrap.delegate('configure', []), expected)
                self.assertIs(cli.docker_image_available, available)
                self.assertIs(cli.docker_run, docker_run)