chore: move trace builders to the correct location

Author: heitorlessa

tests/e2e/utils/data_builder/__init__.py

@@ -5,7 +5,9 @@
     build_metric_name,
     build_metric_query_data,
     build_multiple_add_metric_input,
+)
+from tests.e2e.utils.data_builder.traces import (
     build_put_annotations_input,
     build_put_metadata_input,
+    build_trace_default_query,
 )
-from tests.e2e.utils.data_builder.traces import build_trace_default_query

tests/e2e/utils/data_builder/metrics.py

@@ -1,4 +1,4 @@
-from typing import Any, Dict, List, Optional
+from typing import Dict, List, Optional
 
 from mypy_boto3_cloudwatch.type_defs import DimensionTypeDef, MetricDataQueryTypeDef
 
@@ -112,39 +112,5 @@ def build_add_dimensions_input(**dimensions) -> List[DimensionTypeDef]:
     return [{"Name": name, "Value": value} for name, value in dimensions.items()]
 
 
-def build_put_annotations_input(**annotations: str) -> List[Dict]:
-    """Create trace annotations input to be used with Tracer.put_annotation()
-
-    Parameters
-    ----------
-    annotations : str
-        annotations in key=value form
-
-    Returns
-    -------
-    List[Dict]
-        List of put annotations input
-    """
-    return [{"key": key, "value": value} for key, value in annotations.items()]
-
-
-def build_put_metadata_input(namespace: Optional[str] = None, **metadata: Any) -> List[Dict]:
-    """Create trace metadata input to be used with Tracer.put_metadata()
-
-    All metadata will be under `test` namespace
-
-    Parameters
-    ----------
-    metadata : Any
-        metadata in key=value form
-
-    Returns
-    -------
-    List[Dict]
-        List of put metadata input
-    """
-    return [{"key": key, "value": value, "namespace": namespace} for key, value in metadata.items()]
-
-
 def build_metric_name() -> str:
     return f"test_metric{build_random_value()}"

tests/e2e/utils/data_builder/traces.py

@@ -1,2 +1,39 @@
+from typing import Any, Dict, List, Optional
+
+
 def build_trace_default_query(function_name: str) -> str:
     return f'service("{function_name}")'
+
+
+def build_put_annotations_input(**annotations: str) -> List[Dict]:
+    """Create trace annotations input to be used with Tracer.put_annotation()
+
+    Parameters
+    ----------
+    annotations : str
+        annotations in key=value form
+
+    Returns
+    -------
+    List[Dict]
+        List of put annotations input
+    """
+    return [{"key": key, "value": value} for key, value in annotations.items()]
+
+
+def build_put_metadata_input(namespace: Optional[str] = None, **metadata: Any) -> List[Dict]:
+    """Create trace metadata input to be used with Tracer.put_metadata()
+
+    All metadata will be under `test` namespace
+
+    Parameters
+    ----------
+    metadata : Any
+        metadata in key=value form
+
+    Returns
+    -------
+    List[Dict]
+        List of put metadata input
+    """
+    return [{"key": key, "value": value, "namespace": namespace} for key, value in metadata.items()]