open-telemetry
/
opentelemetry-python-contrib
mirror of https://github.com/open-telemetry/opentelemetry-python-contrib.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

cassandra: fix tests for py313 (#3152) 

* cassandra: fix tests to make CI happy for py313

Signed-off-by: emdneto <9735060+emdneto@users.noreply.github.com>

* ruff

Signed-off-by: emdneto <9735060+emdneto@users.noreply.github.com>

---------

Signed-off-by: emdneto <9735060+emdneto@users.noreply.github.com>
This commit is contained in:
Emídio Neto 2024-12-24 12:14:05 -03:00 committed by GitHub
parent d116ae39d6
commit 16eaec8d03
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
instrumentation/opentelemetry-instrumentation-cassandra/tests/test_cassandra_integration.py
View File

@ -43,6 +43,10 @@ class TestCassandraIntegration(TestBase):
with self.disable_logging():
CassandraInstrumentor().uninstrument()
@property
def _mocked_session(self):
return cassandra.cluster.Session(cluster=mock.Mock(), hosts=[])
def test_instrument_uninstrument(self):
instrumentation = CassandraInstrumentor()
instrumentation.instrument()
@ -67,7 +71,7 @@ class TestCassandraIntegration(TestBase):
):
mock_create_response_future.return_value = mock.Mock()
mock_session_init.return_value = None
mock_connect.return_value = cassandra.cluster.Session()
mock_connect.return_value = self._mocked_session
CassandraInstrumentor().instrument()
@ -100,7 +104,7 @@ class TestCassandraIntegration(TestBase):
):
mock_create_response_future.return_value = mock.Mock()
mock_session_init.return_value = None
mock_connect.return_value = cassandra.cluster.Session()
mock_connect.return_value = self._mocked_session
resource = resources.Resource.create({})
result = self.create_tracer_provider(resource=resource)
@ -124,7 +128,7 @@ class TestCassandraIntegration(TestBase):
):
mock_create_response_future.return_value = mock.Mock()
mock_session_init.return_value = None
mock_connect.return_value = cassandra.cluster.Session()
mock_connect.return_value = self._mocked_session
tracer_provider = trace_api.NoOpTracerProvider()
CassandraInstrumentor().instrument(tracer_provider=tracer_provider)