Mathieu Benoit
1e731114f5
Reflect the current year in all TechnoLibre license headers across script/, test/, and docker/. Generated by Claude Code 2.1.74 model claude-sonnet-4-6 Co-Authored-By: Mathieu Benoit <mathben@technolibre.ca>
121 lines
3.5 KiB
Python
121 lines
3.5 KiB
Python
#!/usr/bin/env python3
|
|
# © 2026 TechnoLibre (http://www.technolibre.ca)
|
|
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl)
|
|
|
|
import os
|
|
import tempfile
|
|
import unittest
|
|
from types import SimpleNamespace
|
|
|
|
from script.docker.docker_update_version import edit_text, edit_docker_prod
|
|
|
|
|
|
class TestEditText(unittest.TestCase):
|
|
"""Test docker-compose.yml image version update."""
|
|
|
|
def _write_compose(self, content):
|
|
f = tempfile.NamedTemporaryFile(
|
|
mode="w", suffix=".yml", delete=False
|
|
)
|
|
f.write(content)
|
|
f.close()
|
|
return f.name
|
|
|
|
def test_updates_image_after_erplibre(self):
|
|
path = self._write_compose(
|
|
"services:\n"
|
|
" ERPLibre:\n"
|
|
" image: old:1.0\n"
|
|
" ports:\n"
|
|
)
|
|
config = SimpleNamespace(
|
|
docker_compose_file=path,
|
|
prod_version="myrepo:2.0",
|
|
)
|
|
edit_text(config)
|
|
with open(path) as f:
|
|
content = f.read()
|
|
self.assertIn("image: myrepo:2.0", content)
|
|
self.assertNotIn("old:1.0", content)
|
|
os.unlink(path)
|
|
|
|
def test_preserves_other_lines(self):
|
|
path = self._write_compose(
|
|
"version: '3'\n"
|
|
"services:\n"
|
|
" ERPLibre:\n"
|
|
" image: old:1.0\n"
|
|
" postgres:\n"
|
|
" image: postgres:14\n"
|
|
)
|
|
config = SimpleNamespace(
|
|
docker_compose_file=path,
|
|
prod_version="new:3.0",
|
|
)
|
|
edit_text(config)
|
|
with open(path) as f:
|
|
content = f.read()
|
|
self.assertIn("version: '3'", content)
|
|
self.assertIn("postgres:14", content)
|
|
os.unlink(path)
|
|
|
|
|
|
class TestEditDockerProd(unittest.TestCase):
|
|
"""Test Dockerfile.prod FROM line update."""
|
|
|
|
def _write_dockerfile(self, content):
|
|
f = tempfile.NamedTemporaryFile(
|
|
mode="w", suffix=".pkg", delete=False
|
|
)
|
|
f.write(content)
|
|
f.close()
|
|
return f.name
|
|
|
|
def test_updates_from_line(self):
|
|
path = self._write_dockerfile(
|
|
"FROM base:old\nRUN apt-get update\n"
|
|
)
|
|
config = SimpleNamespace(
|
|
docker_compose_file="unused",
|
|
docker_prod=path,
|
|
base_version="base:new",
|
|
)
|
|
edit_docker_prod(config)
|
|
with open(path) as f:
|
|
content = f.read()
|
|
self.assertIn("FROM base:new", content)
|
|
self.assertNotIn("FROM base:old", content)
|
|
os.unlink(path)
|
|
|
|
def test_preserves_run_lines(self):
|
|
path = self._write_dockerfile(
|
|
"FROM base:old\nRUN echo hello\nCOPY . /app\n"
|
|
)
|
|
config = SimpleNamespace(
|
|
docker_compose_file="unused",
|
|
docker_prod=path,
|
|
base_version="base:v2",
|
|
)
|
|
edit_docker_prod(config)
|
|
with open(path) as f:
|
|
content = f.read()
|
|
self.assertIn("RUN echo hello", content)
|
|
self.assertIn("COPY . /app", content)
|
|
os.unlink(path)
|
|
|
|
def test_multiple_from_lines(self):
|
|
path = self._write_dockerfile(
|
|
"FROM base:old\nRUN build\nFROM base:old\nRUN run\n"
|
|
)
|
|
config = SimpleNamespace(
|
|
docker_compose_file="unused",
|
|
docker_prod=path,
|
|
base_version="base:v3",
|
|
)
|
|
edit_docker_prod(config)
|
|
with open(path) as f:
|
|
lines = f.readlines()
|
|
from_lines = [l for l in lines if l.startswith("FROM")]
|
|
for line in from_lines:
|
|
self.assertIn("base:v3", line)
|
|
os.unlink(path)
|