1 files changed, 35 insertions, 0 deletions
diff --git a/test/mkmf/test_have_macro.rb b/test/mkmf/test_have_macro.rb
new file mode 100644
index 0000000000..d070829295
--- /dev/null
+++ b/test/mkmf/test_have_macro.rb
@@ -0,0 +1,35 @@
+require_relative 'base'
+require 'tempfile'
+
+class TestMkmf
+  class TestMacro < TestMkmf
+    MACRO_NAME = "RUBY_MKMFTEST_FOOBAR"
+
+    def test_have_macro_opt
+      assert_equal(true, have_macro(MACRO_NAME, nil, "-D#{MACRO_NAME}"), MKMFLOG)
+    end
+
+    def test_have_macro_header
+      Tempfile.open(%w"test_mkmf .h", ".") do |tmp|
+        tmp.puts("#undef #{MACRO_NAME}")
+        tmp.puts("#define #{MACRO_NAME} 1")
+        tmp.close
+        dir, base = File.split(tmp.path)
+        assert_equal(true, have_macro(MACRO_NAME, base, "-I."), MKMFLOG)
+      end
+    end
+
+    def test_not_have_macro_opt
+      assert_equal(false, have_macro(MACRO_NAME, nil, "-U#{MACRO_NAME}"), MKMFLOG)
+    end
+
+    def test_not_have_macro_header
+      Tempfile.open(%w"test_mkmf .h", ".") do |tmp|
+        tmp.puts("#undef #{MACRO_NAME}")
+        tmp.close
+        dir, base = File.split(tmp.path)
+        assert_equal(false, have_macro(MACRO_NAME, base, "-I."), MKMFLOG)
+      end
+    end
+  end
+end