mirror of
https://github.com/containers/podman-compose.git
synced 2025-05-31 23:45:51 +02:00
147 lines
4.9 KiB
Python
147 lines
4.9 KiB
Python
# SPDX-License-Identifier: GPL-2.0
|
|
|
|
import os
|
|
import unittest
|
|
|
|
from tests.integration.test_utils import RunSubprocessMixin
|
|
from tests.integration.test_utils import podman_compose_path
|
|
from tests.integration.test_utils import test_path
|
|
|
|
|
|
def compose_yaml_path(test_ref_folder: str) -> str:
|
|
return os.path.join(test_path(), "service_scale", test_ref_folder, "docker-compose.yml")
|
|
|
|
|
|
class TestComposeScale(unittest.TestCase, RunSubprocessMixin):
|
|
# scale-up using `scale` prarmeter in docker-compose.yml
|
|
def test_scaleup_scale_parameter(self) -> None:
|
|
try:
|
|
output, _, return_code = self.run_subprocess([
|
|
podman_compose_path(),
|
|
"-f",
|
|
compose_yaml_path("scaleup_scale_parameter"),
|
|
"up",
|
|
"-d",
|
|
])
|
|
self.assertEqual(return_code, 0)
|
|
output, _, return_code = self.run_subprocess([
|
|
podman_compose_path(),
|
|
"-f",
|
|
compose_yaml_path("scaleup_scale_parameter"),
|
|
"ps",
|
|
"-q",
|
|
])
|
|
self.assertEqual(len(output.splitlines()), 2)
|
|
finally:
|
|
self.run_subprocess_assert_returncode([
|
|
"podman",
|
|
"rm",
|
|
"--force",
|
|
"-t",
|
|
"0",
|
|
"podman-compose_service1_1",
|
|
"podman-compose_service1_2",
|
|
])
|
|
self.run_subprocess_assert_returncode([
|
|
podman_compose_path(),
|
|
"-f",
|
|
compose_yaml_path("scaleup_scale_parameter"),
|
|
"down",
|
|
"-t",
|
|
"0",
|
|
])
|
|
|
|
# scale-up using `deploy => replicas` prarmeter in docker-compose.yml
|
|
def test_scaleup_deploy_replicas_parameter(self) -> None:
|
|
try:
|
|
output, _, return_code = self.run_subprocess([
|
|
podman_compose_path(),
|
|
"-f",
|
|
compose_yaml_path('scaleup_deploy_replicas_parameter'),
|
|
"up",
|
|
"-d",
|
|
])
|
|
self.assertEqual(return_code, 0)
|
|
output, _, return_code = self.run_subprocess([
|
|
podman_compose_path(),
|
|
"-f",
|
|
compose_yaml_path("scaleup_deploy_replicas_parameter"),
|
|
"ps",
|
|
"-q",
|
|
])
|
|
self.assertEqual(len(output.splitlines()), 3)
|
|
finally:
|
|
self.run_subprocess_assert_returncode([
|
|
"podman",
|
|
"rm",
|
|
"--force",
|
|
"-t",
|
|
"0",
|
|
"podman-compose_service1_1",
|
|
"podman-compose_service1_2",
|
|
"podman-compose_service1_3",
|
|
])
|
|
self.run_subprocess_assert_returncode([
|
|
podman_compose_path(),
|
|
"-f",
|
|
compose_yaml_path('scaleup_deploy_replicas_parameter'),
|
|
"down",
|
|
"-t",
|
|
"0",
|
|
])
|
|
|
|
# scale-up using `--scale <SERVICE>=<number of replicas>` argument in CLI
|
|
def test_scaleup_cli(self) -> None:
|
|
try:
|
|
output, _, return_code = self.run_subprocess([
|
|
podman_compose_path(),
|
|
"-f",
|
|
compose_yaml_path('scaleup_cli'),
|
|
"up",
|
|
"-d",
|
|
])
|
|
self.assertEqual(return_code, 0)
|
|
output, _, return_code = self.run_subprocess([
|
|
podman_compose_path(),
|
|
"-f",
|
|
compose_yaml_path('scaleup_cli'),
|
|
"up",
|
|
"-d",
|
|
"--scale",
|
|
"service1=4",
|
|
])
|
|
# error code 125 is expected as podman-compose complains about already used name
|
|
# "podman-compose_service1_1" for the 1st container
|
|
# Nevertheless, following containers are still created to scale as expected
|
|
# (in test case till 3 containers)
|
|
self.assertEqual(return_code, 125)
|
|
|
|
output, _, return_code = self.run_subprocess([
|
|
podman_compose_path(),
|
|
"-f",
|
|
compose_yaml_path('scaleup_cli'),
|
|
"ps",
|
|
"-q",
|
|
])
|
|
self.assertEqual(len(output.splitlines()), 4)
|
|
finally:
|
|
self.run_subprocess_assert_returncode([
|
|
"podman",
|
|
"rm",
|
|
"--force",
|
|
"-t",
|
|
"0",
|
|
"podman-compose_service1_1",
|
|
"podman-compose_service1_2",
|
|
"podman-compose_service1_3",
|
|
"podman-compose_service1_4",
|
|
])
|
|
self.run_subprocess_assert_returncode([
|
|
podman_compose_path(),
|
|
"-f",
|
|
compose_yaml_path('scaleup_cli'),
|
|
"down",
|
|
"-t",
|
|
"0",
|
|
])
|