diff --git a/test/unit/vagrant/config/v2/loader_test.rb b/test/unit/vagrant/config/v2/loader_test.rb index d88c6724e..564e69940 100644 --- a/test/unit/vagrant/config/v2/loader_test.rb +++ b/test/unit/vagrant/config/v2/loader_test.rb @@ -22,7 +22,7 @@ describe Vagrant::Config::V2::Loader do # Register a plugin for our test register_plugin("2") do |plugin| plugin.config "foo" do - Class.new do + Class.new(Vagrant.plugin("2", "config")) do attr_accessor :bar def finalize! diff --git a/test/unit/vagrant/config/v2/root_test.rb b/test/unit/vagrant/config/v2/root_test.rb index 623269205..fdd8c3d41 100644 --- a/test/unit/vagrant/config/v2/root_test.rb +++ b/test/unit/vagrant/config/v2/root_test.rb @@ -43,7 +43,7 @@ describe Vagrant::Config::V2::Root do describe "#finalize!" do it "should call #finalize!" do - foo_class = Class.new do + foo_class = Class.new(Vagrant.plugin("2", "config")) do attr_accessor :foo def finalize! @@ -59,7 +59,7 @@ describe Vagrant::Config::V2::Root do end it "should call #_finalize!" do - klass = Class.new + klass = Class.new(Vagrant.plugin("2", "config")) klass.any_instance.should_receive(:finalize!) klass.any_instance.should_receive(:_finalize!)