Add header_decl_handler_spec

2 files changed, 16 insertions, 1 deletions

diff --git a/lib/yard/mruby/handlers/c/header/header_decl_handler.rb b/lib/yard/mruby/handlers/c/header/header_decl_handler.rb
index 00c8e11..6649be1 100644
--- a/lib/yard/mruby/handlers/c/header/header_decl_handler.rb
+++ b/lib/yard/mruby/handlers/c/header/header_decl_handler.rb
@@ -1,7 +1,7 @@
 module YARD::MRuby::Handlers::C::Header
   class HeaderDeclHandler < Base
     MATCH = /
-      ^MRB_BEGIN_DECL
+      MRB_BEGIN_DECL
     /mx
 
     handles MATCH
diff --git a/spec/handlers/c/header/header_decl_handler_spec.rb b/spec/handlers/c/header/header_decl_handler_spec.rb
new file mode 100644
index 0000000..4b53971
--- /dev/null
+++ b/spec/handlers/c/header/header_decl_handler_spec.rb
@@ -0,0 +1,15 @@
+require_relative 'spec_helper'
+
+describe YARD::MRuby::Handlers::C::Header::HeaderDeclHandler do
+  it "should find docstrings attached to MRB_BEGIN_DECL tag" do
+    header <<-eof, 'header.h'
+      /* DOCSTRING */
+      MRB_BEGIN_DECL
+
+      MRB_END_DECL
+    eof
+
+    header = Registry.at('header.h')
+    expect(header.docstring).to eq 'DOCSTRING'
+  end
+end