integration test

Author: p

spec/datadog/di/integration/probe_notifier_worker_spec.rb

@@ -0,0 +1,115 @@
+require "datadog/di/spec_helper"
+require 'datadog/di'
+require 'webrick'
+
+RSpec.describe Datadog::DI::ProbeNotifierWorker do
+  let(:worker) do
+    described_class.new(settings, transport, logger)
+  end
+
+  let(:transport) do
+    Datadog::DI::Transport.new(agent_settings)
+  end
+
+  let(:settings) do
+    Datadog::Core::Configuration::Settings.new.tap do |settings|
+      settings.remote.enabled = true
+      settings.dynamic_instrumentation.enabled = true
+      settings.dynamic_instrumentation.internal.development = true
+      settings.dynamic_instrumentation.internal.propagate_all_exceptions = true
+    end
+  end
+
+  let(:agent_settings) do
+    instance_double_agent_settings.tap do |agent_settings|
+      allow(agent_settings).to receive(:hostname).and_return('localhost')
+      allow(agent_settings).to receive(:port).and_return(test_port)
+      allow(agent_settings).to receive(:timeout_seconds).and_return(1)
+      allow(agent_settings).to receive(:ssl).and_return(false)
+    end
+  end
+  
+  let(:test_port) { 48485 }
+
+  di_logger_double
+
+  let(:installed_payload) do
+    {ddsource: 'dd_debugger',
+     debugger: {
+       diagnostics: {
+         parentId: nil,
+         probeId: String,
+         probeVersion: 0,
+         runtimeId: 'test runtime id',
+         status: 'INSTALLED',
+       }
+     },
+     message: 'test message',
+     service: 'rspec',
+     timestamp: 1234567890,}.freeze
+  end
+  
+  let(:diagnostics_payloads) { [] }
+  let(:input_payloads) { [] }
+
+  let(:server) do
+    WEBrick::HTTPServer.new(
+      Port: test_port,
+    ).tap do |server|
+      @received_snapshot_count = 0
+      @received_snapshot_bytes = 0
+
+      server.mount_proc('/debugger/v1/diagnostics') do |req, res|
+        # This request is a multipart form post
+        expect(req.content_type).to match(%r,^multipart/form-data;,)
+        diagnostics_payloads << req.body
+      end
+
+      server.mount_proc('/debugger/v1/input') do |req, res|
+        payload = JSON.parse(req.body)
+        input_payloads << payload
+      end
+    end
+  end
+
+  around do |example|
+    @server_thread = Thread.new do
+      server.start
+    end
+    loop do
+      break if server.status == :Running || !@server_thread.alive?
+      sleep 0.5
+    end
+    expect(@server_thread).to be_alive
+    example.run
+    @server_thread.kill
+    loop do
+      break unless @server_thread.alive?
+      sleep 0.5
+    end
+  end
+
+  after do
+    worker.stop
+  end
+
+  context 'probe status' do
+    it 'sends expected payload' do
+      worker.add_status(installed_payload)
+      worker.flush
+      expect(worker.send(:thread)).to be_alive
+      
+      expect(diagnostics_payloads.length).to be 1
+      expect(diagnostics_payloads.first.gsub("\r\n", "\n").strip).to match(%r~\
+----[-\w]+
+Content-Disposition: form-data; name="event"; filename="event.json"
+Content-Length: 226
+Content-Type: application/json
+Content-Transfer-Encoding: binary
+
+\[{"ddsource":"dd_debugger","debugger":{"diagnostics":{"parentId":null,"probeId":"String","probeVersion":0,"runtimeId":"test runtime id","status":"INSTALLED"}},"message":"test message","service":"rspec","timestamp":1234567890}\]
+----[-\w]+\
+~)
+    end
+  end
+end