test/vagrant/provisioners/base_test.rb in vagrant-0.3.4 vs test/vagrant/provisioners/base_test.rb in vagrant-0.4.0

@@ -5,16 +5,16 @@
     assert Vagrant::Provisioners::Base.include?(Vagrant::Util)
   end
 
   context "base instance" do
     setup do
-      @env = mock_environment
-      @base = Vagrant::Provisioners::Base.new(@env)
+      @vm = mock("vm")
+      @base = Vagrant::Provisioners::Base.new(@vm)
     end
 
     should "set the environment" do
-      base = Vagrant::Provisioners::Base.new(@env)
-      assert_equal @env, base.env
+      base = Vagrant::Provisioners::Base.new(@vm)
+      assert_equal @vm, base.vm
     end
 
     should "implement provision! which does nothing" do
       assert_nothing_raised do
         assert @base.respond_to?(:provision!)