active_record_samplooper 0.0.3 → 0.0.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: edebddabcb8d0f99e3b1023410db3374a7df84a3
-  data.tar.gz: 00460ad5ffeda8801cdc44742b9a14e99e0a4fa1
+  metadata.gz: 93d485e74aa6069680edeb330cc7ce8c4c8a376c
+  data.tar.gz: ab4b473a2b67acc840bbbf937af758f49e0b82b0
 SHA512:
-  metadata.gz: 77d0d17c3d8bee4bae8395d6501a66ce2232de391dfd6d828b4cf276d0313d628df371ed2fd183a03a74380b50fbef0909a84ed98deb912fcd65394cc32480e8
-  data.tar.gz: 5b15e0a17a67a22363b90c1876e2e773492996477a2c7ed2595c11da39006b185144762c1bf40aeeb889bcea7289d1d74e0bc03a58334ca5f38254dd01655688
+  metadata.gz: 360c1d888bbf00a5704752c3e7c75a641e9f33286612c847bf59cee523d39e90ef4d5706d3571760d29c51c3c557bd088e9f228be2c46d33b84c8d7b5af557db
+  data.tar.gz: a5621a045cf40a4cc1d1fbe504e0d68cb4a8a771cbc7e2c28d7ea548670e44126141be7f37a9db7cc7fd93588d5521dc1d76802526e015565cf89ce2dd7afd7c

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    active_record_samplooper (0.0.2)
+    active_record_samplooper (0.0.3)
       rails (~> 4.2.1)
 
 GEM

data/lib/active_record_samplooper.rb

@@ -10,6 +10,32 @@ module ActiveRecordSamplooper
     end
   end
 
+  class ArrayLooper
+    attr_accessor :array
+
+
+    class << self
+      def call(*args)
+        new(*args)
+      end
+    end
+
+
+    def initialize(array)
+      self.array = array.dup
+    end
+
+
+    def find(id)
+      raise ActiveRecord::RecordNotFound unless id
+      array[id - 1] || raise(Gone, id)
+    end
+
+
+    def pluck(*)
+      (1..array.size).to_a
+    end
+  end
 
   class Samplooper
     attr_accessor :klass, :id_store, :rest
@@ -28,20 +54,18 @@ module ActiveRecordSamplooper
     end
 
 
-    def sample
-      find(id_store.sample)
+    def sample(count = 1)
+      count > 1 ? count.times.map { do_sample } : do_sample
     end
 
 
-    def pick
-      return if rest.blank?
-      find(rest.shift)
+    def pick(count = 1)
+      count > 1 ? count.times.map { do_pick } : do_pick
     end
 
 
-    def loop
-      reset! if rest.blank?
-      sample
+    def loop(count = 1)
+      count > 1 ? count.times.map { do_loop } : do_loop
     end
 
 
@@ -54,6 +78,25 @@ module ActiveRecordSamplooper
     def reset!
       self.rest = id_store.dup
     end
+
+
+    private
+    def do_sample
+      find(id_store.sample)
+    end
+
+
+    def do_pick
+      return if rest.blank?
+      find(rest.shift)
+    end
+
+
+    def do_loop
+      reset! if rest.blank?
+      pick
+    end
+
   end
 
   class Gone < StandardError
@@ -66,6 +109,13 @@ module ActiveRecordSamplooper
   end
 end
 
+class ::Array
+  def sampler
+    ActiveRecordSamplooper.(ActiveRecordSamplooper::ArrayLooper.(self))
+  end
+end
+
+
 class ::ActiveRecord::Base
   class << self
     def sample

data/lib/active_record_samplooper/version.rb

@@ -1,3 +1,3 @@
 module ActiveRecordSamplooper
-  VERSION = "0.0.3"
+  VERSION = "0.0.4"
 end

data/spec/models/sample_model_spec.rb

@@ -59,7 +59,7 @@ RSpec.describe SampleModel, type: :model do
       it 'loop get destroyed raise exception' do
         sampler
         SampleModel.first.destroy
-        SampleModel.first.destroy
+        SampleModel.last.destroy
         expect {
           10.times { sampler.loop }
         }.to raise_exception(ActiveRecordSamplooper::Gone)
@@ -89,4 +89,53 @@ RSpec.describe SampleModel, type: :model do
       expect(sampler.pick).to be_nil
     end
   end
+
+  describe 'array samplooper' do
+    let(:array) { (:a..:f).to_a }
+
+    describe 'from sampler' do
+      let(:sampler) { array.sampler }
+
+      describe 'sampler pick' do
+        it 'any times, get SampleModel instance' do
+          36.times { expect(sampler.sample).to be_a(Symbol) }
+        end
+      end
+
+      it 'sample get once each' do
+        6.times { expect(sampler.pick).to be_a(Symbol) }
+        expect(sampler.pick).to be_nil
+      end
+
+      it 'loop get any times each' do
+        100.times do
+          all = (:a..:f).to_a if all.blank?
+
+          expect(all.delete(sampler.loop)).not_to be_nil
+        end
+      end
+
+      it 'sample get destroyed raise exception' do
+        array[4] = nil
+        sampler
+        expect {
+          7.times { sampler.pick }
+        }.to raise_exception(ActiveRecordSamplooper::Gone)
+      end
+
+      it 'loop get destroyed raise exception' do
+        array[4] = nil
+        sampler
+        expect {
+          7.times { sampler.loop }
+        }.to raise_exception(ActiveRecordSamplooper::Gone)
+      end
+
+      it 'sampler not include new instance' do
+        sampler
+        array.push(:aa)
+        1000.times { expect(sampler.sample).not_to eq(:aa) }
+      end
+    end
+  end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: active_record_samplooper
 version: !ruby/object:Gem::Version
-  version: 0.0.3
+  version: 0.0.4
 platform: ruby
 authors:
 - mmmpa
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2015-06-25 00:00:00.000000000 Z
+date: 2015-06-26 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rails