test/vagrant/command/helpers_test.rb in vagrantup-0.6.4 vs test/vagrant/command/helpers_test.rb in vagrantup-0.6.5

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