chore: fix test coverage

Author: heitorlessa

Diff for: aws_lambda_powertools/tracing/tracer.py

@@ -166,19 +166,9 @@ def __init__(
         if self.auto_patch:
             self.patch(modules=patch_modules)
 
-        if "aws_xray_sdk" in self.provider.__module__:
+        if self._is_xray_provider():
             self._disable_xray_trace_batching()
 
-    def _disable_xray_trace_batching(self):
-        """Configure X-Ray SDK to send subsegment individually over batching
-        Known issue: https://github.com/awslabs/aws-lambda-powertools-python/issues/283
-        """
-        if self.disabled:
-            logger.debug("Tracing has been disabled, aborting streaming override")
-            return
-
-        aws_xray_sdk.core.xray_recorder.configure(streaming_threshold=0)
-
     def put_annotation(self, key: str, value: Union[str, numbers.Number, bool]):
         """Adds annotation to existing segment or subsegment
 
@@ -776,5 +766,15 @@ def _patch_xray_provider(self):
 
         return provider
 
+    def _disable_xray_trace_batching(self):
+        """Configure X-Ray SDK to send subsegment individually over batching
+        Known issue: https://github.com/awslabs/aws-lambda-powertools-python/issues/283
+        """
+        if self.disabled:
+            logger.debug("Tracing has been disabled, aborting streaming override")
+            return
+
+        aws_xray_sdk.core.xray_recorder.configure(streaming_threshold=0)
+
     def _is_xray_provider(self):
         return "aws_xray_sdk" in self.provider.__module__