rebrandly-otel 0.2.5__tar.gz → 0.2.6__tar.gz

{rebrandly_otel-0.2.5 → rebrandly_otel-0.2.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: rebrandly_otel
-Version: 0.2.5
+Version: 0.2.6
 Summary: Python OTEL wrapper by Rebrandly
 Home-page: https://github.com/rebrandly/rebrandly-otel-python
 Author: Antonio Romano
@@ -502,6 +502,42 @@ Environment configuration:
 ### More examples
 You can find More examples [here](examples)
 
+## Testing
+
+### Running Tests
+
+The test suite uses [pytest](https://docs.pytest.org/).
+
+Run all tests:
+```bash
+pytest
+```
+
+Run specific test file:
+```bash
+pytest tests/test_flask_support.py -v
+pytest tests/test_fastapi_support.py -v
+pytest tests/test_usage.py -v
+pytest tests/test_pymysql_instrumentation.py -v
+pytest tests/test_metrics_and_logs.py -v
+pytest tests/test_decorators.py -v
+```
+
+Run with coverage:
+```bash
+pytest --cov=src --cov-report=html
+```
+
+### Test Coverage
+
+The test suite includes:
+- **Integration tests** (`test_usage.py`): Core OTEL functionality, Lambda handlers, message processing
+- **Flask integration tests** (`test_flask_support.py`): Flask setup and hooks
+- **FastAPI integration tests** (`test_fastapi_support.py`): FastAPI setup and middleware
+- **PyMySQL instrumentation tests** (`test_pymysql_instrumentation.py`): Database connection instrumentation, query tracing, helper functions
+- **Metrics and logs tests** (`test_metrics_and_logs.py`): Custom metrics creation (counter, histogram, gauge), logging levels (info, warning, debug, error)
+- **Decorators tests** (`test_decorators.py`): Lambda handler decorator, AWS message handler decorator, traces decorator, aws_message_span context manager
+
 ## License
 
 Rebrandly Python SDK is released under the MIT License.

{rebrandly_otel-0.2.5 → rebrandly_otel-0.2.6}/README.md

@@ -475,6 +475,42 @@ Environment configuration:
 ### More examples
 You can find More examples [here](examples)
 
+## Testing
+
+### Running Tests
+
+The test suite uses [pytest](https://docs.pytest.org/).
+
+Run all tests:
+```bash
+pytest
+```
+
+Run specific test file:
+```bash
+pytest tests/test_flask_support.py -v
+pytest tests/test_fastapi_support.py -v
+pytest tests/test_usage.py -v
+pytest tests/test_pymysql_instrumentation.py -v
+pytest tests/test_metrics_and_logs.py -v
+pytest tests/test_decorators.py -v
+```
+
+Run with coverage:
+```bash
+pytest --cov=src --cov-report=html
+```
+
+### Test Coverage
+
+The test suite includes:
+- **Integration tests** (`test_usage.py`): Core OTEL functionality, Lambda handlers, message processing
+- **Flask integration tests** (`test_flask_support.py`): Flask setup and hooks
+- **FastAPI integration tests** (`test_fastapi_support.py`): FastAPI setup and middleware
+- **PyMySQL instrumentation tests** (`test_pymysql_instrumentation.py`): Database connection instrumentation, query tracing, helper functions
+- **Metrics and logs tests** (`test_metrics_and_logs.py`): Custom metrics creation (counter, histogram, gauge), logging levels (info, warning, debug, error)
+- **Decorators tests** (`test_decorators.py`): Lambda handler decorator, AWS message handler decorator, traces decorator, aws_message_span context manager
+
 ## License
 
 Rebrandly Python SDK is released under the MIT License.

{rebrandly_otel-0.2.5 → rebrandly_otel-0.2.6}/rebrandly_otel.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: rebrandly_otel
-Version: 0.2.5
+Version: 0.2.6
 Summary: Python OTEL wrapper by Rebrandly
 Home-page: https://github.com/rebrandly/rebrandly-otel-python
 Author: Antonio Romano
@@ -502,6 +502,42 @@ Environment configuration:
 ### More examples
 You can find More examples [here](examples)
 
+## Testing
+
+### Running Tests
+
+The test suite uses [pytest](https://docs.pytest.org/).
+
+Run all tests:
+```bash
+pytest
+```
+
+Run specific test file:
+```bash
+pytest tests/test_flask_support.py -v
+pytest tests/test_fastapi_support.py -v
+pytest tests/test_usage.py -v
+pytest tests/test_pymysql_instrumentation.py -v
+pytest tests/test_metrics_and_logs.py -v
+pytest tests/test_decorators.py -v
+```
+
+Run with coverage:
+```bash
+pytest --cov=src --cov-report=html
+```
+
+### Test Coverage
+
+The test suite includes:
+- **Integration tests** (`test_usage.py`): Core OTEL functionality, Lambda handlers, message processing
+- **Flask integration tests** (`test_flask_support.py`): Flask setup and hooks
+- **FastAPI integration tests** (`test_fastapi_support.py`): FastAPI setup and middleware
+- **PyMySQL instrumentation tests** (`test_pymysql_instrumentation.py`): Database connection instrumentation, query tracing, helper functions
+- **Metrics and logs tests** (`test_metrics_and_logs.py`): Custom metrics creation (counter, histogram, gauge), logging levels (info, warning, debug, error)
+- **Decorators tests** (`test_decorators.py`): Lambda handler decorator, AWS message handler decorator, traces decorator, aws_message_span context manager
+
 ## License
 
 Rebrandly Python SDK is released under the MIT License.

{rebrandly_otel-0.2.5 → rebrandly_otel-0.2.6}/rebrandly_otel.egg-info/SOURCES.txt

@@ -16,4 +16,9 @@ src/otel_utils.py
 src/pymysql_instrumentation.py
 src/rebrandly_otel.py
 src/traces.py
+tests/test_decorators.py
+tests/test_fastapi_support.py
+tests/test_flask_support.py
+tests/test_metrics_and_logs.py
+tests/test_pymysql_instrumentation.py
 tests/test_usage.py

{rebrandly_otel-0.2.5 → rebrandly_otel-0.2.6}/setup.py

@@ -5,7 +5,7 @@ with open("README.md", "r") as fh:
 
 setuptools.setup(
     name="rebrandly_otel",
-    version="0.2.5",
+    version="0.2.6",
     author="Antonio Romano",
     author_email="antonio@rebrandly.com",
     description="Python OTEL wrapper by Rebrandly",

{rebrandly_otel-0.2.5 → rebrandly_otel-0.2.6}/src/fastapi_support.py

@@ -1,7 +1,7 @@
 # fastapi_integration.py
 """FastAPI integration for Rebrandly OTEL SDK."""
 import json
-from rebrandly_otel import Status, StatusCode, SpanKind
+from opentelemetry.trace import Status, StatusCode, SpanKind
 from .http_utils import filter_important_headers
 from fastapi import HTTPException, Depends
 from starlette.requests import Request

{rebrandly_otel-0.2.5 → rebrandly_otel-0.2.6}/src/flask_support.py

@@ -2,7 +2,7 @@
 """Flask integration for Rebrandly OTEL SDK."""
 
 import json
-from rebrandly_otel import Status, StatusCode, SpanKind
+from opentelemetry.trace import Status, StatusCode, SpanKind
 from .http_utils import filter_important_headers
 
 from flask import request, jsonify

{rebrandly_otel-0.2.5 → rebrandly_otel-0.2.6}/src/http_utils.py

@@ -18,7 +18,8 @@ def filter_important_headers(headers):
         'x-forwarded-proto',
         'x-request-id',
         'x-correlation-id',
-        'x-trace-id'
+        'x-trace-id',
+        'user-agent'
     ]
 
     filtered = {}

{rebrandly_otel-0.2.5 → rebrandly_otel-0.2.6}/src/rebrandly_otel.py

@@ -73,7 +73,7 @@ class RebrandlyOTEL:
              kind: SpanKind = SpanKind.INTERNAL,
              message=None):
         """Create a span using context manager."""
-        with self.tracer.start_span(name, attributes=attributes, kind=kind) as span:
+        with self.tracer.start_span(name=name, attributes=attributes, kind=kind) as span:
             try:
                 yield span
                 span.set_status(Status(StatusCode.OK))
@@ -510,7 +510,13 @@ class RebrandlyOTEL:
                 attributes=combined_attributes,
                 kind=kind
         ) as span:
-            yield span
+            try:
+                yield span
+                span.set_status(Status(StatusCode.OK))
+            except Exception as e:
+                span.record_exception(e)
+                span.set_status(Status(StatusCode.ERROR, str(e)))
+                raise
 
 
 # Create Singleton instance