# encoding: UTF-8

require File.join(File.expand_path(File.dirname(__FILE__)), '../test_helper')

class ResultsTest < ActiveSupport::TestCase

  setup do
    load_test_data
    SkosImporter.new('test/concept_test.nt', 'http://localhost:3000/').run
  end

  test "ranked results" do
    results = Services::SimilarTermsService.ranked("en", {}, "forest")
    assert_equal 2, results.length
    assert_equal Iqvoc::XLLabel.base_class, results[0][0].class
    assert_equal "forest", results[0][0].value
    assert_equal "forest", results[0][1].origin
    assert_equal "woods", results[1][0].value
    assert_equal "forest", results[1][1].origin

    results = Services::SimilarTermsService.ranked("en", {}, "woods", "car")
    assert_equal 4, results.length
    assert_equal "forest", results[0][0].value
    assert_equal "forest", results[0][1].origin
    assert_equal "car", results[1][0].value
    assert_equal "car", results[1][1].origin
    assert_equal "woods", results[2][0].value
    assert_equal "forest", results[2][1].origin
    assert_equal "automobile", results[3][0].value
    assert_equal "car", results[3][1].origin
    assert_equal results[0].length, results[0].uniq.length
  end

  test "weighted results" do
    results = Services::SimilarTermsService.weighted("en", {}, "forest")
    assert_equal 2, results.keys.length
    expected = { "forest" => 5, "woods" => 2 }
    results.each do |label, data|
      assert_equal Iqvoc::XLLabel.base_class, label.class
      assert_equal 2, data.length
      assert_equal expected[label.value], data[0]
      assert_equal Iqvoc::Concept.base_class, data[1].class
    end
  end

  #TODO: add a related concept relation and also test it
  test "inclusion of pref labels of narrower and related concepts" do
    results = Services::SimilarTermsService.weighted("en", {}, "water")
    assert_equal 5, results.length
    assert_equal "water", results.keys.first.value
    assert_equal 5, results[results.keys.first][0]
    assert_equal "real water", results.keys.second.value
    assert_equal 2, results[results.keys.second][0]
    assert_equal "related", results.keys.third.value
    assert_equal 1, results[results.keys.third][0]
    assert_equal "used water", results.keys.fourth.value
    assert_equal 1, results[results.keys.fourth][0]
    assert_equal "new water", results.keys.fifth.value
    assert_equal 1, results[results.keys.fifth][0]
  end

  test "inclusion of pref labels of narrower and related concepts - case insensitive" do
    results = Services::SimilarTermsService.weighted("en", {}, "Water")
    assert_equal 5, results.length
    assert_equal "water", results.keys.first.value
    assert_equal 5, results[results.keys.first][0]
    assert_equal "real water", results.keys.second.value
    assert_equal 2, results[results.keys.second][0]
    assert_equal "related", results.keys.third.value
    assert_equal 1, results[results.keys.third][0]
    assert_equal "used water", results.keys.fourth.value
    assert_equal 1, results[results.keys.fourth][0]
    assert_equal "new water", results.keys.fifth.value
    assert_equal 1, results[results.keys.fifth][0]
  end

  test "no results" do
    results = Services::SimilarTermsService.weighted("de", {}, "water")
    assert_equal 0, results.length
  end

  test "compound returns - case insensitive" do
    SkosImporter.new('test/compound_forms.nt', 'http://hobbies.com/').run
    results = Services::SimilarTermsService.weighted("de", {}, "computer")
    assert_equal 1, results.length
    assert_equal "Computer programming", results.keys.first.value
    assert_equal 1, results[results.keys.first][0]
  end

  test "compound returns" do
    SkosImporter.new('test/compound_forms.nt', 'http://hobbies.com/').run
    results = Services::SimilarTermsService.weighted("de", {}, "Computer")
    assert_equal 1, results.length
    assert_equal "Computer programming", results.keys.first.value
    assert_equal 1, results[results.keys.first][0]
  end

  test "nothing unpublished" do
    concept = Iqvoc::XLLabel.base_class.where(value: "forest").first.concepts.first
    concept.update(published_at: nil)
    results = Services::SimilarTermsService.weighted("en", {}, "forest")
    assert_equal 0, results.length
  end

end