require 'test_helper'

class ModifierTest < ActiveSupport::TestCase
  include SetBuilder::Modifiers


  test "get type with string" do
    assert_equal StringPreposition, SetBuilder::Modifier["StringPreposition"]
  end

  test "get type with symbol" do
    assert_equal StringPreposition, SetBuilder::Modifier[:string]
  end

  test "registering a modifier" do
    assert_raises ArgumentError do
      SetBuilder::Modifier.for(:hash)
    end
    SetBuilder::Modifier.register(:hash, HashModifier)
    assert_nothing_raised ArgumentError do
      SetBuilder::Modifier.for(:hash)
    end
  end

  test "registering an invalid modifier" do
    assert_raises ArgumentError do
      SetBuilder::Modifier.register(:hash, InvalidHashModifier)
    end
  end

  test "converting modifier to json" do
    expected_results = {
      "contains"            => ["string"],
      "does_not_contain"    => ["string"],
      "begins_with"         => ["string"],
      "does_not_begin_with" => ["string"],
      "ends_with"           => ["string"],
      "does_not_end_with"   => ["string"],
      "is"                  => ["string"],
      "is_not"              => ["string"]
    }.to_json
    assert_equal expected_results, SetBuilder::Modifier.for(:string).to_json
  end

  test "converting modifiers to json" do
    expected_results = {
      "date" => {
        "ever"        => [],
        "after"       => ["date"],
        "before"      => ["date"],
        "on"          => ["date"],
        "in_the_last" => ["number", "period"],
        "during_month"=> ["month"],
        "during_year" => ["year"],
        "between"     => ["date", "date"]
      },
      "number"=> {
        "is"=>["number"],
        "is_less_than"=>["number"],
        "is_greater_than"=>["number"],
        "is_between"=>["number", "number"]
      },
      "string" => {
        "contains"            => ["string"],
        "does_not_contain"    => ["string"],
        "begins_with"         => ["string"],
        "does_not_begin_with" => ["string"],
        "ends_with"           => ["string"],
        "does_not_end_with"   => ["string"],
        "is"                  => ["string"],
        "is_not"              => ["string"]
      }
    }
    assert_equal expected_results, $friend_traits.modifiers.to_hash
  end


end


class InvalidHashModifier
end

class HashModifier < SetBuilder::Modifier::Verb

  def self.operators
    [:has_key]
  end

end