git.oblomov.eu Git - ohcount/blob - test/unit/source_file_test.rb

   1 require File.dirname(__FILE__) + '/../test_helper'
   2 include Ohcount
   3
   4 class SourceFileTest < Ohcount::Test
   5
   6         def test_initialize
   7                 assert SourceFile.new("foo.rb")
   8         end
   9
  10         def test_contents
  11                 contents = "foobar"
  12                 f = SourceFile.new("foo.rb", :contents => contents)
  13                 assert_equal contents, f.contents
  14         end
  15
  16         def test_language_breakdowns
  17                 contents = "x = 5"
  18                 f = SourceFile.new("foo.rb", :contents => contents)
  19                 assert_equal 'ruby', f.language_breakdowns[0].name
  20                 assert_equal contents, f.language_breakdowns('ruby').code
  21         end
  22
  23         def test_parse
  24                 ruby_code = ''
  25                 contents = "x = 5"
  26                 f = SourceFile.new("foo.rb", :contents => contents)
  27                 f.parse do |l, s, line|
  28                         ruby_code << line if l == :ruby && s == :code
  29                 end
  30                 assert_equal contents, ruby_code
  31         end
  32 end
  33