diff --git a/catalystwan/integration_tests/profile_builder/test_pb_uc_voice.py b/catalystwan/integration_tests/profile_builder/test_pb_uc_voice.py index 8bced1e5..0d77ccd7 100644 --- a/catalystwan/integration_tests/profile_builder/test_pb_uc_voice.py +++ b/catalystwan/integration_tests/profile_builder/test_pb_uc_voice.py @@ -7,8 +7,12 @@ TranslationProfileParcel, TranslationRuleParcel, ) +from catalystwan.models.configuration.feature_profile.sdwan.uc_voice.analog_interface import ( + Association, + ModuleType, + SlotId, +) from catalystwan.models.configuration.feature_profile.sdwan.uc_voice.translation_rule import Action, RuleSettings -from catalystwan.models.configuration.feature_profile.sdwan.uc_voice.analog_interface import Association, SlotId, ModuleType from catalystwan.tests.builders.uc_voice import as_default @@ -21,7 +25,9 @@ def setUp(self) -> None: feature_profile=FeatureProfileCreationPayload(name=self.fp_name, description=self.fp_description) ) self.api = self.session.api.sdwan_feature_profiles.transport - self.tp = TranslationProfileParcel(parcel_name="TPP", parcel_description="TTP_Desc", translation_profile_settings=[]) + self.tp = TranslationProfileParcel( + parcel_name="TPP", parcel_description="TTP_Desc", translation_profile_settings=[] + ) self.tr_calling = TranslationRuleParcel( parcel_name="2", parcel_description="desc", @@ -57,7 +63,9 @@ def test_when_build_profile_with_translation_profile_and_rules_expect_success(se # Assert assert len(report.failed_parcels) == 0 - def test_when_build_profile_with_analog_interface_and_translation_profile_and_rules_assosiations_expect_success(self): + def test_when_build_profile_with_analog_interface_and_translation_profile_and_rules_assosiations_expect_success( + self, + ): # Arrange ai = AnalogInterfaceParcel( parcel_name="Ai", @@ -71,9 +79,9 @@ def test_when_build_profile_with_analog_interface_and_translation_profile_and_ru translation_profile=RefIdItem(ref_id=as_global("TPP")), trunk_group=RefIdItem(ref_id=as_default(None)), trunk_group_priority=as_default(None), - translation_rule_direction=as_default(None) + translation_rule_direction=as_default(None), ) - ] + ], ) self.builder.add_translation_profile(self.tp, self.tr_calling, self.tr_called) self.builder.add_parcel_with_associations(ai)