stoobly-agent 0.21.0__py3-none-any.whl → 0.22.3__py3-none-any.whl

stoobly_agent/app/cli/helpers/print_service.py

@@ -11,6 +11,14 @@ def select_print_options(kwargs):
 
     return print_options
 
+def print_projects(projects, **kwargs):
+    tabulate_print(
+        projects, 
+        filter=['created_at', 'is_deleted', 'organization_id', 'project_id', 'starred', 'updated_at'],
+        headers=not kwargs.get('without_headers'),
+        select=kwargs.get('select') or []
+    )
+
 def print_requests(requests, **kwargs):
     tabulate_print(
       requests, 

stoobly_agent/app/cli/helpers/scenario_facade.py

@@ -3,53 +3,35 @@ import requests
 
 from stoobly_agent.app.cli.helpers.iterate_group_by import iterate_group_by
 from stoobly_agent.app.cli.helpers.replay_facade import ReplayCliOptions, ReplayFacade, TestCliOptions
+from stoobly_agent.app.models.scenario_model import ScenarioModel
 from stoobly_agent.app.proxy.replay.replay_scenario_service import inject_replay
 from stoobly_agent.app.proxy.replay.trace_context import TraceContext
 from stoobly_agent.app.settings import Settings
 from stoobly_agent.config.constants import mode
 from stoobly_agent.lib.api.interfaces.scenarios import ScenarioShowResponse, ScenariosIndexResponse
 from stoobly_agent.lib.api.keys import ProjectKey, ScenarioKey
-from stoobly_agent.lib.api.scenarios_resource import ScenariosResource
 
 class ScenarioFacade(ReplayFacade):
 
   def __init__(self, settings: Settings):
-    self.__api = ScenariosResource(settings.remote.api_url, settings.remote.api_key)
     super().__init__(settings)
+    self.__model = ScenarioModel(settings)
 
   def create(self, project_key: str, name: str, description: str = ''):
-    res: requests.Response = self.__api.from_project_key(
-      project_key, 
-      lambda project_id: self.__api.create(
-        project_id, {
-          'description': description,
-          'name': name,
-        }
-      )
-    )
-
-    if not res.ok:
-      raise AssertionError(res.content)
-
-    return res.json()
+    key = ProjectKey(project_key)
+    return self.__model.create(**{
+      'project_id': key.id, 
+      'description': description,
+      'name': name,
+    })
 
   def index(self, project_key, cli_options: dict) -> ScenariosIndexResponse:
     key = ProjectKey(project_key)
-    res = self.__api.index(**{ 'project_id': key.id, **cli_options})
-
-    if not res.ok:
-      raise AssertionError(res.content)
-
-    return res.json()
+    return self.__model.index(**{ 'project_id': key.id, **cli_options})
 
   def show(self, scenario_key: str) -> ScenarioShowResponse:
     key = ScenarioKey(scenario_key)
-    res = self.__api.show(key.id)
-
-    if not res.ok:
-      raise AssertionError(res.content)
-
-    return res.json()
+    return self.__model.show(key.id)
 
   def replay(self, source_key: str, cli_options: ReplayCliOptions):
     return self.__replay(source_key, {

stoobly_agent/app/cli/intercept.py

@@ -3,15 +3,59 @@ import pdb
 
 from stoobly_agent.app.cli.decorators.config import ConfigDecorator
 from stoobly_agent.app.settings import Settings
+from stoobly_agent.config.constants import mode
 
 settings = Settings.instance()
+mode_options = [mode.MOCK, mode.RECORD, mode.REPLAY]
+
+if settings.cli.features.remote:
+    mode_options.append(mode.TEST)
 
 @click.group(
     epilog="Run 'stoobly-agent intercept COMMAND --help' for more information on a command.",
-    help="Toggle request intercept"
+    help="Manage request intercept"
 )
 @click.pass_context
 def intercept(ctx):
     pass
 
-ConfigDecorator(intercept, settings, 'proxy.intercept.active').decorate()
+@intercept.command(
+    help="Enable intercept"
+)
+@click.option('--mode', type=click.Choice(mode_options))
+def enable(**kwargs):
+    settings = Settings.instance()
+
+    if kwargs['mode']:
+        settings.proxy.intercept.mode = kwargs['mode']
+
+    settings.proxy.intercept.active = True
+
+    settings.commit()
+
+    print("Intercept enabled!")
+
+@intercept.command(
+    help="Disable intercept"
+)
+def disable(**kwargs):
+    settings = Settings.instance()
+
+    settings.proxy.intercept.active = False
+
+    settings.commit()
+
+    print("Intercept disabled!")
+
+@intercept.command(
+    help="Show intercept"
+)
+def show(**kwargs):
+    settings = Settings.instance()
+
+    mode = settings.proxy.intercept.mode
+
+    if not mode:
+        print('No intercept mode set')
+    else:
+        print(f"{mode.capitalize()} {'enabled' if settings.proxy.intercept.active else 'disabled'}")

stoobly_agent/app/cli/main_group.py

@@ -1,6 +1,5 @@
 import click
 import collections
-import pdb
 
 from functools import reduce
 from typing import List, TypedDict
@@ -27,23 +26,23 @@ class MainGroup(click.Group):
     command_groups: List[CommandGroup] = [
       {
         'name': 'Commands',
-        'commands': ['dev-tools', 'exec', 'feature'],
+        'commands': ['dev-tools', 'exec', 'feature', 'init', 'mock', 'record'],
       },
       {
         'name': 'Proxy Commands',
         'commands': ['ca-cert', 'config', 'intercept', 'run'],
-      }  
+      }
     ]
 
     if self.__settings.cli.features.remote:
       command_groups.append({
-        'name': 'Remote Commands',
-        'commands': ['project', 'report', 'scenario', 'request', 'trace'],
+        'name': 'Remote Resource Commands',
+        'commands': ['project', 'report', 'request', 'scenario', 'trace'],
       })
     else:
       command_groups.append({
-        'name': 'Local Commands',
-        'commands': ['request'],
+        'name': 'Local Resource Commands',
+        'commands': ['request', 'scenario'],
       })
 
     self.__print(formatter, command_groups)
@@ -69,6 +68,6 @@ class MainGroup(click.Group):
 
   def __get_commands(self, command_names: List[str]):
     return list(filter(
-      lambda c: c != None, 
+      lambda c: c != None,
       map(lambda c: self.commands.get(c), command_names)
-    ))
+    ))

stoobly_agent/app/cli/project_cli.py

@@ -5,7 +5,7 @@ import sys
 from stoobly_agent.app.settings import Settings
 
 from .helpers import ProjectFacade
-from .helpers.tabulate_print_service import tabulate_print
+from .helpers.print_service import print_projects, select_print_options
 from .helpers.validations import *
 
 @click.group(
@@ -25,7 +25,7 @@ def project(ctx):
 @click.option('--without-headers', is_flag=True, default=False, help='Disable printing column headers.')
 @click.argument('name')
 def create(**kwargs):
-    print_options = __select_print_options(kwargs)
+    print_options = select_print_options(kwargs)
 
     project = ProjectFacade(Settings.instance())
 
@@ -34,7 +34,7 @@ def create(**kwargs):
     except AssertionError as e:
         return print(e, file=sys.stderr)
 
-    __print([project_response], **print_options)
+    print_projects([project_response], **print_options)
 
 @project.command(
     help="Show all projects"
@@ -47,7 +47,7 @@ def create(**kwargs):
 @click.option('--without-headers', is_flag=True, default=False, help='Disable printing column headers.')
 @click.argument('organization_key', required=False)
 def list(**kwargs):
-    print_options = __select_print_options(kwargs)
+    print_options = select_print_options(kwargs)
     settings = Settings.instance()
     project = ProjectFacade(settings)
 
@@ -57,7 +57,9 @@ def list(**kwargs):
     if not organization_key:
         try:
             project_key = ProjectKey(settings.proxy.intercept.project_key)
-            organization_key = OrganizationKey.encode(project_key.organization_id)
+
+            if not project_key.is_local:
+                organization_key = OrganizationKey.encode(project_key.organization_id)
         except InvalidProjectKey as e:
             pass
 
@@ -81,7 +83,7 @@ def list(**kwargs):
     if len(projects_response['list']) == 0:
         print('No projects found.', file=sys.stderr)
     else:
-        __print(projects_response['list'], **print_options)
+        print_projects(projects_response['list'], **print_options)
 
 @project.command(
     help="Describe project"
@@ -90,7 +92,7 @@ def list(**kwargs):
 @click.option('--without-headers', is_flag=True, default=False, help='Disable printing column headers.')
 @click.argument('project_key', required=False)
 def show(**kwargs):
-    print_options = __select_print_options(kwargs)
+    print_options = select_print_options(kwargs)
 
     settings = Settings.instance()
     project_key = resolve_project_key_and_validate(kwargs, settings)
@@ -101,23 +103,4 @@ def show(**kwargs):
     except AssertionError as e:
         return print(e, file=sys.stderr)
 
-    __print([project_response], **print_options)
-
-def __print(projects, **kwargs):
-    tabulate_print(
-        projects, 
-        filter=['created_at', 'organization_id', 'project_id', 'starred', 'updated_at'],
-        headers=not kwargs.get('without_headers'),
-        select=kwargs.get('select') or []
-    )
-
-def __select_print_options(kwargs):
-    print_options = {
-        'select': kwargs['select'],
-        'without_headers': kwargs['without_headers']
-    }
-
-    del kwargs['without_headers']
-    del kwargs['select']
-
-    return print_options
+    print_projects([project_response], **print_options)

stoobly_agent/app/cli/request_cli.py

@@ -38,7 +38,7 @@ def request(ctx):
 )
 @click.option('--page', default=0)
 @ConditionalDecorator(lambda f: click.option('--project-key')(f), is_remote)
-@ConditionalDecorator(lambda f: click.option('--scenario-key')(f), is_remote)
+@click.option('--scenario-key')
 @click.option('--select', multiple=True, help='Select column(s) to display.')
 @click.option('--sort-by', default='created_at', help='created_at|path')
 @click.option('--sort-order', default='desc', help='asc | desc')
@@ -68,16 +68,16 @@ def list(**kwargs):
 @request.command(
   help="Replay a request"
 )
-@click.option(
+@ConditionalDecorator(lambda f: click.option(
     '--alias-resolve-strategy', 
     default=alias_resolve_strategy.NONE, 
     type=click.Choice([alias_resolve_strategy.NONE, alias_resolve_strategy.FIFO, alias_resolve_strategy.LIFO]), 
     help='Strategy for resolving dynamic values for aliases.'
-)
-@click.option('--assign', multiple=True, help='Assign alias values. Format: <NAME>=<VALUE>')
+)(f), is_remote)
+@ConditionalDecorator(lambda f: click.option('--assign', multiple=True, help='Assign alias values. Format: <NAME>=<VALUE>')(f), is_remote)
 @click.option('--format', default=DEFAULT_FORMAT, type=click.Choice([DEFAULT_FORMAT, JSON_FORMAT]), help='Format replay response.')
 @click.option('--host', help='Rewrite request host.')
-@click.option('--group-by', help='Repeat for each alias name.')
+@ConditionalDecorator(lambda f: click.option('--group-by', help='Repeat for each alias name.')(f), is_remote)
 @click.option('--lifecycle-hooks-script-path', help='Path to lifecycle hooks script.')
 @click.option(
   '--log-level', default=logger.WARNING, type=click.Choice(log_levels), 
@@ -86,10 +86,10 @@ def list(**kwargs):
   '''
 )
 @click.option('--record', is_flag=True, default=False, help='Replay and record request.')
-@ConditionalDecorator(lambda f: click.option('--scenario-key', help='Record to scenario.')(f), is_remote)
+@click.option('--scenario-key', help='Record to scenario.')
 @click.option('--scheme', help='Rewrite request scheme.')
-@click.option('--trace-id', help='Use existing trace.')
-@click.option('--validate', multiple=True, help='Validate one or more aliases. Format: <NAME>=?<TYPE>')
+@ConditionalDecorator(lambda f: click.option('--trace-id', help='Use existing trace.')(f), is_remote)
+@ConditionalDecorator(lambda f: click.option('--validate', multiple=True, help='Validate one or more aliases. Format: <NAME>=?<TYPE>')(f), is_remote)
 @click.argument('request_key')
 def replay(**kwargs):
   os.environ[env_vars.LOG_LEVEL] = kwargs['log_level']
@@ -103,7 +103,7 @@ def replay(**kwargs):
 
     validate_scenario_key(kwargs['scenario_key'])
 
-  if len(kwargs['validate']):
+  if 'validate' in kwargs and len(kwargs['validate']):
       validate_aliases(kwargs['validate'], assign=kwargs['assign'], format=kwargs['format'], trace_id=kwargs['trace_id'])
 
   __assign_default_alias_resolve_strategy(kwargs)
@@ -203,9 +203,11 @@ def get(**kwargs):
 
   print(res.content)
 
-@response.command()
+@response.command(
+  help="Query properties in response"
+)
 @click.option('--query', required=True)
-@click.argument('request-key')
+@click.argument('request_key')
 def query(**kwargs):
   validate_request_key(kwargs['request_key'])
 
@@ -218,12 +220,6 @@ def query(**kwargs):
 
   print(jmespath.search(kwargs['query'], decoded_response))
 
-@response.command(
-  help="Set new mocked response"
-)
-def set(**kwargs):
-  print("Not yet implemented. Stay tuned!")
-
 request.add_command(response)
 
 def __replay(handler, kwargs) -> requests.Response:
@@ -238,5 +234,5 @@ def __replay(handler, kwargs) -> requests.Response:
 
 def __assign_default_alias_resolve_strategy(kwargs):
     # If we have assigned values to aliases, it's likely we want to also have them resolved
-    if len(kwargs['assign']) > 0 and kwargs['alias_resolve_strategy'] == alias_resolve_strategy.NONE:
+    if 'assign' in kwargs and len(kwargs['assign']) > 0 and kwargs['alias_resolve_strategy'] == alias_resolve_strategy.NONE:
         kwargs['alias_resolve_strategy'] = alias_resolve_strategy.FIFO

stoobly_agent/app/cli/scenario_cli.py

@@ -11,10 +11,14 @@ from stoobly_agent.app.cli.helpers.context import ReplayContext
 from stoobly_agent.app.settings import Settings
 from stoobly_agent.config.constants import alias_resolve_strategy, env_vars, test_filter, test_strategy
 from stoobly_agent.lib import logger
+from stoobly_agent.lib.utils.conditional_decorator import ConditionalDecorator
 
 from .helpers.scenario_facade import ScenarioFacade
 from .helpers.validations import *
 
+settings = Settings.instance()
+is_remote = settings.cli.features.remote
+
 @click.group(
     epilog="Run 'stoobly-agent scenario COMMAND --help' for more information on a command.",
     help="Manage request scenarios"
@@ -26,8 +30,8 @@ def scenario(ctx):
 @scenario.command(
     help="Create a scenario"
 )
-@click.option('--description', help='Scenario description.')
-@click.option('--project-key', help='Project to create scenario in.')
+@click.option('--description', default='', help='Scenario description.')
+@ConditionalDecorator(lambda f: click.option('--project-key', help='Project to create scenario in.')(f), is_remote)
 @click.option('--select', multiple=True, help='Select column(s) to display.')
 @click.option('--without-headers', is_flag=True, default=False, help='Disable printing column headers.')
 @click.argument('name')
@@ -49,15 +53,15 @@ def create(**kwargs):
 @scenario.command(
     help="Replay a scenario"
 )
-@click.option(
+@ConditionalDecorator(lambda f: click.option(
     '--alias-resolve-strategy', 
     default=alias_resolve_strategy.NONE, 
     type=click.Choice([alias_resolve_strategy.NONE, alias_resolve_strategy.FIFO, alias_resolve_strategy.LIFO]), 
     help='Strategy for resolving dynamic values for aliases.'
-)
-@click.option('--assign', multiple=True, help='Assign alias values. Format: <NAME>=<VALUE>')
+)(f), is_remote)
+@ConditionalDecorator(lambda f: click.option('--assign', multiple=True, help='Assign alias values. Format: <NAME>=<VALUE>')(f), is_remote)
 @click.option('--format', default=DEFAULT_FORMAT, type=click.Choice([DEFAULT_FORMAT, JSON_FORMAT]), help='Format replay response.')
-@click.option('--group-by', help='Repeat for each alias name.')
+@ConditionalDecorator(lambda f: click.option('--group-by', help='Repeat for each alias name.')(f), is_remote)
 @click.option('--host', help='Rewrite request host.')
 @click.option('--lifecycle-hooks-script-path', help='Path to lifecycle hooks script.')
 @click.option(
@@ -69,8 +73,8 @@ def create(**kwargs):
 @click.option('--record', is_flag=True, default=False, help='Replay and record scenario.')
 @click.option('--scenario-key', help='Record to scenario.')
 @click.option('--scheme', help='Rewrite request scheme.')
-@click.option('--trace-id', help='Use existing trace.')
-@click.option('--validate', multiple=True, help='Validate one or more aliases. Format: <NAME>=?<TYPE>')
+@ConditionalDecorator(lambda f: click.option('--trace-id', help='Use existing trace.')(f), is_remote)
+@ConditionalDecorator(lambda f: click.option('--validate', multiple=True, help='Validate one or more aliases. Format: <NAME>=?<TYPE>')(f), is_remote)
 @click.argument('key')
 def replay(**kwargs):
     os.environ[env_vars.LOG_LEVEL] = kwargs['log_level']
@@ -84,7 +88,7 @@ def replay(**kwargs):
 
         validate_scenario_key(kwargs['scenario_key'])
 
-    if len(kwargs['validate']):
+    if 'validate' in kwargs and len(kwargs['validate']):
         validate_aliases(kwargs['validate'], assign=kwargs['assign'], format=kwargs['format'], trace_id=kwargs['trace_id'])
 
     __assign_default_alias_resolve_strategy(kwargs)
@@ -99,86 +103,11 @@ def replay(**kwargs):
     scenario = ScenarioFacade(Settings.instance())
     scenario.replay(kwargs.get('key'), kwargs)
 
-@scenario.command(
-    help="Replay and test a scenario"
-)
-@click.option('--aggregate-failures', default=False, is_flag=True, help='Toggles whether to continue execution on failure.')
-@click.option(
-    '--alias-resolve-strategy', 
-    default=alias_resolve_strategy.NONE, 
-    type=click.Choice([alias_resolve_strategy.NONE, alias_resolve_strategy.FIFO, alias_resolve_strategy.LIFO]), 
-    help='Strategy for resolving dynamic values for aliases.'
-)
-@click.option('--assign', multiple=True, help='Assign alias values. Format: <NAME>=<VALUE>')
-@click.option(
-    '--filter', 
-    default=test_filter.ALL, 
-    type=click.Choice([test_filter.ALL, test_filter.ALIAS, test_filter.LINK]), 
-    help='For iterable responses, selectively test properties.'
-)
-@click.option('--format', default=DEFAULT_FORMAT, type=click.Choice([DEFAULT_FORMAT, JSON_FORMAT]), help='Format replay response.')
-@click.option('--group-by', help='Repeat for each alias name.')
-@click.option('--host', help='Rewrite request host.')
-@click.option('--lifecycle-hooks-script-path', help='Path to lifecycle hooks script.')
-@click.option(
-    '--log-level', default=logger.WARNING, type=click.Choice([logger.DEBUG, logger.INFO, logger.WARNING, logger.ERROR]), 
-    help='''
-        Log levels can be "debug", "info", "warning", or "error"
-    '''
-)
-@click.option('--report-key', help='Save to report.')
-@click.option('--scheme', help='Rewrite request scheme.')
-@click.option(
-    '--strategy', 
-    default=test_strategy.DIFF, 
-    type=click.Choice([test_strategy.CUSTOM, test_strategy.DIFF, test_strategy.FUZZY]), 
-    help='How to test responses.'
-)
-@click.option('--trace-id', help='Use existing trace.')
-@click.option('--validate', multiple=True, help='Validate one or more aliases. Format: <NAME>=?<TYPE>')
-@click.argument('key')
-def test(**kwargs):
-    os.environ[env_vars.LOG_LEVEL] = kwargs['log_level']
-    logger.Logger.reload()
-
-    settings = Settings.instance()
-    scenario_key = validate_scenario_key(kwargs['key'])
-
-    if kwargs.get('report_key'):
-        validate_report_key(kwargs['report_key'])
-
-    if len(kwargs['validate']):
-        validate_aliases(kwargs['validate'], assign=kwargs['assign'], format=kwargs['format'], trace_id=kwargs['trace_id'])
-
-    __assign_default_alias_resolve_strategy(kwargs)
-
-    session_context: SessionContext = { 
-        'aggregate_failures': kwargs['aggregate_failures'], 
-        'passed': 0, 
-        'project_id': scenario_key.project_id, 
-        'test_facade': TestFacade(settings), 
-        'total': 0 
-    }
-
-    kwargs['before_replay'] = lambda context: handle_before_replay(
-      context, kwargs['format']
-    )
-    kwargs['after_replay'] = lambda context: __handle_on_test_response(
-        context, session_context, kwargs['format']
-    )
-
-    scenario = ScenarioFacade(settings)
-    scenario.test(kwargs['key'], kwargs)
-
-    handle_test_session_complete(session_context, kwargs['format'])
-
-    exit_on_failure(session_context, format=kwargs['format'])
-
 @scenario.command(
     help="Show all scenarios"
 )
 @click.option('--page', default=0)
-@click.option('--project-key', help='Project to list scenarios from.')
+@ConditionalDecorator(lambda f: click.option('--project-key', help='Project to create scenario in.')(f), is_remote)
 @click.option('--select', multiple=True, help='Select column(s) to display.')
 @click.option('--size', default=10)
 @click.option('--sort-by', default='created_at', help='created_at|name')
@@ -188,8 +117,10 @@ def list(**kwargs):
     print_options = select_print_options(kwargs)
 
     settings = Settings.instance()
+
     project_key = resolve_project_key_and_validate(kwargs, settings)
-    del kwargs['project_key']
+    if 'project_key' in kwargs:
+        del kwargs['project_key']
 
     scenario = ScenarioFacade(settings)
 
@@ -223,6 +154,82 @@ def show(**kwargs):
 
     print_scenarios([scenario_response], **print_options)
 
+if is_remote:
+    @scenario.command(
+        help="Replay and test a scenario"
+    )
+    @click.option('--aggregate-failures', default=False, is_flag=True, help='Toggles whether to continue execution on failure.')
+    @click.option(
+        '--alias-resolve-strategy', 
+        default=alias_resolve_strategy.NONE, 
+        type=click.Choice([alias_resolve_strategy.NONE, alias_resolve_strategy.FIFO, alias_resolve_strategy.LIFO]), 
+        help='Strategy for resolving dynamic values for aliases.'
+    )
+    @click.option('--assign', multiple=True, help='Assign alias values. Format: <NAME>=<VALUE>')
+    @click.option(
+        '--filter', 
+        default=test_filter.ALL, 
+        type=click.Choice([test_filter.ALL, test_filter.ALIAS, test_filter.LINK]), 
+        help='For iterable responses, selectively test properties.'
+    )
+    @click.option('--format', default=DEFAULT_FORMAT, type=click.Choice([DEFAULT_FORMAT, JSON_FORMAT]), help='Format replay response.')
+    @click.option('--group-by', help='Repeat for each alias name.')
+    @click.option('--host', help='Rewrite request host.')
+    @click.option('--lifecycle-hooks-script-path', help='Path to lifecycle hooks script.')
+    @click.option(
+        '--log-level', default=logger.WARNING, type=click.Choice([logger.DEBUG, logger.INFO, logger.WARNING, logger.ERROR]), 
+        help='''
+            Log levels can be "debug", "info", "warning", or "error"
+        '''
+    )
+    @click.option('--report-key', help='Save to report.')
+    @click.option('--scheme', help='Rewrite request scheme.')
+    @click.option(
+        '--strategy', 
+        default=test_strategy.DIFF, 
+        type=click.Choice([test_strategy.CUSTOM, test_strategy.DIFF, test_strategy.FUZZY]), 
+        help='How to test responses.'
+    )
+    @click.option('--trace-id', help='Use existing trace.')
+    @click.option('--validate', multiple=True, help='Validate one or more aliases. Format: <NAME>=?<TYPE>')
+    @click.argument('key')
+    def test(**kwargs):
+        os.environ[env_vars.LOG_LEVEL] = kwargs['log_level']
+        logger.Logger.reload()
+
+        settings = Settings.instance()
+        scenario_key = validate_scenario_key(kwargs['key'])
+
+        if kwargs.get('report_key'):
+            validate_report_key(kwargs['report_key'])
+
+        if len(kwargs['validate']):
+            validate_aliases(kwargs['validate'], assign=kwargs['assign'], format=kwargs['format'], trace_id=kwargs['trace_id'])
+
+        __assign_default_alias_resolve_strategy(kwargs)
+
+        session_context: SessionContext = { 
+            'aggregate_failures': kwargs['aggregate_failures'], 
+            'passed': 0, 
+            'project_id': scenario_key.project_id, 
+            'test_facade': TestFacade(settings), 
+            'total': 0 
+        }
+
+        kwargs['before_replay'] = lambda context: handle_before_replay(
+        context, kwargs['format']
+        )
+        kwargs['after_replay'] = lambda context: __handle_on_test_response(
+            context, session_context, kwargs['format']
+        )
+
+        scenario = ScenarioFacade(settings)
+        scenario.test(kwargs['key'], kwargs)
+
+        handle_test_session_complete(session_context, kwargs['format'])
+
+        exit_on_failure(session_context, format=kwargs['format'])
+
 def __handle_on_test_response(replay_context: ReplayContext, session_context: SessionContext, format = None):
     handle_test_complete(replay_context, session_context, format)
 
@@ -231,5 +238,5 @@ def __handle_on_test_response(replay_context: ReplayContext, session_context: Se
 
 def __assign_default_alias_resolve_strategy(kwargs):
     # If we have assigned values to aliases, it's likely we want to also have them resolved
-    if len(kwargs['assign']) > 0 and kwargs['alias_resolve_strategy'] == alias_resolve_strategy.NONE:
+    if 'assign' in kwargs and len(kwargs['assign']) > 0 and kwargs['alias_resolve_strategy'] == alias_resolve_strategy.NONE:
         kwargs['alias_resolve_strategy'] = alias_resolve_strategy.FIFO

stoobly_agent/app/models/adapters/mitmproxy/__init__.py

@@ -0,0 +1,2 @@
+from .request import MitmproxyRequestAdapterFactory
+from .response import MitmproxyResponseAdapterFactory

stoobly_agent/app/models/adapters/mitmproxy/request/__init__.py

@@ -0,0 +1,12 @@
+import pdb
+import requests
+
+from .python_adapter import PythonRequestAdapter
+
+class MitmproxyRequestAdapterFactory():
+
+  def __init__(self, request: requests.Request):
+    self.__request = request
+
+  def python_request(self):
+    return PythonRequestAdapter(self.__request).adapt()

stoobly_agent/app/models/adapters/mitmproxy/request/python_adapter.py

@@ -0,0 +1,16 @@
+import requests
+
+from mitmproxy.http import Request as MitmproxyRequest
+
+class PythonRequestAdapter():
+  
+  def __init__(self, request: MitmproxyRequest):
+    self.__request = request
+
+  def adapt(self):
+    return requests.Request(
+      method=self.__request.method,
+      url=self.__request.url,
+      headers=self.__request.headers,
+      data=self.__request.content
+    )

stoobly_agent/app/models/adapters/mitmproxy/response/__init__.py

@@ -0,0 +1,12 @@
+import pdb
+import requests
+
+from .python_adapter import PythonResponseAdapter
+
+class MitmproxyResponseAdapterFactory():
+
+  def __init__(self, response: requests.Response):
+    self.__response = response
+
+  def python_response(self):
+    return PythonResponseAdapter(self.__response).adapt()