Integration tests for container scaling changes

Signed-off-by: Yashodhan Pise <technoy@gmail.com>
This commit is contained in:
Yashodhan Pise 2025-03-31 18:44:12 +05:30 committed by Povilas Kanapickas
parent 11879d3e94
commit fbc4c7da80
4 changed files with 133 additions and 0 deletions

View File

@ -0,0 +1,6 @@
name: podman-compose
services:
service1:
image: docker.io/library/busybox:latest
tty: true

View File

@ -0,0 +1,9 @@
name: podman-compose
services:
service1:
image: docker.io/library/busybox:latest
tty: true
deploy:
mode: replicated
replicas: 2

View File

@ -0,0 +1,7 @@
name: podman-compose
services:
service1:
image: docker.io/library/busybox:latest
tty: true
scale: 2

View File

@ -0,0 +1,111 @@
# 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",
])