hashtrain-acts_as_random_id 1.0.3 → 1.1.0

data/Rakefile

@@ -38,7 +38,7 @@ PKG_FILES = FileList[
 
 spec = Gem::Specification.new do |s| 
    s.name             = "acts_as_random_id" 
-   s.version          = "1.0.0"
+   s.version          = "1.1.0"
    s.author           = "hashtrain.com and author idea Stanislav Pogrebnyak (stanislav.pogrebnyak@gmail.com)"
    s.email            = "mail@hashtrain.com"
    s.homepage         = "http://github.com/hashtrain/acts_as_random_id/"

data/lib/acts_as_random_id.rb

@@ -3,20 +3,22 @@ module ActsAsRandomId
   def self.included(base)
     base.send :extend, ClassMethods
     
-    def ensure_unique_id
+    def ensure_unique_id(options)
       begin
-        self.id = yield
-      end while self.class.exists?(:id => self.id)
+        self.send "#{options[:field]}=", yield
+      end while self.class.exists?(options[:field] => self.send(options[:field]))
     end
   end
 
   module ClassMethods
-    def acts_as_random_id(options={}, &block)
+    # Default options:
+    # :field => :id
+    def acts_as_random_id(options={:field => :id}, &block)
       before_create do |record|
         if block
-          record.ensure_unique_id(&block)
+          record.ensure_unique_id(options, &block)
         else
-          record.ensure_unique_id do
+          record.ensure_unique_id(options) do
             rand(2_147_483_647) + 1 # mysql and SQLite type integer
           end
         end

data/test/acts_as_random_id_test.rb

@@ -7,6 +7,12 @@ class ActsAsRandomIdTest < Test::Unit::TestCase
   class Comment < ActiveRecord::Base
     acts_as_random_id
   end
+  
+  class Payment < ActiveRecord::Base
+    acts_as_random_id :field => :secure_identifier do
+      Time.now.to_i
+    end
+  end
 
   class Group < ActiveRecord::Base
     acts_as_random_id do
@@ -36,4 +42,8 @@ class ActsAsRandomIdTest < Test::Unit::TestCase
     assert_equal Time.now.to_i, a.id
   end
 
+  def test_should_assign_custom_field
+    p = Payment.create
+    assert_equal Time.now.to_i, p.secure_identifier
+  end
 end

data/test/schema.rb

@@ -17,5 +17,10 @@ ActiveRecord::Schema.define(:version => 20090217091952) do
     t.datetime "created_at"
     t.datetime "updated_at"
   end
-
+  
+  create_table "payments", :force => true do |t|
+    t.text     "secure_identifier"
+    t.datetime "created_at"
+    t.datetime "updated_at"
+  end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: hashtrain-acts_as_random_id
 version: !ruby/object:Gem::Version 
-  version: 1.0.3
+  version: 1.1.0
 platform: ruby
 authors: 
 - hashtrain.com