From 531b07cd506d3710d4d5f26a8a8955309815a75d Mon Sep 17 00:00:00 2001 From: Oleg Pudeyev Date: Mon, 26 Aug 2024 15:16:08 -0400 Subject: [PATCH] integration test --- .../di/integration/instrumentation_spec.rb | 18 +++++++++--------- 1 file changed, 9 insertions(+), 9 deletions(-) diff --git a/spec/datadog/di/integration/instrumentation_spec.rb b/spec/datadog/di/integration/instrumentation_spec.rb index aae17bed5e7..234453a03fc 100644 --- a/spec/datadog/di/integration/instrumentation_spec.rb +++ b/spec/datadog/di/integration/instrumentation_spec.rb @@ -66,17 +66,17 @@ it 'invokes probe' do remote_processor.process(probe_rc_spec) -=begin - expect(Datadog::DI::ProbeNotifier).to receive(:notify_executed).and_call_original do |_probe, **opts| - p _probe - p opts[:tracepoint].binding.local_variables - end - expect(Datadog::DI::ProbeNotifier).to receive(:notify_snapshot).and_call_original do |_probe, **opts| - p _probe - p opts[:tracepoint].binding.local_variables + expect(InstrumentationIntegrationTestClass.new.test_method).to eq(42) + end + + it 'assembles expected notification payload' do + remote_processor.process(probe_rc_spec) + payload = nil + expect(Datadog::DI.component.probe_notifier_worker).to receive(:add_snapshot) do |_payload| + payload = _payload end -=end expect(InstrumentationIntegrationTestClass.new.test_method).to eq(42) + expect(payload).to be_a(Hash) end end end