From 08743d99db767cfc8db3a937e8fa96908c895537 Mon Sep 17 00:00:00 2001
From: Oleg Pudeyev <156273877+p-datadog@users.noreply.github.com>
Date: Tue, 14 Jan 2025 10:52:32 -0500
Subject: [PATCH] DEBUG-3316 verify code tracking works in forked processes
 (#4284)

---
 spec/datadog/di/code_tracker_spec.rb         | 37 ++++++++++++++++++++
 spec/datadog/di/code_tracker_test_class_4.rb |  2 ++
 spec/datadog/di/code_tracker_test_class_5.rb |  2 ++
 3 files changed, 41 insertions(+)
 create mode 100644 spec/datadog/di/code_tracker_test_class_4.rb
 create mode 100644 spec/datadog/di/code_tracker_test_class_5.rb

diff --git a/spec/datadog/di/code_tracker_spec.rb b/spec/datadog/di/code_tracker_spec.rb
index 896eef6e6dc..053ce338783 100644
--- a/spec/datadog/di/code_tracker_spec.rb
+++ b/spec/datadog/di/code_tracker_spec.rb
@@ -97,6 +97,43 @@
         expect(tracker.send(:registry)).to be_empty
       end
     end
+
+    context 'when process forks' do
+      it 'continues tracking in the fork' do
+        # Load rspec assertion code
+        expect(1).to eq(1)
+        expect(1).to equal(1)
+
+        expect(tracker.send(:registry)).to be_empty
+        tracker.start
+
+        require_relative 'code_tracker_test_class_4'
+        expect(tracker.send(:registry).length).to eq(1)
+        path = tracker.send(:registry).to_a.dig(0, 0)
+        expect(File.basename(path)).to eq("code_tracker_test_class_4.rb")
+
+        expect_in_fork do
+          expect(tracker.send(:registry).length).to eq(1)
+          path = tracker.send(:registry).to_a.dig(0, 0)
+          expect(File.basename(path)).to eq("code_tracker_test_class_4.rb")
+
+          require_relative 'code_tracker_test_class_5'
+          expect(tracker.send(:registry).length).to eq(2)
+          path = tracker.send(:registry).to_a.dig(1, 0)
+          expect(File.basename(path)).to eq("code_tracker_test_class_5.rb")
+        end
+
+        begin
+          Process.waitpid
+        rescue Errno::ECHILD
+        end
+
+        # Verify parent did not change
+        expect(tracker.send(:registry).length).to eq(1)
+        path = tracker.send(:registry).to_a.dig(0, 0)
+        expect(File.basename(path)).to eq("code_tracker_test_class_4.rb")
+      end
+    end
   end
 
   describe "#active?" do
diff --git a/spec/datadog/di/code_tracker_test_class_4.rb b/spec/datadog/di/code_tracker_test_class_4.rb
new file mode 100644
index 00000000000..99c8cc88a08
--- /dev/null
+++ b/spec/datadog/di/code_tracker_test_class_4.rb
@@ -0,0 +1,2 @@
+class CodeTrackerTestClass4
+end
diff --git a/spec/datadog/di/code_tracker_test_class_5.rb b/spec/datadog/di/code_tracker_test_class_5.rb
new file mode 100644
index 00000000000..fd32977582e
--- /dev/null
+++ b/spec/datadog/di/code_tracker_test_class_5.rb
@@ -0,0 +1,2 @@
+class CodeTrackerTestClass5
+end