require 'linguist/language'

require 'test/unit'

class TestLanguage < Test::Unit::TestCase
  include Linguist

  def test_find_by_name
    ruby = Language['Ruby']
    assert_equal ruby, Language.find_by_name('Ruby')
    assert_equal ruby, Language.find_by_name('ruby')
    assert_equal ruby, Language.find_by_name('RUBY')
  end

  def test_find_all_by_name
    Language.all.each do |language|
      assert_equal language, Language.find_by_name(language.name)
    end
  end

  def test_find_by_extension
    assert_equal Language['Ruby'], Language.find_by_extension('.rb')
    assert_equal Language['Ruby'], Language.find_by_extension('rb')
    assert_nil Language.find_by_extension('.kt')
  end

  def test_find_all_by_extension
    Language.all.each do |language|
      language.extensions.each do |extension|
        assert_equal language, Language.find_by_extension(extension)
      end
    end
  end

  def test_find_by_lexer
    assert_equal Language['C'], Language.find_by_lexer('c')
    assert_equal Language['C++'], Language.find_by_lexer('cpp')
    assert_equal Language['Java'], Language.find_by_lexer('java')
    assert_equal Language['JavaScript'], Language.find_by_lexer('javascript')
    assert_equal Language['OCaml'], Language.find_by_lexer('ocaml')
    assert_equal Language['Perl'], Language.find_by_lexer('perl')
    assert_equal Language['Python'], Language.find_by_lexer('python')
    assert_equal Language['Ruby'], Language.find_by_lexer('ruby')
    assert_equal Language['Scheme'], Language.find_by_lexer('scheme')
    assert_equal Language['Shell'], Language.find_by_lexer('bash')
    assert_equal Language['Text'], Language.find_by_lexer('kt')
  end

  def test_find
    assert_equal "Ruby", Language['Ruby'].name
    assert_equal "Ruby", Language['ruby'].name
    assert_equal "Ruby", Language['RUBY'].name
    assert_equal "C++", Language['C++'].name
    assert_equal "C++", Language['c++'].name
    assert_equal "C++", Language['cpp'].name
    assert_equal "C#", Language['C#'].name
    assert_equal "C#", Language['c#'].name
    assert_equal "C#", Language['csharp'].name
    assert_equal "C#", Language['CSHARP'].name
    assert_equal "Text", Language['defunkt'].name
  end

  def test_name
    assert_equal "Perl",    Language['Perl'].name
    assert_equal "Python",  Language['Python'].name
    assert_equal "Ruby",    Language['Ruby'].name
  end

  # Used for code search indexing. Changing any of these values may
  # require reindexing repositories.
  def test_search_term
    assert_equal "perl",        Language['Perl'].search_term
    assert_equal "python",      Language['Python'].search_term
    assert_equal "ruby",        Language['Ruby'].search_term
    assert_equal "common-lisp", Language['Common Lisp'].search_term
    assert_equal "html+erb",    Language['HTML+ERB'].search_term
    assert_equal "max/msp",     Language['Max/MSP'].search_term
    assert_equal "pure-data",   Language['Pure Data'].search_term

    assert_equal "aspx-vb",    Language['ASP'].search_term
    assert_equal "as3",        Language['ActionScript'].search_term
    assert_equal "nasm",       Language['Assembly'].search_term
    assert_equal "bat",        Language['Batchfile'].search_term
    assert_equal "csharp",     Language['C#'].search_term
    assert_equal "cpp",        Language['C++'].search_term
    assert_equal "java",       Language['ChucK'].search_term
    assert_equal "cfm",        Language['ColdFusion'].search_term
    assert_equal "dpatch",     Language['Darcs Patch'].search_term
    assert_equal "scheme",     Language['Emacs Lisp'].search_term
    assert_equal "ocaml",      Language['F#'].search_term
    assert_equal "bash",       Language['Gentoo Ebuild'].search_term
    assert_equal "bash",       Language['Gentoo Eclass'].search_term
    assert_equal "pot",        Language['Gettext Catalog'].search_term
    assert_equal "irc",        Language['IRC log'].search_term
    assert_equal "java",       Language['Groovy'].search_term
    assert_equal "javascript", Language['JSON'].search_term
    assert_equal "lhs",        Language['Literate Haskell'].search_term
    assert_equal "ruby",       Language['Mirah'].search_term
    assert_equal "scheme",     Language['Nu'].search_term
    assert_equal "pir",        Language['Parrot Internal Representation'].search_term
    assert_equal "pytb",       Language['Python traceback'].search_term
    assert_equal "raw",        Language['Raw token data'].search_term
    assert_equal "bash",       Language['Shell'].search_term
    assert_equal "vim",        Language['VimL'].search_term
    assert_equal "c",          Language['XS'].search_term
    assert_equal "jsp",        Language['Java Server Pages'].search_term
    assert_equal "rst",        Language['reStructuredText'].search_term
  end

  def test_error_without_name
    assert_raise ArgumentError do
      Language.new :name => nil
    end
  end

  def test_lexer
    assert_equal Lexer['Perl'],   Language['Perl'].lexer
    assert_equal Lexer['Python'], Language['Python'].lexer
    assert_equal Lexer['Ruby'],   Language['Ruby'].lexer
    assert_equal Lexer['C++'],    Language['C++'].lexer
    assert_equal Lexer['Bash'],   Language['Gentoo Ebuild'].lexer
    assert_equal Lexer['Scheme'], Language['Nu'].lexer
  end

  def test_extensions
    assert Language['Perl'].extensions.include?('.pl')
    assert Language['Python'].extensions.include?('.py')
    assert Language['Ruby'].extensions.include?('.rb')
  end

  def test_eql
    assert Language['Ruby'].eql?(Language['Ruby'])
    assert !Language['Ruby'].eql?(Language['Python'])
    assert !Language['Ruby'].eql?(Language.new(:name => 'Ruby'))
  end

  def test_popular
    assert Language['Ruby'].popular?
    assert Language['Perl'].popular?
    assert Language['Python'].popular?
    assert Language['Assembly'].unpopular?
    assert Language['Brainfuck'].unpopular?
  end

  def test_common
    assert Language['Perl'].common?
    assert Language['Python'].common?
    assert Language['Ruby'].common?
    assert !Language['Brainfuck'].common?
    assert !Language['Makefile'].common?
  end

  def test_colorize
    assert_equal <<-HTML, Language['Text'].colorize("Hello")
<div class="highlight"><pre>Hello
</pre>
</div>
    HTML

    assert_equal <<-HTML, Language['Ruby'].colorize("def foo\n  'foo'\nend\n")
<div class="highlight"><pre><span class="k">def</span> <span class="nf">foo</span>
  <span class="s1">&#39;foo&#39;</span>
<span class="k">end</span>
</pre>
</div>
    HTML
  end

  def test_colorize_without_wrapper
    assert_equal <<-HTML, Language['Text'].colorize_without_wrapper("Hello")
Hello
    HTML

    assert_equal <<-HTML, Language['Ruby'].colorize_without_wrapper("def foo\n  'foo'\nend\n")
<span class="k">def</span> <span class="nf">foo</span>
  <span class="s1">&#39;foo&#39;</span>
<span class="k">end</span>
    HTML
  end
end