diff options
Diffstat (limited to 'tests/conftest.py')
| -rw-r--r-- | tests/conftest.py | 30 |
1 files changed, 24 insertions, 6 deletions
diff --git a/tests/conftest.py b/tests/conftest.py index 39d92322..a9452410 100644 --- a/tests/conftest.py +++ b/tests/conftest.py | |||
| @@ -299,9 +299,18 @@ class VdkrRunner: | |||
| 299 | raise AssertionError(error_msg) | 299 | raise AssertionError(error_msg) |
| 300 | return result | 300 | return result |
| 301 | 301 | ||
| 302 | def memres_start(self, timeout=120): | 302 | def memres_start(self, timeout=120, port_forwards=None): |
| 303 | """Start memory resident mode.""" | 303 | """Start memory resident mode. |
| 304 | return self.run("memres", "start", timeout=timeout) | 304 | |
| 305 | Args: | ||
| 306 | timeout: Command timeout in seconds | ||
| 307 | port_forwards: List of port forwards, e.g., ["8080:80", "2222:22"] | ||
| 308 | """ | ||
| 309 | args = ["memres", "start"] | ||
| 310 | if port_forwards: | ||
| 311 | for pf in port_forwards: | ||
| 312 | args.extend(["-p", pf]) | ||
| 313 | return self.run(*args, timeout=timeout) | ||
| 305 | 314 | ||
| 306 | def memres_stop(self, timeout=30): | 315 | def memres_stop(self, timeout=30): |
| 307 | """Stop memory resident mode.""" | 316 | """Stop memory resident mode.""" |
| @@ -506,9 +515,18 @@ class VpdmnRunner: | |||
| 506 | raise AssertionError(error_msg) | 515 | raise AssertionError(error_msg) |
| 507 | return result | 516 | return result |
| 508 | 517 | ||
| 509 | def memres_start(self, timeout=120): | 518 | def memres_start(self, timeout=120, port_forwards=None): |
| 510 | """Start memory resident mode.""" | 519 | """Start memory resident mode. |
| 511 | return self.run("memres", "start", timeout=timeout) | 520 | |
| 521 | Args: | ||
| 522 | timeout: Command timeout in seconds | ||
| 523 | port_forwards: List of port forwards, e.g., ["8080:80", "2222:22"] | ||
| 524 | """ | ||
| 525 | args = ["memres", "start"] | ||
| 526 | if port_forwards: | ||
| 527 | for pf in port_forwards: | ||
| 528 | args.extend(["-p", pf]) | ||
| 529 | return self.run(*args, timeout=timeout) | ||
| 512 | 530 | ||
| 513 | def memres_stop(self, timeout=30): | 531 | def memres_stop(self, timeout=30): |
| 514 | """Stop memory resident mode.""" | 532 | """Stop memory resident mode.""" |