Set RUBY_MEMCHECK_RUNNING environment variable

Shopify · Aug 4, 2023 · d415e82 · d415e82
1 parent a295057
commit d415e82
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/lib/ruby_memcheck/test_task_reporter.rb b/lib/ruby_memcheck/test_task_reporter.rb
@@ -20,6 +20,7 @@ def run_ruby_with_valgrind(&block)
 
     def setup
       ENV["RUBY_MEMCHECK_LOADED_FEATURES_FILE"] = File.expand_path(configuration.loaded_features_file)
+      ENV["RUBY_MEMCHECK_RUNNING"] = "1"
     end
 
     def report_valgrind_errors

diff --git a/test/ruby_memcheck/shared_test_task_reporter_tests.rb b/test/ruby_memcheck/shared_test_task_reporter_tests.rb
@@ -223,6 +223,18 @@ def test_test_helper_is_loaded
       end
     end
 
+    def test_envionment_variable_RUBY_MEMCHECK_RUNNING
+      Tempfile.create do |tempfile|
+        ok = run_with_memcheck(<<~RUBY, raise_on_failure: false)
+          File.write(#{tempfile.path.inspect}, ENV["RUBY_MEMCHECK_RUNNING"])
+        RUBY
+
+        assert(ok)
+        assert_empty(@test_task.reporter.errors)
+        assert_includes(tempfile.read, "1")
+      end
+    end
+
     private
 
     def run_with_memcheck(code, raise_on_failure: true, spawn_opts: {})