1818
1919from opentelemetry ._opamp .agent import OpAMPAgent , _safe_invoke
2020from opentelemetry ._opamp .agent import _Job as Job
21- from opentelemetry ._opamp .callbacks import Callbacks , MessageData
21+ from opentelemetry ._opamp .callbacks import MessageData , OpAMPCallbacks
2222from opentelemetry ._opamp .proto import opamp_pb2
2323
2424
25- class _NoOpCallbacks (Callbacks ):
25+ class _NoOpCallbacks (OpAMPCallbacks ):
2626 pass
2727
2828
@@ -48,7 +48,7 @@ def test_agent_start_will_send_connection_and_disconnetion_messages():
4848 mock_message .flags = 0
4949 client_mock .send .return_value = mock_message
5050
51- cb = mock .create_autospec (Callbacks , instance = True )
51+ cb = mock .create_autospec (OpAMPCallbacks , instance = True )
5252 agent = OpAMPAgent (interval = 30 , client = client_mock , callbacks = cb )
5353 agent .start ()
5454 # wait for the queue to be consumed
@@ -100,7 +100,7 @@ def test_agent_send_warns_without_worker_thread(caplog):
100100def test_agent_retries_before_max_attempts (caplog ):
101101 caplog .set_level (logging .DEBUG , logger = "opentelemetry._opamp.agent" )
102102
103- cb = mock .create_autospec (Callbacks , instance = True )
103+ cb = mock .create_autospec (OpAMPCallbacks , instance = True )
104104 client_mock = mock .Mock ()
105105 connection_message = mock .Mock ()
106106 connection_message .HasField .return_value = False
@@ -136,7 +136,7 @@ def test_agent_retries_before_max_attempts(caplog):
136136def test_agent_stops_after_max_attempts (caplog ):
137137 caplog .set_level (logging .DEBUG , logger = "opentelemetry._opamp.agent" )
138138
139- cb = mock .create_autospec (Callbacks , instance = True )
139+ cb = mock .create_autospec (OpAMPCallbacks , instance = True )
140140 client_mock = mock .Mock ()
141141 connection_message = mock .Mock ()
142142 connection_message .HasField .return_value = False
@@ -171,7 +171,7 @@ def test_agent_stops_after_max_attempts(caplog):
171171
172172
173173def test_agent_send_enqueues_job ():
174- cb = mock .create_autospec (Callbacks , instance = True )
174+ cb = mock .create_autospec (OpAMPCallbacks , instance = True )
175175 client_mock = mock .Mock ()
176176 msg = mock .Mock ()
177177 msg .HasField .return_value = False
@@ -194,7 +194,7 @@ def test_agent_send_enqueues_job():
194194
195195
196196def test_on_error_called_without_on_message_for_error_response ():
197- cb = mock .create_autospec (Callbacks , instance = True )
197+ cb = mock .create_autospec (OpAMPCallbacks , instance = True )
198198 client_mock = mock .Mock ()
199199
200200 error_response = opamp_pb2 .ServerErrorResponse (
@@ -224,7 +224,7 @@ def test_on_error_called_without_on_message_for_error_response():
224224
225225
226226def test_on_error_not_called_without_error_response ():
227- cb = mock .create_autospec (Callbacks , instance = True )
227+ cb = mock .create_autospec (OpAMPCallbacks , instance = True )
228228 client_mock = mock .Mock ()
229229
230230 server_msg = opamp_pb2 .ServerToAgent ()
@@ -255,7 +255,7 @@ def test_dispatch_order_with_error():
255255 error_response = error_response ,
256256 )
257257
258- class OrderTrackingCallbacks (Callbacks ):
258+ class OrderTrackingCallbacks (OpAMPCallbacks ):
259259 def on_connect (self , agent , client ):
260260 call_order .append ("on_connect" )
261261
@@ -286,7 +286,7 @@ def test_dispatch_order_without_error():
286286
287287 server_msg = opamp_pb2 .ServerToAgent ()
288288
289- class OrderTrackingCallbacks (Callbacks ):
289+ class OrderTrackingCallbacks (OpAMPCallbacks ):
290290 def on_connect (self , agent , client ):
291291 call_order .append ("on_connect" )
292292
@@ -311,7 +311,7 @@ def on_error(self, agent, client, error_response):
311311
312312
313313def test_report_full_state_flag_triggers_full_state_send ():
314- cb = mock .create_autospec (Callbacks , instance = True )
314+ cb = mock .create_autospec (OpAMPCallbacks , instance = True )
315315 client_mock = mock .Mock ()
316316
317317 conn_msg = opamp_pb2 .ServerToAgent ()
0 commit comments