Ejemplo n.º 1
0
 def __init__(self, method_name, config_file=None, recording_name=None,
              recording_processors=None, replay_processors=None, recording_patches=None, replay_patches=None):
     self.name_replacer = GeneralNameReplacer()
     super(ScenarioTest, self).__init__(
         method_name,
         config_file=config_file,
         recording_processors=recording_processors or [
             SubscriptionRecordingProcessor(MOCKED_SUBSCRIPTION_ID),
             OAuthRequestResponsesFilter(),
             LargeRequestBodyProcessor(),
             LargeResponseBodyProcessor(),
             DeploymentNameReplacer(),
             RequestUrlNormalizer(),
             self.name_replacer,
         ],
         replay_processors=replay_processors or [
             LargeResponseBodyReplacer(),
             DeploymentNameReplacer(),
             RequestUrlNormalizer(),
         ],
         recording_patches=recording_patches or [patch_main_exception_handler],
         replay_patches=replay_patches or [
             patch_main_exception_handler,
             patch_time_sleep_api,
             patch_long_run_operation_delay,
             patch_load_cached_subscriptions,
             patch_retrieve_token_for_user,
             patch_progress_controller,
         ],
         recording_dir=find_recording_dir(inspect.getfile(self.__class__)),
         recording_name=recording_name
     )
Ejemplo n.º 2
0
    def __init__(self,
                 method_name,
                 config_file=None,
                 recording_name=None,
                 recording_processors=None,
                 replay_processors=None,
                 recording_patches=None,
                 replay_patches=None):
        self.cli_ctx = get_dummy_cli()
        self.name_replacer = GeneralNameReplacer()
        self.kwargs = {}
        self.test_guid_count = 0
        default_recording_processors = [
            SubscriptionRecordingProcessor(MOCKED_SUBSCRIPTION_ID),
            OAuthRequestResponsesFilter(),
            LargeRequestBodyProcessor(),
            LargeResponseBodyProcessor(),
            DeploymentNameReplacer(),
            RequestUrlNormalizer(), self.name_replacer,
            StorageAccountKeyReplacer()
        ]

        default_replay_processors = [
            LargeResponseBodyReplacer(),
            DeploymentNameReplacer(),
            RequestUrlNormalizer(),
        ]

        default_recording_patches = [patch_main_exception_handler]

        default_replay_patches = [
            patch_main_exception_handler,
            patch_time_sleep_api,
            patch_long_run_operation_delay,
            patch_load_cached_subscriptions,
            patch_retrieve_token_for_user,
            patch_progress_controller,
        ]

        def _merge_lists(base, patches):
            merged = list(base)
            if patches and not isinstance(patches, list):
                patches = [patches]
            if patches:
                merged = list(set(merged).union(set(patches)))
            return merged

        super(ScenarioTest, self).__init__(
            method_name,
            config_file=config_file,
            recording_processors=_merge_lists(default_recording_processors,
                                              recording_processors),
            replay_processors=_merge_lists(default_replay_processors,
                                           replay_processors),
            recording_patches=_merge_lists(default_recording_patches,
                                           recording_patches),
            replay_patches=_merge_lists(default_replay_patches,
                                        replay_patches),
            recording_dir=find_recording_dir(inspect.getfile(self.__class__)),
            recording_name=recording_name)
Ejemplo n.º 3
0
 def _get_recording_processors(self):
     return [
         self.scrubber,
         OAuthRequestResponsesFilter(),
         DeploymentNameReplacer(),
         RequestUrlNormalizer()
     ]