Making specs works

3 files changed, 76 insertions, 0 deletions

diff --git a/spec/handlers/c/mruby_class_handler_spec.rb b/spec/handlers/c/mruby_class_handler_spec.rb
new file mode 100644
index 0000000..521ad62
--- /dev/null
+++ b/spec/handlers/c/mruby_class_handler_spec.rb
@@ -0,0 +1,32 @@
+require_relative 'spec_helper'
+
+describe YARD::Handlers::C::MRubyClassHandler do
+  it "should register classes" do
+    parse_init 'cFoo = mrb_define_class(mrb, "Foo", mrb->object_class);'
+
+    expect(Registry.at('Foo').type).to be :class
+  end
+
+  it "should register classes under namespaces" do
+    parse_init 'cFoo = mrb_define_class_under(mrb, cBar, "Foo", rb_cObject);'
+    expect(Registry.at('Bar::Foo').type).to be :class
+  end
+
+  it "should remember symbol defined with class" do
+    parse_init(<<-eof)
+      cXYZ = mrb_define_class(mrb, "Foo", mrb->object_class);
+      mrb_define_method(mrb, cXYZ, "bar", bar, MRB_ARGS_NONE());
+    eof
+    expect(Registry.at('Foo').type).to be :class
+    expect(Registry.at('Foo#bar')).not_to be_nil
+  end
+
+  it "should lookup superclass symbol name" do
+    parse_init(<<-eof)
+      cXYZ = mrb_define_class(mrb,"Foo", mrb->object_class);
+      cBar = mrb_define_class(mrb,"Bar", cXYZ);
+    eof
+    expect(Registry.at('Bar').superclass).to eq Registry.at('Foo')
+  end
+
+end
diff --git a/spec/handlers/c/spec_helper.rb b/spec/handlers/c/spec_helper.rb
new file mode 100644
index 0000000..9424e3e
--- /dev/null
+++ b/spec/handlers/c/spec_helper.rb
@@ -0,0 +1,13 @@
+require_relative '../spec_helper'
+
+def parse(src, file = '(stdin)')
+  YARD::Registry.clear
+  parser = YARD::Parser::SourceParser.new(:c)
+  parser.file = file
+  parser.parse(StringIO.new(src))
+end
+
+def parse_init(src)
+  YARD::Registry.clear
+  YARD.parse_string("void mrb_foo_gem_init(mrb_state *mrb) {\n#{src}\n}", :c)
+end
diff --git a/spec/handlers/spec_helper.rb b/spec/handlers/spec_helper.rb
new file mode 100644
index 0000000..1eba194
--- /dev/null
+++ b/spec/handlers/spec_helper.rb
@@ -0,0 +1,31 @@
+require_relative '../spec_helper'
+require 'stringio'
+
+def undoc_error(code)
+  lambda { StubbedSourceParser.parse_string(code) }.should raise_error(Parser::UndocumentableError)
+end
+
+def with_parser(parser_type, &block)
+  tmp = StubbedSourceParser.parser_type
+  StubbedSourceParser.parser_type = parser_type
+  yield
+  StubbedSourceParser.parser_type = tmp
+end
+
+class StubbedProcessor < YARD::Handlers::Processor
+  def process(statements)
+    statements.each_with_index do |stmt, index|
+      find_handlers(stmt).each do |handler|
+        handler.new(self, stmt).process
+      end
+    end
+  end
+end
+
+class StubbedSourceParser < YARD::Parser::SourceParser
+  StubbedSourceParser.parser_type = :ruby
+  def post_process
+    post = StubbedProcessor.new(self)
+    post.process(@parser.enumerator)
+  end
+end