Sorting tons of stuff around

1 files changed, 41 insertions, 0 deletions

diff --git a/spec/handlers/c/source/class_handler_spec.rb b/spec/handlers/c/source/class_handler_spec.rb
new file mode 100644
index 0000000..bfac3b6
--- /dev/null
+++ b/spec/handlers/c/source/class_handler_spec.rb
@@ -0,0 +1,41 @@
+require_relative 'spec_helper'
+
+describe YARD::MRuby::Handlers::C::Source::ClassHandler 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
+
+  it "should associate declaration comments as class docstring" do
+    parse_init(<<-eof)
+      /* Docstring! */
+      cFoo = mrb_define_class(mrb, "Foo", mrb->object_class);
+    eof
+
+    expect(Registry.at('Foo').docstring).not_to be_blank
+  end
+
+end