test/vagrant/command/helpers_test.rb in vagrantup-0.6.8 vs test/vagrant/command/helpers_test.rb in vagrantup-0.6.9

@@ -37,11 +37,11 @@
       assert instance.target_vms.equal?(result)
     end
 
     context "without multivm" do
       setup do
-        @env.stubs(:vms).returns({ :one => 1, :two => 2 })
+        @env.stubs(:vms_ordered => [1, 2], :vms => {:one => 1, :two => 2})
       end
 
       should "raise an exception if a name is specified" do
         instance = command(["foo"], vagrant_env)
         assert_raises(Vagrant::Errors::MultiVMEnvironmentRequired) {
@@ -57,10 +57,10 @@
       end
     end
 
     context "with multivm" do
       setup do
-        @env.stubs(:vms).returns(:one => 1, :two => 2)
+        @env.stubs(:vms_ordered => [1, 2], :vms => {:one => 1, :two => 2})
       end
 
       should "return all the VMs if no name is specified" do
         instance = command([], @env)
         assert_equal @env.vms.values, instance.target_vms