# -*- encoding: utf-8 -*-
require 'helper'

describe FuzzyMatch do
  describe '#find' do
    it %{identifies the best match based on string similarity} do
      d = FuzzyMatch.new %w{ RATZ CATZ }
      d.find('RITZ').must_equal 'RATZ'
      d.find('RíTZ').must_equal 'RATZ'

      d = FuzzyMatch.new [ 'X' ]
      d.find('X').must_equal 'X'
      d.find('A').must_be_nil
    end
  
    it %{not return any result if the maximum score is zero} do
      FuzzyMatch.new(['a']).find('b').must_be_nil
    end
  end
  
  describe '#find_all' do
    it %{return all records in sorted order} do
      d = FuzzyMatch.new [ 'X', 'X22', 'Y', 'Y4' ], :groupings => [ /X/, /Y/ ], :must_match_grouping => true
      d.find_all('X').must_equal ['X', 'X22' ]
      d.find_all('A').must_equal []
    end
  end

  describe '#find_best' do
    it %{returns one or more records with the best score} do
      d = FuzzyMatch.new [ 'X', 'X', 'X22', 'Y', 'Y', 'Y4' ], :groupings => [ /X/, /Y/ ], :must_match_grouping => true
      d.find_best('X').must_equal ['X', 'X' ]
      d.find_best('A').must_equal []
    end
  end

  describe '#find_all_with_score' do
    it %{return records with 2 scores} do
      d = FuzzyMatch.new [ 'X', 'X22', 'Y', 'Y4' ], :groupings => [ /X/, /Y/ ], :must_match_grouping => true
      d.find_all_with_score('X').must_equal [ ['X', 1, 1], ['X22', 0, 0.33333333333333337] ]
      d.find_all_with_score('A').must_equal []
    end
  end

  describe '#explain' do
    before do
      require 'stringio'
      @capture = StringIO.new
      @old_stdout = $stdout
      $stdout = @capture
    end
    after do
      $stdout = @old_stdout
    end
      
    it %{print a basic explanation to stdout} do
      d = FuzzyMatch.new %w{ RATZ CATZ }
      d.explain('RITZ')
      @capture.rewind
      @capture.read.must_include 'CATZ'
    end
    
    it %{explains match failures} do
      FuzzyMatch.new(['aaa']).explain('bbb')
      @capture.rewind
      @capture.read.must_match %r{No winner assigned.*aaa.*bbb}
    end
  end

  describe "normalizers" do
    it %{sometimes gets false results without them} do
      d = FuzzyMatch.new ['BOEING 737-100/200', 'BOEING 737-900']
      d.find('BOEING 737100 number 900').must_equal 'BOEING 737-900'
    end

    it %{can be used to improve results} do
      normalizers = [
        %r{(7\d)(7|0)-?(\d{1,3})} # tighten 737-100/200 => 737100, which will cause it to win over 737-900
      ]
      d = FuzzyMatch.new ['BOEING 737-100/200', 'BOEING 737-900'], :normalizers => normalizers
      d.find('BOEING 737100 number 900').must_equal 'BOEING 737-100/200'
    end
  end

  describe "identities" do
    it %{sometimes gets false results without them} do
      # false positive without identity
      d = FuzzyMatch.new %w{ foo bar }
      d.find('baz').must_equal 'bar'
    end

    it %{can be used to improve results} do
      d = FuzzyMatch.new %w{ foo bar }, :identities => [ /ba(.)/ ]
      d.find('baz').must_be_nil
    end
  end

  describe 'groupings' do
    it %{sometimes gets false results without them} do
      d = FuzzyMatch.new [ 'Barack Obama', 'George Bush' ]
      d.find('Barack Bush').must_equal 'Barack Obama' # luke i am your father
      d.find('George Obama').must_equal 'George Bush' # nooooooooooooooooooo
    end
    
    it %{can be used to improve results} do
      d = FuzzyMatch.new [ 'Barack Obama', 'George Bush' ], :groupings => [ /Obama/, /Bush/ ]
      d.find('Barack Bush').must_equal 'George Bush'
      d.find('George Obama').must_equal 'Barack Obama'
    end
  end
  
  describe "the :must_match_grouping option" do
    it %{optionally only attempt matches with records that fit into a grouping} do
      d = FuzzyMatch.new [ 'Barack Obama', 'George Bush' ], :groupings => [ /Obama/, /Bush/ ], :must_match_grouping => true
      d.find('George Clinton').must_be_nil

      d = FuzzyMatch.new [ 'Barack Obama', 'George Bush' ], :groupings => [ /Obama/, /Bush/ ]
      d.find('George Clinton', :must_match_grouping => true).must_be_nil
    end
  end
  
  describe "the :first_grouping_decides option" do
    it %{optionally force the first grouping to decide} do
      d = FuzzyMatch.new [ 'Boeing 747', 'Boeing 747SR', 'Boeing ER6' ], :groupings => [ /(boeing \d{3})/i, /boeing/i ]
      d.find_all('Boeing 747').must_equal [ 'Boeing 747', 'Boeing 747SR', 'Boeing ER6' ]

      d = FuzzyMatch.new [ 'Boeing 747', 'Boeing 747SR', 'Boeing ER6' ], :groupings => [ /(boeing \d{3})/i, /boeing/i ], :first_grouping_decides => true
      d.find_all('Boeing 747').must_equal [ 'Boeing 747', 'Boeing 747SR' ]

      # first_grouping_decides refers to the needle
      d = FuzzyMatch.new [ 'Boeing 747', 'Boeing 747SR', 'Boeing ER6' ], :groupings => [ /(boeing \d{3})/i, /boeing/i ], :first_grouping_decides => true
      d.find_all('Boeing ER6').must_equal ["Boeing ER6", "Boeing 747", "Boeing 747SR"]

      d = FuzzyMatch.new [ 'Boeing 747', 'Boeing 747SR', 'Boeing ER6' ], :groupings => [ /(boeing \d{3})/i, /boeing (7|E)/i, /boeing/i ], :first_grouping_decides => true
      d.find_all('Boeing ER6').must_equal [ 'Boeing ER6' ]

      # or equivalently with an identity
      d = FuzzyMatch.new [ 'Boeing 747', 'Boeing 747SR', 'Boeing ER6' ], :groupings => [ /(boeing \d{3})/i, /boeing/i ], :first_grouping_decides => true, :identities => [ /boeing (7|E)/i ]
      d.find_all('Boeing ER6').must_equal [ 'Boeing ER6' ]
    end
  end

  describe "the :read option" do
    it %{interpret a Numeric as an array index} do
      ab = ['a', 'b']
      ba = ['b', 'a']
      haystack = [ab, ba]
      by_first = FuzzyMatch.new haystack, :read => 0
      by_last = FuzzyMatch.new haystack, :read => 1
      by_first.find('a').must_equal ab
      by_last.find('b').must_equal ab
      by_first.find('b').must_equal ba
      by_last.find('a').must_equal ba
    end

    it %{interpret a Symbol, etc. as hash key} do
      ab = { :one => 'a', :two => 'b' }
      ba = { :one => 'b', :two => 'a' }
      haystack = [ab, ba]
      by_first = FuzzyMatch.new haystack, :read => :one
      by_last = FuzzyMatch.new haystack, :read => :two
      by_first.find('a').must_equal ab
      by_last.find('b').must_equal ab
      by_first.find('b').must_equal ba
      by_last.find('a').must_equal ba
    end

    MyStruct = Struct.new(:one, :two)
    it %{interpret a Symbol as a method id (if the object responds to it)} do
      ab = MyStruct.new('a', 'b')
      ba = MyStruct.new('b', 'a')
      haystack = [ab, ba]
      by_first = FuzzyMatch.new haystack, :read => :one
      by_last = FuzzyMatch.new haystack, :read => :two
      by_first.read.must_equal :one
      by_last.read.must_equal :two
      by_first.find('a').must_equal ab
      by_last.find('b').must_equal ab
      by_first.find('b').must_equal ba
      by_last.find('a').must_equal ba
    end
  end
  
  describe 'the :must_match_at_least_one_word option' do
    it %{optionally require that the matching record share at least one word with the needle} do
      d = FuzzyMatch.new %w{ RATZ CATZ }, :must_match_at_least_one_word => true
      d.find('RITZ').must_be_nil

      d = FuzzyMatch.new ["Foo's Bar"], :must_match_at_least_one_word => true
      d.find("Foo's").must_equal "Foo's Bar"
      d.find("'s").must_be_nil
      d.find("Foo").must_be_nil

      d = FuzzyMatch.new ["Bolivia, Plurinational State of"], :must_match_at_least_one_word => true
      d.find("Bolivia").must_equal "Bolivia, Plurinational State of"
    end

    it %{use STOP WORDS} do
      d = FuzzyMatch.new [ 'A HOTEL', 'B HTL' ]
      d.find('A HTL', :must_match_at_least_one_word => true).must_equal 'B HTL'

      d = FuzzyMatch.new [ 'A HOTEL', 'B HTL' ], :must_match_at_least_one_word => true
      d.find('A HTL').must_equal 'B HTL'

      d = FuzzyMatch.new [ 'A HOTEL', 'B HTL' ], :must_match_at_least_one_word => true, :stop_words => [ %r{HO?TE?L} ]
      d.find('A HTL').must_equal 'A HOTEL'
    end
    
    it %{not be fooled by substrings (but rather compare whole words to whole words)} do
      d = FuzzyMatch.new [ 'PENINSULA HOTELS' ], :must_match_at_least_one_word => true
      d.find('DOLCE LA HULPE BXL FI').must_be_nil
    end

    it %{not be case-sensitive when checking for sharing of words} do
      d = FuzzyMatch.new [ 'A', 'B' ]
      d.find('a', :must_match_at_least_one_word => true).must_equal 'A'
    end
  end
  
  describe "the :gather_last_result option" do
    it %{not gather metadata about the last result by default} do
      d = FuzzyMatch.new %w{ NISSAN HONDA }
      d.find('MISSAM')
      lambda do
        d.last_result
      end.must_raise ::RuntimeError, /gather_last_result/
    end

    it %{optionally gather metadata about the last result} do
      d = FuzzyMatch.new %w{ NISSAN HONDA }
      d.find 'MISSAM', :gather_last_result => true
      d.last_result.score.must_equal 0.6
      d.last_result.winner.must_equal 'NISSAN'
    end
  end
  
  describe 'quirks' do
    it %{should not return false negatives because of one-letter similarities} do
      # dices coefficient doesn't think these two are similar at all because it looks at pairs
      FuzzyMatch.score_class.new('X foo', 'X bar').dices_coefficient_similar.must_equal 0
      # so we must compensate for that somewhere
      d = FuzzyMatch.new ['X foo', 'randomness']
      d.find('X bar').must_equal 'X foo'
      # without making false positives
      d.find('Y bar').must_be_nil
    end

    it %{finds possible matches even when pair distance fails} do
      d = FuzzyMatch.new ['XX', '2 A']
      d.find('2A').must_equal '2 A'
      d = FuzzyMatch.new ['XX', '2A']
      d.find('2 A').must_equal '2A'
    end

    it %{weird blow ups} do
      d = FuzzyMatch.new ['XX', '2 A']
      d.find('A').must_equal '2 A'
      d = FuzzyMatch.new ['XX', 'A']
      d.find('2 A').must_equal 'A'
    end

  end

  describe 'deprecations' do
    it %{takes :must_match_blocking as :must_match_grouping} do
      d = FuzzyMatch.new [], :must_match_blocking => :a
      d.default_options[:must_match_grouping].must_equal :a
    end

    it %{takes :first_blocking_decides as :first_grouping_decides} do
      d = FuzzyMatch.new [], :first_blocking_decides => :b
      d.default_options[:first_grouping_decides].must_equal :b
    end

    it %{takes :haystack_reader as :read} do
      d = FuzzyMatch.new [], :haystack_reader => :c
      d.read.must_equal :c
    end

    it %{takes :blockings as :groupings} do
      d = FuzzyMatch.new [], :blockings => [ /X/, /Y/ ]
      d.groupings.must_equal [ FuzzyMatch::Rule::Grouping.new(/X/), FuzzyMatch::Rule::Grouping.new(/Y/) ]
    end

    it %{takes :tighteners as :normalizers} do
      d = FuzzyMatch.new [], :tighteners => [ /X/, /Y/ ]
      d.normalizers.must_equal [ FuzzyMatch::Rule::Normalizer.new(/X/), FuzzyMatch::Rule::Normalizer.new(/Y/) ]
    end

    it %{receives #free method, but doesn't do anything} do
      d = FuzzyMatch.new %w{ A B }
      d.free
      d.find('A').wont_be_nil
    end
  end
  
  it %{defaults to a pure-ruby engine, but also has amatch} do
    if defined?($testing_amatch) and $testing_amatch
      FuzzyMatch.engine.must_equal :amatch
    else
      FuzzyMatch.engine.must_equal :pure_ruby
    end
  end
end