stages 1.0.1 → 1.1.1

data/.travis.yml

@@ -0,0 +1,8 @@
+language: ruby
+rvm:
+  - 1.9.2
+  - 1.9.3
+script: "rake test"
+before_install:
+  - gem update --system 1.8.25
+  - gem --version

data/README.md

@@ -1,3 +1,5 @@
+[![Build Status](https://travis-ci.org/iGoDigital-LLC/stages.png)](https://travis-ci.org/iGoDigital-LLC/stages)
+
 Stages
 ------
 

data/VERSION

@@ -1 +1 @@
-1.0.1
+1.1.1

data/lib/stage_base.rb

@@ -1,5 +1,6 @@
 module Stages
   class Stage
+    include Enumerable
     attr_accessor :source
 
     def initialize(&block)
@@ -56,6 +57,18 @@ module Stages
       end
     end
 
+    def to_enum
+      Enumerator.new do |y|
+        while !done?
+          y << run
+        end
+      end
+    end
+
+    def each(&block)
+      to_enum.each(&block)
+    end
+
     def handle_value(value)
       output value
     end

data/lib/stages/each.rb

@@ -9,9 +9,8 @@ module Stages
     def process
       if @things
         process_things
-      else
-        process_inputs
       end
+      process_inputs
     end
 
     def process_inputs
@@ -19,7 +18,7 @@ module Stages
         v = input
         v = @block.call(v) if @block
         v.each do |v|
-          output v
+          handle_value v
         end
       end
     end

data/lib/stages/limit.rb

@@ -0,0 +1,21 @@
+module Stages
+  class Limit < Stage
+    def initialize(allowed)
+      @allowed = allowed
+      @sent = 0
+      super()
+    end
+
+    def process
+      while @sent < @allowed
+        handle_value input
+        @sent += 1
+      end
+    end
+
+     def reset
+       @sent = 0
+       super()
+     end
+  end
+end

data/lib/sugar.rb

@@ -43,5 +43,9 @@ module Stages
     def feeder(*args, &block)
       Feeder.new(*args, &block)
     end
+
+    def limit(*args, &block)
+      Limit.new(*args, &block)
+    end
   end
 end

data/stages.gemspec

@@ -5,11 +5,11 @@
 
 Gem::Specification.new do |s|
   s.name = "stages"
-  s.version = "1.0.1"
+  s.version = "1.1.1"
 
   s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
   s.authors = ["Nathan D Acuff"]
-  s.date = "2013-03-01"
+  s.date = "2013-08-21"
   s.description = "pipeline builder"
   s.email = "support@igodigital.com"
   s.extra_rdoc_files = [
@@ -17,6 +17,7 @@ Gem::Specification.new do |s|
   ]
   s.files = [
     ".autotest",
+    ".travis.yml",
     "Gemfile",
     "Gemfile.lock",
     "README.md",
@@ -33,6 +34,7 @@ Gem::Specification.new do |s|
     "lib/stages/exhaust.rb",
     "lib/stages/exhaust_count.rb",
     "lib/stages/feeder.rb",
+    "lib/stages/limit.rb",
     "lib/stages/map.rb",
     "lib/stages/select.rb",
     "lib/stages/unique.rb",

data/test/test_pipeline.rb

@@ -61,9 +61,30 @@ class TestPipeline < MiniTest::Unit::TestCase
     assert_equal(2, pipeline.run)
   end
 
+  test 'reset resets limit' do
+    pipeline = Each.new([1, 2, 3, 4, 5]) | Limit.new(2)
+    assert_equal(1, pipeline.run)
+    assert_equal(2, pipeline.run)
+    pipeline.reset
+    assert_equal(1, pipeline.run)
+    assert_equal(2, pipeline.run)
+    assert_equal(:stages_eos, pipeline.run)
+  end
+
   test 'nil stages are skipped/no-ops' do
     pipeline = Evens.new | nil | Select.new{ |x| x > 3}
     assert_equal(4, pipeline.run)
     assert_equal(6, pipeline.run)
   end
+
+  test "pipeline as enum" do
+    pipeline = Each.new([1,2,3,4,5]) | Select.new{ |x| x % 2 == 0}
+    output = pipeline.map{ |x| x }
+    assert_equal([2,4], output)
+
+    pipeline.reset
+    pipeline.each_slice(2) do |thing|
+      assert_equal([2, 4], thing)
+    end
+  end
 end

data/test/test_stages.rb

@@ -164,6 +164,27 @@ class TestStages < MiniTest::Unit::TestCase
     assert pipeline.done?
   end
 
+  test 'each with input and incoming data prepends given data' do
+    pipeline = Emit.new([1,2,3]) | Each.new([4,5,6]) | Exhaust.new
+    assert_equal([4,5,6,1,2,3], pipeline.run)
+  end
+
+  test 'limit stage only ticks X times' do
+    pipeline = Each.new([1,2,3,4,4]) | Limit.new(2) | Exhaust.new
+    assert_equal([1,2], pipeline.run)
+  end
+
+  test "stage as enum" do
+    pipeline = Each.new([1,2,3,4,5])
+    output = pipeline.to_enum.map{ |x| x }
+    assert_equal([1,2,3,4,5], output)
+
+    pipeline.reset
+    output = []
+    pipeline.to_enum.each_slice(2){ |x| output << x }
+    assert_equal([[1,2],[3,4],[5]], output)
+  end
+
   def sing
     { :do => 'doe a deer a female deer',
       :re => 'ray a drop of golden sun',

data/test/test_syntax.rb

@@ -52,4 +52,9 @@ class TestSyntax < MiniTest::Unit::TestCase
     assert_equal(6, pipeline.run)
     assert pipeline.done?
   end
+
+  test 'limit' do
+    pipeline = each([1,2,3]) | limit(2) | run_until_exhausted
+    assert_equal([1, 2], pipeline.run)
+  end
 end

metadata

@@ -2,7 +2,7 @@
 name: stages
 version: !ruby/object:Gem::Version 
   prerelease: 
-  version: 1.0.1
+  version: 1.1.1
 platform: ruby
 authors: 
 - Nathan D Acuff
@@ -10,7 +10,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2013-03-01 00:00:00 Z
+date: 2013-08-21 00:00:00 Z
 dependencies: 
 - !ruby/object:Gem::Dependency 
   name: rake
@@ -33,6 +33,7 @@ extra_rdoc_files:
 - README.md
 files: 
 - .autotest
+- .travis.yml
 - Gemfile
 - Gemfile.lock
 - README.md
@@ -49,6 +50,7 @@ files:
 - lib/stages/exhaust.rb
 - lib/stages/exhaust_count.rb
 - lib/stages/feeder.rb
+- lib/stages/limit.rb
 - lib/stages/map.rb
 - lib/stages/select.rb
 - lib/stages/unique.rb
@@ -72,7 +74,7 @@ required_ruby_version: !ruby/object:Gem::Requirement
   requirements: 
   - - ">="
     - !ruby/object:Gem::Version 
-      hash: 152651317607917436
+      hash: 730254182358055923
       segments: 
       - 0
       version: "0"