2 files changed, 22 insertions, 2 deletions
diff --git a/spec/handlers/c/header/define_handler_spec.rb b/spec/handlers/c/header/define_handler_spec.rb
new file mode 100644
index 0000000..c37e347
--- /dev/null
+++ b/spec/handlers/c/header/define_handler_spec.rb
@@ -0,0 +1,22 @@
+require_relative 'spec_helper'
+
+describe YARD::MRuby::Handlers::C::Header::DefineHandler do
+  it "should register defines" do
+    header_line <<-eof
+      # define MRB_FOO bar
+    eof
+
+    puts Registry.send(:thread_local_store).inspect
+    expect(Registry.at('MRB_FOO')).not_to be_nil
+  end
+
+  it "should find docstrings attached to defines" do
+    header_line <<-eof
+      /* DOCSTRING */
+      # define MRB_FOO bar
+    eof
+
+    define = Registry.at('MRB_FOO')
+    expect(define.docstring).to eq 'DOCSTRING'
+  end
+end
diff --git a/spec/handlers/c/header/function_handler_spec.rb b/spec/handlers/c/header/function_handler_spec.rb
index 2270584..dc42c91 100644
--- a/spec/handlers/c/header/function_handler_spec.rb
+++ b/spec/handlers/c/header/function_handler_spec.rb
@@ -6,8 +6,6 @@ describe YARD::MRuby::Handlers::C::Header::FunctionHandler do
       MRB_API void mrb_foo( void );
     eof
     expect(Registry.at('mrb_foo')).not_to be_nil
-
-    # puts Registry.send(:thread_local_store).inspect
   end
 
   it "should find docstrings attached to functions" do