Add directive specs for lib objects

spec/compiler/crystal/tools/doc/directives_spec.cr

@@ -47,6 +47,21 @@ describe Crystal::Doc::Generator do
       a_def.visibility.should eq("private")
     end
 
+    it "shows documentation for nested objects if a lib is marked with :showdoc:" do
+      program = top_level_semantic(<<-CRYSTAL, wants_doc: true).program
+        # :showdoc:
+        lib Foo
+          # docs for `foo`
+          fun foo
+        end
+        CRYSTAL
+
+      generator = Doc::Generator.new program, [""]
+
+      generator.must_include?(program.types["Foo"]).should be_true
+      generator.type(program.types["Foo"]).lookup_method("foo").should_not be_nil
+    end
+
     it "does not include documentation for methods within a :nodoc: namespace" do
       program = top_level_semantic(<<-CRYSTAL, wants_doc: true).program
         # :nodoc:
@@ -91,6 +106,19 @@ describe Crystal::Doc::Generator do
       generator.must_include?(generator.type(program.types["Foo"]).lookup_path("Baz")).should be_false
     end
 
+    it "does not include documentation for a :showdoc: fun inside a lib not marked with :showdoc:" do
+      program = top_level_semantic(<<-CRYSTAL, wants_doc: true).program
+        lib Foo
+          # :showdoc:
+          fun foo
+        end
+        CRYSTAL
+
+      generator = Doc::Generator.new program, [""]
+
+      generator.must_include?(program.types["Foo"]).should be_false
+    end
+
     it "doesn't show a method marked :nodoc: within a :showdoc: namespace" do
       program = top_level_semantic(<<-CRYSTAL, wants_doc: true).program
         # :showdoc:
@@ -105,5 +133,22 @@ describe Crystal::Doc::Generator do
       generator = Doc::Generator.new program, [""]
       generator.type(program.types["Foo"]).lookup_method("foo").should be_nil
     end
+
+    it "doesn't show a fun marked :nodoc: within a :showdoc: lib" do
+      program = top_level_semantic(<<-CRYSTAL, wants_doc: true).program
+        # :showdoc:
+        lib Foo
+          # :nodoc:
+          # Some docs for `foo`
+          fun foo
+
+          fun bar
+        end
+        CRYSTAL
+
+      generator = Doc::Generator.new program, [""]
+      generator.type(program.types["Foo"]).lookup_method("foo").should be_nil
+      generator.type(program.types["Foo"]).lookup_method("bar").should_not be_nil
+    end
   end
 end