home *** CD-ROM | disk | FTP | other *** search
/ OS/2 Shareware BBS: 10 Tools / 10-Tools.zip / ruby164.zip / rbemx164.zip / ruby / share / doc / rdtool-0.6.10 / test / test-output-format-visitor.rb < prev    next >
Text File  |  2001-04-15  |  1KB  |  56 lines
  1. require 'runit/testcase'
  2. require 'runit/cui/mytestrunner'
  3. require 'runit/topublic'
  4.  
  5. require 'test/temp-dir'
  6.  
  7. require 'rd/output-format-visitor'
  8. require 'rd/tree'
  9. require 'rd/element'
  10. require 'rd/block-element'
  11. require 'rd/inline-element'
  12. require 'rd/rd-struct'
  13.  
  14. include RD
  15.  
  16. class TestOFVisitor < RUNIT::TestCase
  17.   include RUNIT::ToPublic
  18.   def setup
  19.   end
  20.  
  21.   def test_initialize
  22.     vis = OutputFormatVisitor.new
  23.     assert_equal([], vis.include_suffix)
  24.     vis = SampleVisitor.new
  25.     assert_equal(["html"], vis.include_suffix)    
  26.   end
  27.  
  28.   def test_apply_to_include
  29.     tree = Tree.new_with_document_struct(DocumentStructure::RD,
  30.                      ["test/data/sub", "test/data/"])
  31.     de = DocumentElement.new
  32.     # test/data/includee1.html
  33.     inc1 = Include.new("includee1")
  34.     # test/data/includee2.html, test/data/sub/includee2.html
  35.     inc2 = Include.new("includee2")
  36.     # test/data/includee3.nothtml
  37.     inc3 = Include.new("includee3")
  38.     tree.root = de
  39.     de.add_child(inc1)
  40.     de.add_child(inc2)
  41.     de.add_child(inc3)
  42.     vis = SampleVisitor.new
  43.     assert_equal("includee1.html", vis.apply_to_Include(inc1))
  44.     assert_equal("sub/includee2.html", vis.apply_to_Include(inc2))
  45.     assert_equal(nil, vis.apply_to_Include(inc3))
  46.   end
  47. end
  48.  
  49. class SampleVisitor < OutputFormatVisitor
  50.   INCLUDE_SUFFIX = ["html"]
  51. end
  52.  
  53. if $0 == __FILE__
  54.   RUNIT::CUI::MyTestRunner.run(TestOFVisitor.suite)
  55. end
  56.