Skip to content

Commit 6ca0d3c

Browse files
authored
Add --mode and --clean parameters to openstack stress command (#2147)
Sync with new openstack-simple-stress parameters: --mode (rolling/block execution mode) and --clean (cleanup leftover resources matching prefix). Signed-off-by: Christian Berendt <berendt@osism.tech>
1 parent c7311c4 commit 6ca0d3c

1 file changed

Lines changed: 15 additions & 0 deletions

File tree

osism/commands/stress.py

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -43,6 +43,11 @@ def get_parser(self, prog_name):
4343
action="store_true",
4444
help="Do not wait for resources",
4545
)
46+
parser.add_argument(
47+
"--clean",
48+
action="store_true",
49+
help="Clean up leftover resources matching the prefix",
50+
)
4651

4752
# Integer parameters with defaults
4853
parser.add_argument(
@@ -149,6 +154,13 @@ def get_parser(self, prog_name):
149154
default="__DEFAULT__",
150155
help="Volume type (default: %(default)s)",
151156
)
157+
parser.add_argument(
158+
"--mode",
159+
type=str,
160+
default="rolling",
161+
choices=["rolling", "block"],
162+
help="Execution mode (default: %(default)s)",
163+
)
152164

153165
return parser
154166

@@ -187,6 +199,8 @@ def take_action(self, parsed_args):
187199
command.append("--no-boot-volume")
188200
if parsed_args.no_wait:
189201
command.append("--no-wait")
202+
if parsed_args.clean:
203+
command.append("--clean")
190204

191205
# Add integer parameters
192206
command.extend(["--interval", str(parsed_args.interval)])
@@ -207,6 +221,7 @@ def take_action(self, parsed_args):
207221
command.extend(["--storage-zone", parsed_args.storage_zone])
208222
command.extend(["--affinity", parsed_args.affinity])
209223
command.extend(["--volume-type", parsed_args.volume_type])
224+
command.extend(["--mode", parsed_args.mode])
210225

211226
logger.debug(
212227
f"Executing OpenStack stress test with command: {' '.join(command)}"

0 commit comments

Comments
 (0)