add new methods: Book#catalog=(catalog) and Catalog.new(obj)

kmuto · Mar 5, 2016 · 93691d0 · 93691d0
1 parent aaaa19c
commit 93691d0
Show file tree

Hide file tree

Showing 3 changed files with 26 additions and 1 deletion.
diff --git a/lib/review/book/base.rb b/lib/review/book/base.rb
@@ -201,6 +201,10 @@ def catalog
         @catalog
       end
 
+      def catalog=(catalog)
+        @catalog = catalog
+      end
+
       def read_CHAPS
         if catalog
           catalog.chaps

diff --git a/lib/review/catalog.rb b/lib/review/catalog.rb
@@ -3,7 +3,11 @@
 module ReVIEW
   class Catalog
     def initialize(file)
-      @yaml = YAML.load(file.read)
+      if file.respond_to? :read
+        @yaml = YAML.load(file.read)
+      else  ## as Object
+        @yaml = file
+      end
       @yaml ||= {}
     end
 

diff --git a/test/test_catalog.rb b/test/test_catalog.rb
@@ -81,6 +81,16 @@ def test_postdef
     assert_equal(exp.chomp, sut.postdef)
   end
 
+  def test_from_object
+    sut = Catalog.new(yaml_hash)
+    exp =<<-EOS
+ch01.re
+ch02.re
+    EOS
+    assert_equal(exp.chomp, sut.chaps)
+  end
+
+
   private
   def yaml
     StringIO.new <<-EOS
@@ -103,6 +113,13 @@ def yaml
     EOS
   end
 
+  def yaml_hash
+    {"PREDEF" => %w(pre01.re pre02.re),
+     "CHAPS" => %w(ch01.re ch02.re),
+     "APPENDIX" => %w(post01.re post02.re),
+     "POSTDEF" => %w(back01.re back02.re)}
+  end
+
   def yaml_with_parts
     StringIO.new <<-EOS
 CHAPS: