has_constant 0.4.9 → 0.5.0

data/VERSION

@@ -1 +1 @@
-0.4.9
+0.5.0

data/has_constant.gemspec

@@ -5,7 +5,7 @@
 
 Gem::Specification.new do |s|
   s.name = %q{has_constant}
-  s.version = "0.4.9"
+  s.version = "0.5.0"
 
   s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
   s.authors = ["mattbeedle"]
@@ -27,10 +27,12 @@ Gem::Specification.new do |s|
     "VERSION",
     "has_constant.gemspec",
     "lib/has_constant.rb",
+    "lib/has_constant/orm/active_model.rb",
     "lib/has_constant/orm/active_record.rb",
     "lib/has_constant/orm/mongoid.rb",
     "test/has_constant_test.rb",
     "test/helper.rb",
+    "test/unit/orm/active_model_test.rb",
     "test/unit/orm/active_record_test.rb",
     "test/unit/orm/mongoid_test.rb"
   ]
@@ -41,6 +43,7 @@ Gem::Specification.new do |s|
   s.test_files = [
     "test/has_constant_test.rb",
     "test/helper.rb",
+    "test/unit/orm/active_model_test.rb",
     "test/unit/orm/active_record_test.rb",
     "test/unit/orm/mongoid_test.rb"
   ]

data/lib/has_constant/orm/active_model.rb

@@ -0,0 +1,65 @@
+module HasConstant
+  module Orm
+    module ActiveModel
+      extend ActiveSupport::Concern
+
+      module ClassMethods
+        def has_constant( name, values = lambda { I18n.t(name) }, options = {} )
+          super(name, values, options)
+
+          singular = (options[:accessor] || name.to_s.singularize).to_s
+
+          # Add the getter method. This returns the string representation of
+          # the stored value
+          define_method(singular) do
+            if read_attribute(singular)
+              self.class.send(name)[read_attribute(singular).to_i]
+            end
+          end
+
+          define_method("#{singular}=") do |val|
+            if val.instance_of?(String)
+              write_attribute singular.to_sym, self.class.send(name.to_s).index(val)
+            else
+              write_attribute singular.to_sym, val
+            end
+          end
+
+          class_eval do
+            if respond_to?(:scope)
+              scope :by_constant, lambda { |constant,value| { :conditions =>
+                { constant.to_sym =>
+                  eval("#{self.to_s}.#{constant.pluralize}.index(value)") } } }
+
+              scope "#{singular}_is".to_sym, lambda { |*values| { :conditions =>
+                { singular.to_sym => indexes_for(name, values) }
+              } }
+
+              scope "#{singular}_is_not".to_sym, lambda { |*values| {
+                :conditions => ["#{singular} NOT IN (?)",
+                                indexes_for(name, values)] } }
+            else
+              named_scope :by_constant, lambda { |constant,value| {
+                :conditions =>
+                { constant.to_sym =>
+                  eval("#{self.to_s}.#{constant.pluralize}.index(value)") } } }
+
+              named_scope "#{singular}_is".to_sym, lambda { |*values| {
+                :conditions => { singular.to_sym => indexes_for(name, values) }
+              } }
+
+              named_scope "#{singular}_is_not".to_sym, lambda { |*values| {
+                :conditions => ["#{singular} NOT IN (?)",
+                                indexes_for(name, values)] } }
+            end
+          end
+        end
+
+      private
+        def indexes_for( name, values )
+          values.map { |v| self.send(name.to_sym).index(v) }
+        end
+      end
+    end
+  end
+end

data/lib/has_constant.rb

@@ -1,11 +1,19 @@
 require 'active_support'
 require 'has_constant/orm/mongoid'
-require 'has_constant/orm/active_record'
+require 'has_constant/orm/active_model'
 require 'active_support/inflector'
 
 module HasConstant
   extend ActiveSupport::Concern
 
+  included do
+    if defined?(Mongoid) && ancestors.include?(Mongoid::Document)
+      send(:include, HasConstant::Orm::Mongoid)
+    elsif defined?(ActiveModel) && !ancestors.map(&:to_s).grep(/^ActiveModel/).blank?
+      send(:include, HasConstant::Orm::ActiveModel)
+    end
+  end
+
   # HasConstant takes a Proc containing an array of possible values for a field name
   # The field name is inferred as the singular of the has constant name. For example
   # has_constant :titles

data/test/unit/orm/active_model_test.rb

@@ -0,0 +1,69 @@
+require 'helper'
+require 'mocha'
+
+setup_active_record
+
+class User < ActiveRecord::Base
+  include HasConstant
+end
+
+class ActiveModelTest < Test::Unit::TestCase
+  context 'Instance' do
+    should 'save values as integers' do
+      User.has_constant :salutations, %w(Mr Mrs)
+      u = User.new(:salutation => 'Mr')
+      u.save!
+      assert_equal 'Mr', u.salutation
+      assert_equal 0, u.attributes['salutation']
+    end
+
+    should 'default values to translated values list' do
+      I18n.stubs(:t).returns(['a', 'b'])
+      User.has_constant :salutations
+      assert_equal ['a', 'b'], User.salutations
+    end
+
+    context 'accessor' do
+      setup do
+        User.has_constant :titles, ['Mr', 'Mrs'], :accessor => :salutation
+        @user = User.new(:salutation => 'Mr')
+      end
+
+      should 'work store the values in the correct field ' do
+        @user.save!
+        assert_equal 0, @user.attributes['salutation']
+      end
+
+      should 'use the accessor to get values' do
+        assert_equal 'Mr', @user.salutation
+      end
+
+      should 'use constant name for values list' do
+        assert_equal %w(Mr Mrs), User.titles
+      end
+    end
+  end
+
+  context 'scopes' do
+    setup do
+      User.has_constant :salutations, %w(Mr Mrs)
+      @man = User.create!(:salutation => 'Mr')
+      @woman = User.create!(:salutation => 'Mrs')
+    end
+
+    should 'provide by_constant scope' do
+      assert_equal 1, User.by_constant('salutation', 'Mr').count
+      assert_equal @man, User.by_constant('salutation', 'Mr').first
+    end
+
+    should 'provide singular_is scope' do
+      assert_equal 1, User.salutation_is('Mr').count
+      assert_equal @man, User.salutation_is('Mr').first
+    end
+
+    should 'provide singular_is_not scope' do
+      assert_equal 1, User.salutation_is_not('Mr').count
+      assert_equal @woman, User.salutation_is_not('Mr').first
+    end
+  end
+end if defined?(ActiveRecord)

data/test/unit/orm/active_record_test.rb

@@ -4,7 +4,6 @@ setup_active_record
 
 class User < ActiveRecord::Base
   include HasConstant
-  include HasConstant::Orm::ActiveRecord
 
   has_constant :salutations, ['Mr', 'Mrs']
 end

data/test/unit/orm/mongoid_test.rb

@@ -1,11 +1,11 @@
 require 'helper'
+require 'mocha'
 
 setup_mongoid
 
 class MongoUser
   include Mongoid::Document
   include HasConstant
-  include HasConstant::Orm::Mongoid
 
   has_constant :salutations, ['Mr', 'Mrs']
 end if defined?(Mongoid)
@@ -13,7 +13,6 @@ end if defined?(Mongoid)
 class MongoUserWithProc
   include Mongoid::Document
   include HasConstant
-  include HasConstant::Orm::Mongoid
 
   has_constant :salutations, lambda { ['Mr', 'Mrs'] }
 end if defined?(Mongoid)
@@ -21,7 +20,6 @@ end if defined?(Mongoid)
 class MongoUserWithout
   include Mongoid::Document
   include HasConstant
-  include HasConstant::Orm::Mongoid
 end
 
 class MongoidTest < Test::Unit::TestCase

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification 
 name: has_constant
 version: !ruby/object:Gem::Version 
-  hash: 29
+  hash: 11
   prerelease: false
   segments: 
   - 0
-  - 4
-  - 9
-  version: 0.4.9
+  - 5
+  - 0
+  version: 0.5.0
 platform: ruby
 authors: 
 - mattbeedle
@@ -80,10 +80,12 @@ files:
 - VERSION
 - has_constant.gemspec
 - lib/has_constant.rb
+- lib/has_constant/orm/active_model.rb
 - lib/has_constant/orm/active_record.rb
 - lib/has_constant/orm/mongoid.rb
 - test/has_constant_test.rb
 - test/helper.rb
+- test/unit/orm/active_model_test.rb
 - test/unit/orm/active_record_test.rb
 - test/unit/orm/mongoid_test.rb
 has_rdoc: true
@@ -123,5 +125,6 @@ summary: Allows certain fields to be limited to a set of values
 test_files: 
 - test/has_constant_test.rb
 - test/helper.rb
+- test/unit/orm/active_model_test.rb
 - test/unit/orm/active_record_test.rb
 - test/unit/orm/mongoid_test.rb