streaming_step_spec.rb in elasticity-2.5.5

- old
+ new

@@ -1,29 +1,30 @@
 describe Elasticity::StreamingStep do
 
   subject do
-    Elasticity::StreamingStep.new('INPUT_BUCKET', 'OUTPUT_BUCKET', 'MAPPER', 'REDUCER')
+    Elasticity::StreamingStep.new('INPUT_BUCKET', 'OUTPUT_BUCKET', 'MAPPER', 'REDUCER', '-ARG1', 'VALUE1')
   end
 
   it { should be_a Elasticity::JobFlowStep }
 
   its(:name) { should == 'Elasticity Streaming Step' }
   its(:action_on_failure) { should == 'TERMINATE_JOB_FLOW' }
   its(:input_bucket) { should == 'INPUT_BUCKET' }
   its(:output_bucket) { should == 'OUTPUT_BUCKET' }
   its(:mapper) { should == 'MAPPER' }
   its(:reducer) { should == 'REDUCER' }
+  its(:arguments) { should == %w(-ARG1 VALUE1) }
 
   describe '#to_aws_step' do
 
     it 'should convert to aws step format' do
       subject.to_aws_step(Elasticity::JobFlow.new('_', '_')).should == {
         :name => 'Elasticity Streaming Step',
         :action_on_failure => 'TERMINATE_JOB_FLOW',
         :hadoop_jar_step => {
           :jar => '/home/hadoop/contrib/streaming/hadoop-streaming.jar',
-          :args => %w(-input INPUT_BUCKET -output OUTPUT_BUCKET -mapper MAPPER -reducer REDUCER),
+          :args => %w(-input INPUT_BUCKET -output OUTPUT_BUCKET -mapper MAPPER -reducer REDUCER -ARG1 VALUE1),
         },
       }
     end
 
   end
@@ -32,6 +33,6 @@
     it 'should not require installation' do
       Elasticity::StreamingStep.requires_installation?.should be_false
     end
   end
 
-end
\ No newline at end of file
+end