mirror of
https://github.com/containers/podman-compose.git
synced 2025-04-23 17:09:23 +02:00
112 lines
3.6 KiB
Python
112 lines
3.6 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):
|
|
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):
|
|
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_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):
|
|
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_scale_parameter"),
|
|
"ps",
|
|
"-q",
|
|
])
|
|
self.assertEqual(len(output.splitlines()), 2)
|
|
finally:
|
|
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):
|
|
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=2",
|
|
])
|
|
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_compose_path(),
|
|
"-f",
|
|
compose_yaml_path('scaleup_cli'),
|
|
"down",
|
|
"-t",
|
|
"0",
|
|
])
|