Add a custom flag to behave clear_screen=true

Author: jameslinnell

pyproject.toml

@@ -56,6 +56,7 @@ boto3-stubs = { extras = [
 ], version = "^1.34.37" }
 pytest-timeout = "^2.2.0"
 lz4 = "^4.3.3"
+click = "^8.1.7"
 
 [tool.poetry.group.local]
 optional = true

src/api/tests/feature_tests/environment.py

@@ -1,6 +1,8 @@
+import time
 from contextlib import nullcontext
 from pathlib import Path
 
+import click
 from behave import use_fixture
 from behave.model import Feature, Scenario, Step
 from event.aws.client import dynamodb_client
@@ -30,6 +32,8 @@
 
 
 def before_all(context: Context):
+    clear_screen = context.config.userdata.get("clear_screen", "false")
+    context.clear_screen = clear_screen.lower() == "true"
     context.postman_collection = PostmanCollection()
     context.test_mode = TestMode.parse(config=context.config)
     context.table_name = LOCAL_TABLE_NAME
@@ -114,6 +118,9 @@ def after_scenario(context: Context, scenario: Scenario):
 def after_feature(context: Context, feature: Feature):
     if context.postman_feature:
         context.postman_collection.item.append(context.postman_feature)
+    if context.clear_screen:
+        click.clear()
+        time.sleep(0.5)
 
 
 def after_all(context: Context):