enumerize 0.8.0 → 2.6.1

data/test/mongo_mapper_test.rb

@@ -1,22 +1,27 @@
+# frozen_string_literal: true
+
 require 'test_helper'
 
+begin
+
 silence_warnings do
   require 'mongo_mapper'
 end
 
-MongoMapper.connection = Mongo::Connection.new('localhost', 27017)
-MongoMapper.database   = 'enumerize-test-suite-of-mongomapper'
+MongoMapper.connection = Mongo::Client.new(['localhost:27017'], database: 'enumerize-test-suite-of-mongomapper')
 
-describe Enumerize do
+class MongoMapperTest < MiniTest::Spec
   class MongoMapperUser
     include MongoMapper::Document
     extend Enumerize
 
     key :sex
     key :role
+    key :foo
 
-    enumerize :sex, :in => %w[male female]
+    enumerize :sex,  :in => %w[male female]
     enumerize :role, :in => %w[admin user], :default => 'user'
+    enumerize :foo,  :in => %w[bar baz], :skip_validations => true
   end
 
   before { $VERBOSE = nil }
@@ -27,7 +32,7 @@ describe Enumerize do
   it 'sets nil if invalid value is passed' do
     user = model.new
     user.sex = :invalid
-    user.sex.must_equal nil
+    expect(user.sex).must_be_nil
   end
 
   it 'saves value' do
@@ -35,7 +40,7 @@ describe Enumerize do
     user = model.new
     user.sex = :female
     user.save!
-    user.sex.must_equal 'female'
+    expect(user.sex).must_equal 'female'
   end
 
   it 'loads value' do
@@ -43,19 +48,25 @@ describe Enumerize do
     model.create!(:sex => :male)
     store_translations(:en, :enumerize => {:sex => {:male => 'Male'}}) do
       user = model.first
-      user.sex.must_equal 'male'
-      user.sex_text.must_equal 'Male'
+      expect(user.sex).must_equal 'male'
+      expect(user.sex_text).must_equal 'Male'
     end
   end
 
   it 'has default value' do
-    model.new.role.must_equal 'user'
+    expect(model.new.role).must_equal 'user'
   end
 
   it 'validates inclusion' do
     user = model.new
     user.role = 'wrong'
-    user.wont_be :valid?
+    expect(user).wont_be :valid?
+  end
+
+  it 'does not validate inclusion when :skip_validations option passed' do
+    user = model.new
+    user.foo = 'wrong'
+    expect(user).must_be :valid?
   end
 
   it 'assigns value on loaded record' do
@@ -63,6 +74,10 @@ describe Enumerize do
     model.create!(:sex => :male)
     user = model.first
     user.sex = :female
-    user.sex.must_equal 'female'
+    expect(user.sex).must_equal 'female'
   end
 end
+
+rescue LoadError
+  # Skip
+end

data/test/mongoid_test.rb

@@ -1,25 +1,35 @@
+# frozen_string_literal: true
+
 require 'test_helper'
 
+begin
+
 silence_warnings do
   require 'mongoid'
 end
 
 Mongoid.configure do |config|
-  config.sessions = { :default => { :database => 'enumerize-test-suite', hosts: ['127.0.0.1:27017'] } }
-  config.use_utc = true
-  config.include_root_in_json = true
+  config.connect_to('enumerize-test-suite')
+  config.options = { use_utc: true, include_root_in_json: true }
 end
 
-describe Enumerize do
+class MongoidTest < MiniTest::Spec
   class MongoidUser
     include Mongoid::Document
     extend Enumerize
 
     field :sex
     field :role
-    enumerize :sex, :in => %w[male female]
-    enumerize :role, :in => %w[admin user], :default => 'user'
-    enumerize :mult, :in => %w[one two three four], :multiple => true
+    field :foo
+    field :skill
+
+    enumerize :sex,    :in => %w[male female], scope: true
+    enumerize :status, :in => %w[notice warning error], scope: true
+    enumerize :role,   :in => %w[admin user], :default => 'user', scope: :having_role
+    enumerize :mult,   :in => %w[one two three four], :multiple => true
+    enumerize :foo,    :in => %w[bar baz], :skip_validations => true
+    enumerize :skill,  :in => { noob: 0, casual: 1, pro: 2 }, scope: :shallow
+    enumerize :account_type, :in => %w[basic premium], scope: :shallow
   end
 
   before { $VERBOSE = nil }
@@ -30,7 +40,7 @@ describe Enumerize do
   it 'sets nil if invalid value is passed' do
     user = model.new
     user.sex = :invalid
-    user.sex.must_equal nil
+    expect(user.sex).must_be_nil
   end
 
   it 'saves value' do
@@ -38,7 +48,7 @@ describe Enumerize do
     user = model.new
     user.sex = :female
     user.save!
-    user.sex.must_equal 'female'
+    expect(user.sex).must_equal 'female'
   end
 
   it 'loads value' do
@@ -46,19 +56,48 @@ describe Enumerize do
     model.create!(:sex => :male)
     store_translations(:en, :enumerize => {:sex => {:male => 'Male'}}) do
       user = model.first
-      user.sex.must_equal 'male'
-      user.sex_text.must_equal 'Male'
+      expect(user.sex).must_equal 'male'
+      expect(user.sex_text).must_equal 'Male'
     end
   end
 
   it 'has default value' do
-    model.new.role.must_equal 'user'
+    expect(model.new.role).must_equal 'user'
+  end
+
+  it 'uses after_initialize callback to set default value' do
+    model.delete_all
+    model.create!(sex: 'male', role: nil)
+
+    user = model.where(sex: 'male').first
+    expect(user.role).must_equal 'user'
+  end
+
+  it 'does not set default value for not selected attributes' do
+    model.delete_all
+    model.create!(sex: :male)
+
+    assert_equal ['_id'], model.only(:id).first.attributes.keys
   end
 
   it 'validates inclusion' do
     user = model.new
     user.role = 'wrong'
-    user.wont_be :valid?
+    expect(user).wont_be :valid?
+  end
+
+  it 'does not validate inclusion when :skip_validations option passed' do
+    user = model.new
+    user.foo = 'wrong'
+    expect(user).must_be :valid?
+  end
+
+  it 'sets value to enumerized field from db when record is reloaded' do
+    user = model.create!(mult: [:one])
+    model.find(user.id).update(mult: %i[two three])
+    expect(user.mult).must_equal %w[one]
+    user.reload
+    expect(user.mult).must_equal %w[two three]
   end
 
   it 'assigns value on loaded record' do
@@ -66,7 +105,7 @@ describe Enumerize do
     model.create!(:sex => :male)
     user = model.first
     user.sex = :female
-    user.sex.must_equal 'female'
+    expect(user.sex).must_equal 'female'
   end
 
   it 'loads multiple properly' do
@@ -74,6 +113,52 @@ describe Enumerize do
 
     model.create!(:mult => ['one', 'two'])
     user = model.first
-    user.mult.to_a.must_equal ['one', 'two']
+    expect(user.mult.to_a).must_equal ['one', 'two']
+  end
+
+  it 'adds scope' do
+    model.delete_all
+
+    user_1 = model.create!(sex: :male, skill: :noob, role: :admin, account_type: :basic)
+    user_2 = model.create!(sex: :female, skill: :noob, role: :user, account_type: :basic)
+    user_3 = model.create!(skill: :pro, account_type: :premium)
+
+    expect(model.with_sex(:male).to_a).must_equal [user_1]
+    expect(model.with_sex(:female).to_a).must_equal [user_2]
+    expect(model.with_sex(:male, :female).to_set).must_equal [user_1, user_2].to_set
+
+    expect(model.without_sex(:male).to_set).must_equal [user_2, user_3].to_set
+    expect(model.without_sex(:female).to_set).must_equal [user_1, user_3].to_set
+    expect(model.without_sex(:male, :female).to_a).must_equal [user_3]
+
+    expect(model.having_role(:admin).to_a).must_equal [user_1]
+    expect(model.having_role(:user).to_a).must_equal [user_2, user_3]
+
+    expect(model.pro.to_a).must_equal [user_3]
+    expect(model.premium.to_a).must_equal [user_3]
+
+    expect(model.not_pro.to_set).must_equal [user_1, user_2].to_set
+    expect(model.not_premium.to_set).must_equal [user_1, user_2].to_set
+  end
+
+  it 'chains scopes' do
+    model.delete_all
+
+    user_1 = model.create!(status: :notice)
+    user_2 = model.create!(status: :warning)
+    user_3 = model.create!(status: :error)
+
+    expect(model.with_status(:notice, :warning).with_status(:notice, :error).to_a).must_equal [user_1]
+    expect(model.with_status(:notice, :warning).union.with_status(:notice, :error).to_a).must_equal [user_1, user_2, user_3]
   end
+
+  it 'ignores not enumerized values that passed to the scope method' do
+    model.delete_all
+
+    expect(model.with_sex(:foo)).must_equal []
+  end
+end
+
+rescue LoadError
+  # Skip
 end

data/test/multiple_test.rb

@@ -1,36 +1,65 @@
+# frozen_string_literal: true
+
 require 'test_helper'
 
-describe Enumerize::Base do
-  let(:klass) do
+class MultipleTest < MiniTest::Spec
+  let(:kklass) do
     Class.new do
       extend Enumerize
     end
   end
 
   let(:subklass) do
-    Class.new(klass)
+    Class.new(kklass)
   end
 
-  let(:object) { klass.new }
+  let(:object) { kklass.new }
 
   it 'returns [] when not set' do
-    klass.enumerize :foos, in: %w(a b), multiple: true
-    object.foos.must_equal []
+    kklass.enumerize :foos, in: %w(a b), multiple: true
+    expect(object.foos).must_equal []
   end
 
   it 'returns setted array' do
-    klass.enumerize :foos, in: %w(a b c), multiple: true
+    kklass.enumerize :foos, in: %w(a b c), multiple: true
     object.foos = %w(a c)
-    object.foos.must_equal %w(a c)
+    expect(object.foos).must_equal %w(a c)
+  end
+
+  it 'sets default value as single value' do
+    kklass.enumerize :foos, in: %w(a b c), default: 'b', multiple: true
+    expect(object.foos).must_equal %w(b)
+  end
+
+  it 'sets default value as array of one element' do
+    kklass.enumerize :foos, in: %w(a b c), default: %w(b), multiple: true
+    expect(object.foos).must_equal %w(b)
+  end
+
+  it 'sets default value as array of several elements' do
+    kklass.enumerize :foos, in: %w(a b c), default: %w(b c), multiple: true
+    expect(object.foos).must_equal %w(b c)
   end
 
   it "doesn't define _text method" do
-    klass.enumerize :foos, in: %w(a b c), multiple: true
-    object.wont_respond_to :foos_text
+    kklass.enumerize :foos, in: %w(a b c), multiple: true
+    expect(object).wont_respond_to :foos_text
   end
 
   it "doesn't define _value method" do
-    klass.enumerize :foos, in: %w(a b c), multiple: true
-    object.wont_respond_to :foos_value
+    kklass.enumerize :foos, in: %w(a b c), multiple: true
+    expect(object).wont_respond_to :foos_value
+  end
+
+  it "cannot define multiple with scope" do
+    assert_raises ArgumentError do
+      kklass.enumerize :foos, in: %w(a b c), multiple: true, scope: true
+    end
+  end
+
+  it 'assign a name with the first letter capitalized' do
+    kklass.enumerize :Foos, in: %w(a b c), multiple: true
+    object.Foos = %w(a c)
+    expect(object.Foos).must_equal %w(a c)
   end
 end

data/test/predicates_test.rb

@@ -1,57 +1,65 @@
+# frozen_string_literal: true
+
 require 'test_helper'
 
-describe Enumerize::Predicates do
-  let(:klass) do
+class PredicatesTest < MiniTest::Spec
+  let(:kklass) do
     Class.new do
       extend Enumerize
     end
   end
 
-  let(:object) { klass.new }
+  let(:object) { kklass.new }
 
   it 'creates predicate methods' do
-    klass.enumerize(:foo, in: %w(a b), predicates: true)
-    object.must_respond_to :a?
-    object.must_respond_to :b?
+    kklass.enumerize(:foo, in: %w(a b), predicates: true)
+    expect(object).must_respond_to :a?
+    expect(object).must_respond_to :b?
+  end
+
+  it 'creates predicate methods when enumerized values have dash in it' do
+    kklass.enumerize(:foo, in: %w(foo-bar bar-foo), predicates: true)
+    expect(object).must_respond_to :foo_bar?
+    expect(object).must_respond_to :bar_foo?
   end
 
   it 'creates predicate methods on multiple attribute' do
-    klass.enumerize(:foo, in: %w(a b), predicates: true, multiple: true)
-    object.must_respond_to :a?
-    object.must_respond_to :b?
+    kklass.enumerize(:foo, in: %w(a b), predicates: true, multiple: true)
+    expect(object).must_respond_to :a?
+    expect(object).must_respond_to :b?
   end
 
   it 'checks values' do
-    klass.enumerize(:foo, in: %w(a b), predicates: true)
+    kklass.enumerize(:foo, in: %w(a b), predicates: true)
     object.foo = 'a'
-    object.a?.must_equal true
-    object.b?.must_equal false
+    expect(object.a?).must_equal true
+    expect(object.b?).must_equal false
   end
 
   it 'checks values on multiple attribute' do
-    klass.enumerize(:foo, in: %w(a b), predicates: true, multiple: true)
+    kklass.enumerize(:foo, in: %w(a b), predicates: true, multiple: true)
     object.foo << :a
-    object.a?.must_equal true
-    object.b?.must_equal false
+    expect(object.a?).must_equal true
+    expect(object.b?).must_equal false
   end
 
   it 'prefixes methods' do
-    klass.enumerize(:foo, in: %w(a b), predicates: { prefix: 'bar' })
-    object.wont_respond_to :a?
-    object.wont_respond_to :b?
-    object.must_respond_to :bar_a?
-    object.must_respond_to :bar_b?
+    kklass.enumerize(:foo, in: %w(a b), predicates: { prefix: 'bar' })
+    expect(object).wont_respond_to :a?
+    expect(object).wont_respond_to :b?
+    expect(object).must_respond_to :bar_a?
+    expect(object).must_respond_to :bar_b?
   end
 
   it 'accepts only option' do
-    klass.enumerize(:foo, in: %w(a b), predicates: { only: :a })
-    object.must_respond_to :a?
-    object.wont_respond_to :b?
+    kklass.enumerize(:foo, in: %w(a b), predicates: { only: :a })
+    expect(object).must_respond_to :a?
+    expect(object).wont_respond_to :b?
   end
 
   it 'accepts except option' do
-    klass.enumerize(:foo, in: %w(a b), predicates: { except: :a })
-    object.wont_respond_to :a?
-    object.must_respond_to :b?
+    kklass.enumerize(:foo, in: %w(a b), predicates: { except: :a })
+    expect(object).wont_respond_to :a?
+    expect(object).must_respond_to :b?
   end
 end

data/test/rails_admin_test.rb

@@ -1,25 +1,27 @@
+# frozen_string_literal: true
+
 require 'test_helper'
 
 class RailsAdminSpec < MiniTest::Spec
-  let(:klass) do
+  let(:kklass) do
     Class.new do
       extend Enumerize
     end
   end
 
-  let(:object) { klass.new }
+  let(:object) { kklass.new }
 
   it 'defines enum method' do
     store_translations(:en, :enumerize => {:foo => {:a => 'a text', :b => 'b text'}}) do
-      klass.enumerize(:foo, in: [:a, :b])
-      object.foo_enum.must_equal [['a text', 'a'], ['b text', 'b']]
+      kklass.enumerize(:foo, in: [:a, :b])
+      expect(object.foo_enum).must_equal [['a text', 'a'], ['b text', 'b']]
     end
   end
 
   it 'defines enum properly for custom values enumerations' do
     store_translations(:en, :enumerize => {:foo => {:a => 'a text', :b => 'b text'}}) do
-      klass.enumerize(:foo, in: {:a => 1, :b => 2})
-      object.foo_enum.must_equal [['a text', 1], ['b text', 2]]
+      kklass.enumerize(:foo, in: {:a => 1, :b => 2})
+      expect(object.foo_enum).must_equal [['a text', 'a'], ['b text', 'b']]
     end
   end
 end