spec/spec_helper.rb in amee-data-abstraction-2.2.0 vs spec/spec_helper.rb in amee-data-abstraction-2.2.1

@@ -259,24 +259,29 @@
           mock_di.should_receive(:value).with(k).and_return(v)
         else
           mock_di.should_receive(:value).with(k).and_return(v).once
         end
       end
-      test.flexmock(AMEE::Data::Item).should_receive(:get).
-        with(connection,dipath,{}).
-        at_least.once.
-        and_return(mock_di)
+      if once
+        test.flexmock(AMEE::Data::Item).should_receive(:get).
+          with(connection,dipath,{}).
+          at_least.once.
+          and_return(mock_di)
+      else
+        test.flexmock(AMEE::Data::Item).should_receive(:get).
+          with(connection,dipath,{}).
+          and_return(mock_di)
+      end
     end
     if once
       test.flexmock(AMEE::Profile::Item).should_receive(:get).
       with(connection,pipath,{}).
       at_least.once.
       and_return(mock_pi)
     else
-    test.flexmock(AMEE::Profile::Item).should_receive(:get).
-      with(connection,pipath,{}).
-      at_least.once.
-      and_return(mock_pi)
+      test.flexmock(AMEE::Profile::Item).should_receive(:get).
+        with(connection,pipath,{}).
+        and_return(mock_pi)
     end
     return self
   end
 
   def delete
\ No newline at end of file