63 lines
3.0 KiB
Python
63 lines
3.0 KiB
Python
#!/usr/bin/env python3
|
|
"""Check deployment preparation without host compilers or a Docker daemon."""
|
|
import json
|
|
import os
|
|
from pathlib import Path
|
|
import shutil
|
|
import subprocess
|
|
import sys
|
|
import tempfile
|
|
import unittest
|
|
|
|
SCRIPT = Path(__file__).with_name('prepare-deploy.py')
|
|
|
|
|
|
class PrepareDeployment(unittest.TestCase):
|
|
def test_docker_only_build_and_failure_propagation(self):
|
|
with tempfile.TemporaryDirectory(prefix='chublog deploy ') as temporary:
|
|
root = Path(temporary)
|
|
(root / 'scripts').mkdir()
|
|
shutil.copy2(SCRIPT, root / 'scripts/prepare-deploy.py')
|
|
for name in ['external/ChuSQL-RustDriver/Cargo.toml',
|
|
'external/ChuSQL/chusql-core/storage/Cargo.toml']:
|
|
manifest = root / name
|
|
manifest.parent.mkdir(parents=True)
|
|
manifest.touch()
|
|
(root / '.env.example').touch()
|
|
tools = root / 'tools'
|
|
tools.mkdir()
|
|
docker = tools / 'docker'
|
|
docker.write_text(f'#!{sys.executable}\nimport json, os, sys\n'
|
|
'from pathlib import Path\n'
|
|
'Path(os.environ["CALL_LOG"]).write_text(json.dumps(sys.argv[1:]))\n'
|
|
'sys.exit(int(os.environ.get("DOCKER_EXIT", "0")))\n')
|
|
docker.chmod(0o755)
|
|
log = root / 'call.json'
|
|
env = {**os.environ, 'PATH': str(tools), 'CALL_LOG': str(log)}
|
|
command = [sys.executable, str(root / 'scripts/prepare-deploy.py')]
|
|
result = subprocess.run(command + ['--build-database'], env=env, capture_output=True, text=True)
|
|
self.assertEqual(result.returncode, 0, result.stderr)
|
|
call = json.loads(log.read_text())
|
|
self.assertEqual(call[:3], ['compose', '--env-file', str(root / '.env.example')])
|
|
self.assertIn('db', call)
|
|
self.assertIn('web', call)
|
|
self.assertFalse((root / 'deploy/bin').exists())
|
|
override = root / 'public sources.env'
|
|
override.write_text('REGISTRY_PREFIX=public.ecr.aws/docker/library/\n')
|
|
result = subprocess.run(command + ['--sources-env-file', str(override)],
|
|
env=env, capture_output=True, text=True)
|
|
self.assertEqual(result.returncode, 0, result.stderr)
|
|
self.assertEqual(json.loads(log.read_text())[3:5], ['--env-file', str(override)])
|
|
(root / '.env').touch()
|
|
result = subprocess.run(command, env={**env, 'DOCKER_EXIT': '17'}, capture_output=True, text=True)
|
|
self.assertEqual(result.returncode, 17)
|
|
self.assertEqual(json.loads(log.read_text())[2], str(root / '.env'))
|
|
docker.unlink()
|
|
result = subprocess.run(command, env=env, capture_output=True, text=True)
|
|
self.assertEqual(result.returncode, 2)
|
|
self.assertIn('Docker with the Compose plugin is required', result.stderr)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|