puppet 6.4.5-universal-darwin → 6.5.0-universal-darwin

data/spec/unit/application/device_spec.rb

@@ -370,19 +370,9 @@ describe Puppet::Application::Device do
         allow(configurer).to receive(:run)
         allow(Puppet::Configurer).to receive(:new).and_return(configurer)
 
-        allow(Puppet::FileSystem).to receive(:exist?)
-        allow(Puppet::FileSystem).to receive(:symlink)
-        allow(Puppet::FileSystem).to receive(:dir_mkpath).and_return(true)
-        allow(Puppet::FileSystem).to receive(:dir_exist?).and_return(true)
-
         allow(plugin_handler).to receive(:download_plugins)
       end
 
-      it "sets ssldir relative to the global confdir" do
-        expect(Puppet).to receive(:[]=).with(:ssldir, make_absolute("/dummy/devices/device1/ssl"))
-        expect { device.main }.to exit_with 1
-      end
-
       it "sets vardir to the device vardir" do
         expect(Puppet).to receive(:[]=).with(:vardir, make_absolute("/dummy/devices/device1"))
         expect { device.main }.to exit_with 1
@@ -400,22 +390,6 @@ describe Puppet::Application::Device do
       end
 
       context 'with --target=device1' do
-        it "symlinks the ssl directory if it doesn't exist" do
-          allow(device.options).to receive(:[]).with(:target).and_return('device1')
-          allow(Puppet::FileSystem).to receive(:exist?).and_return(false)
-
-          expect(Puppet::FileSystem).to receive(:symlink).with(Puppet[:ssldir], File.join(Puppet[:confdir], 'ssl')).and_return(true)
-          expect { device.main }.to exit_with 1
-        end
-
-        it "creates the device confdir under the global confdir" do
-          allow(device.options).to receive(:[]).with(:target).and_return('device1')
-          allow(Puppet::FileSystem).to receive(:dir_exist?).and_return(false)
-
-          expect(Puppet::FileSystem).to receive(:dir_mkpath).with(Puppet[:ssldir]).and_return(true)
-          expect { device.main }.to exit_with 1
-        end
-
         it "manages the specified target" do
           allow(device.options).to receive(:[]).with(:target).and_return('device1')
 
@@ -464,7 +438,7 @@ describe Puppet::Application::Device do
           allow(device.options).to receive(:[]).with(:to_yaml).and_return(true)
           allow(device.command_line).to receive(:args).and_return(['user'])
           expect(Puppet::Resource.indirection).to receive(:search).with('user/', {}).and_return(resources)
-          expect(device).to receive(:puts).with("---\nuser:\n  title: {}\n")
+          expect(device).to receive(:puts).with("user:\n  title:\n")
           expect { device.main }.to exit_with 0
         end
       end

data/spec/unit/application/ssl_spec.rb

@@ -1,5 +1,6 @@
 require 'spec_helper'
 require 'puppet/application/ssl'
+require 'webmock/rspec'
 require 'openssl'
 require 'puppet/test_ca'
 
@@ -22,17 +23,22 @@ describe Puppet::Application::Ssl, unless: Puppet::Util::Platform.jruby? do
   end
 
   before do
+    WebMock.disable_net_connect!
+
+    allow_any_instance_of(Net::HTTP).to receive(:start)
+    allow_any_instance_of(Net::HTTP).to receive(:finish)
+
     Puppet.settings.use(:main)
     Puppet[:certname] = name
     Puppet[:vardir] = tmpdir("ssl_testing")
 
     # Host assumes ca cert and crl are present
-    File.open(Puppet[:localcacert], 'w') { |f| f.write(@ca_cert.to_pem) }
-    File.open(Puppet[:hostcrl], 'w') { |f| f.write(@crl.to_pem) }
+    File.write(Puppet[:localcacert], @ca_cert.to_pem)
+    File.write(Puppet[:hostcrl], @crl.to_pem)
 
     # Setup our ssl client
-    File.open(Puppet[:hostprivkey], 'w') { |f| f.write(@host[:private_key].to_pem) }
-    File.open(Puppet[:hostpubkey], 'w') { |f| f.write(@host[:private_key].public_key.to_pem) }
+    File.write(Puppet[:hostprivkey], @host[:private_key].to_pem)
+    File.write(Puppet[:hostpubkey], @host[:private_key].public_key.to_pem)
   end
 
   def expects_command_to_pass(expected_output = nil)
@@ -100,11 +106,22 @@ describe Puppet::Application::Ssl, unless: Puppet::Util::Platform.jruby? do
 
     it_behaves_like 'an ssl action'
 
-    it 'registers OIDs' do
+    it 'generates an RSA private key' do
+      File.unlink(Puppet[:hostprivkey])
+
+      stub_request(:put, %r{puppet-ca/v1/certificate_request/#{name}}).to_return(status: 200)
+      stub_request(:get, %r{puppet-ca/v1/certificate/#{name}}).to_return(status: 404)
+
+      expects_command_to_pass(%r{Submitted certificate request for '#{name}' to https://.*})
+    end
+
+    it 'generates an EC private key' do
+      Puppet[:key_type] = 'ec'
+      File.unlink(Puppet[:hostprivkey])
+
       stub_request(:put, %r{puppet-ca/v1/certificate_request/#{name}}).to_return(status: 200)
       stub_request(:get, %r{puppet-ca/v1/certificate/#{name}}).to_return(status: 404)
 
-      expect(Puppet::SSL::Oids).to receive(:register_puppet_oids)
       expects_command_to_pass(%r{Submitted certificate request for '#{name}' to https://.*})
     end
 
@@ -168,7 +185,7 @@ describe Puppet::Application::Ssl, unless: Puppet::Util::Platform.jruby? do
     end
 
     it 'overwrites an existing cert' do
-      File.open(Puppet[:hostcert], 'w') { |f| f.write "existing certificate" }
+      File.write(Puppet[:hostcert], "existing certificate")
 
       stub_request(:get, %r{puppet-ca/v1/certificate/#{name}}).to_return(status: 200, body: @host[:cert].to_pem)
 
@@ -178,12 +195,12 @@ describe Puppet::Application::Ssl, unless: Puppet::Util::Platform.jruby? do
     end
 
     it "reports an error if the downloaded cert's public key doesn't match our private key" do
-      File.open(Puppet[:hostcert], 'w') { |f| f.write "existing cert" }
+      File.write(Puppet[:hostcert], "existing cert")
 
       # generate a new host key, whose public key doesn't match the cert
       private_key = OpenSSL::PKey::RSA.new(512)
-      File.open(Puppet[:hostprivkey], 'w') { |f| f.write(private_key.to_pem) }
-      File.open(Puppet[:hostpubkey], 'w') { |f| f.write(private_key.public_key.to_pem) }
+      File.write(Puppet[:hostprivkey], private_key.to_pem)
+      File.write(Puppet[:hostpubkey], private_key.public_key.to_pem)
 
       stub_request(:get, %r{puppet-ca/v1/certificate/#{name}}).to_return(status: 200, body: @host[:cert].to_pem)
 
@@ -203,7 +220,7 @@ describe Puppet::Application::Ssl, unless: Puppet::Util::Platform.jruby? do
     before do
       ssl.command_line.args << 'verify'
 
-      File.open(Puppet[:hostcert], 'w') { |f| f.write(@host[:cert].to_pem) }
+      File.write(Puppet[:hostcert], @host[:cert].to_pem)
     end
 
     it 'reports if the key is missing' do
@@ -222,8 +239,8 @@ describe Puppet::Application::Ssl, unless: Puppet::Util::Platform.jruby? do
       # generate new keys
       private_key = OpenSSL::PKey::RSA.new(512)
       public_key = private_key.public_key
-      File.open(Puppet[:hostprivkey], 'w') { |f| f.write(private_key.to_pem) }
-      File.open(Puppet[:hostpubkey], 'w') { |f| f.write(public_key.to_pem) }
+      File.write(Puppet[:hostprivkey], private_key.to_pem)
+      File.write(Puppet[:hostpubkey], public_key.to_pem)
 
       expects_command_to_fail(%r{The certificate for 'CN=ssl-client' does not match its private key})
     end
@@ -231,19 +248,35 @@ describe Puppet::Application::Ssl, unless: Puppet::Util::Platform.jruby? do
     it 'reports if the cert verification fails' do
       # generate a new CA to force an error
       new_ca = Puppet::TestCa.new
-      File.open(Puppet[:localcacert], 'w') { |f| f.write(new_ca.ca_cert.to_pem) }
+      File.write(Puppet[:localcacert], new_ca.ca_cert.to_pem)
 
       # and CRL for that CA
-      File.open(Puppet[:hostcrl], 'w') { |f| f.write(new_ca.ca_crl.to_pem) }
+      File.write(Puppet[:hostcrl], new_ca.ca_crl.to_pem)
 
       expects_command_to_fail(%r{Invalid signature for certificate 'CN=ssl-client'})
     end
 
     it 'reports when verification succeeds' do
-      allow_any_instance_of(OpenSSL::X509::Store).to receive(:verify).and_return(true)
-
       expects_command_to_pass(%r{Verified client certificate 'CN=ssl-client' fingerprint})
     end
+
+    it 'reports when verification succeeds with a password protected private key' do
+      FileUtils.cp(File.join(PuppetSpec::FIXTURE_DIR, 'ssl', 'encrypted-key.pem'), Puppet[:hostprivkey])
+      FileUtils.cp(File.join(PuppetSpec::FIXTURE_DIR, 'ssl', 'signed.pem'), Puppet[:hostcert])
+
+      Puppet[:passfile] = file_containing('passfile', '74695716c8b6')
+
+      expects_command_to_pass(%r{Verified client certificate 'CN=signed' fingerprint})
+    end
+
+    it 'reports if the private key password is incorrect' do
+      FileUtils.cp(File.join(PuppetSpec::FIXTURE_DIR, 'ssl', 'encrypted-key.pem'), Puppet[:hostprivkey])
+      FileUtils.cp(File.join(PuppetSpec::FIXTURE_DIR, 'ssl', 'signed.pem'), Puppet[:hostcert])
+
+      Puppet[:passfile] = file_containing('passfile', 'wrongpassword')
+
+      expects_command_to_fail(/Failed to load private key for host 'ssl-client'/)
+    end
   end
 
   context 'when cleaning' do
@@ -252,32 +285,32 @@ describe Puppet::Application::Ssl, unless: Puppet::Util::Platform.jruby? do
     end
 
     it 'deletes the hostcert' do
-      File.open(Puppet[:hostcert], 'w') { |f| f.write(@host[:cert].to_pem) }
+      File.write(Puppet[:hostcert], @host[:cert].to_pem)
 
       expects_command_to_pass(%r{Removed certificate #{Puppet[:cert]}})
     end
 
     it 'deletes the private key' do
-      File.open(Puppet[:hostprivkey], 'w') { |f| f.write(@host[:private_key].to_pem) }
+      File.write(Puppet[:hostprivkey], @host[:private_key].to_pem)
 
       expects_command_to_pass(%r{Removed private key #{Puppet[:hostprivkey]}})
     end
 
     it 'deletes the public key' do
-      File.open(Puppet[:hostpubkey], 'w') { |f| f.write(@host[:private_key].public_key.to_pem) }
+      File.write(Puppet[:hostpubkey], @host[:private_key].public_key.to_pem)
 
       expects_command_to_pass(%r{Removed public key #{Puppet[:hostpubkey]}})
     end
 
     it 'deletes the request' do
       path = File.join(Puppet[:requestdir], "#{Puppet[:certname]}.pem")
-      File.open(path, 'w') { |f| f.write(@host[:csr].to_pem) }
+      File.write(path, @host[:csr].to_pem)
 
       expects_command_to_pass(%r{Removed certificate request #{path}})
     end
 
     it 'deletes the passfile' do
-      File.open(Puppet[:passfile], 'w') { |_| }
+      FileUtils.touch(Puppet[:passfile])
 
       expects_command_to_pass(%r{Removed private key password file #{Puppet[:passfile]}})
     end
@@ -306,13 +339,13 @@ describe Puppet::Application::Ssl, unless: Puppet::Util::Platform.jruby? do
       end
 
       it 'deletes the local CA cert' do
-        File.open(Puppet[:localcacert], 'w') { |f| f.write(@ca_cert.to_pem) }
+        File.write(Puppet[:localcacert], @ca_cert.to_pem)
 
         expects_command_to_pass(%r{Removed local CA certificate #{Puppet[:localcacert]}})
       end
 
       it 'deletes the local CRL' do
-        File.open(Puppet[:hostcrl], 'w') { |f| f.write(@crl.to_pem) }
+        File.write(Puppet[:hostcrl], @crl.to_pem)
 
         expects_command_to_pass(%r{Removed local CRL #{Puppet[:hostcrl]}})
       end
@@ -331,7 +364,7 @@ describe Puppet::Application::Ssl, unless: Puppet::Util::Platform.jruby? do
       end
 
       it "cleans the cert when the CA is local and the CA has already cleaned its cert" do
-        File.open(Puppet[:hostcert], 'w') { |f| f.write(@host[:cert].to_pem) }
+        File.write(Puppet[:hostcert], @host[:cert].to_pem)
 
         stub_request(:get, %r{puppet-ca/v1/certificate/puppetserver}).to_return(status: 404)
 
@@ -339,7 +372,7 @@ describe Puppet::Application::Ssl, unless: Puppet::Util::Platform.jruby? do
       end
 
       it "cleans the cert when run on a puppetserver that isn't the CA" do
-        File.open(Puppet[:hostcert], 'w') { |f| f.write(@host[:cert].to_pem) }
+        File.write(Puppet[:hostcert], @host[:cert].to_pem)
 
         Puppet[:ca_server] = 'caserver'
 
@@ -357,7 +390,7 @@ describe Puppet::Application::Ssl, unless: Puppet::Util::Platform.jruby? do
         device_cert_path = File.join(Puppet[:devicedir], 'device.example.com', 'ssl', 'certs')
         device_cert_file = File.join(device_cert_path, 'device.example.com.pem')
         FileUtils.mkdir_p(device_cert_path)
-        File.open(device_cert_file, 'w') { |f| f.write('device.example.com') }
+        File.write(device_cert_file, 'device.example.com')
         expects_command_to_pass(%r{Removed certificate #{device_cert_file}})
      end
     end
@@ -368,14 +401,6 @@ describe Puppet::Application::Ssl, unless: Puppet::Util::Platform.jruby? do
       ssl.command_line.args << 'bootstrap'
     end
 
-    it 'registers the OIDs' do
-      expect_any_instance_of(Puppet::SSL::StateMachine).to receive(:ensure_client_certificate).and_return(
-        double('ssl_context')
-      )
-      expect(Puppet::SSL::Oids).to receive(:register_puppet_oids)
-      expects_command_to_pass
-    end
-
     it 'returns an SSLContext with the loaded CA certs, CRLs, private key and client cert' do
       expect_any_instance_of(Puppet::SSL::StateMachine).to receive(:ensure_client_certificate).and_return(
         double('ssl_context')

data/spec/unit/configurer_spec.rb

@@ -1,90 +1,108 @@
 require 'spec_helper'
 require 'puppet/configurer'
-require 'webmock/rspec'
 
 describe Puppet::Configurer do
   before do
-    Puppet::Node::Facts.indirection.terminus_class = :memory
-    Puppet::Node::Facts.indirection.save(facts)
-
+    allow(Puppet.settings).to receive(:use).and_return(true)
+    @agent = Puppet::Configurer.new
+    allow(@agent).to receive(:init_storage)
+    allow(Puppet::Util::Storage).to receive(:store)
     Puppet[:server] = "puppetmaster"
     Puppet[:report] = true
-
-    catalog.add_resource(resource)
   end
 
-  let(:configurer) { Puppet::Configurer.new }
-  let(:report) { Puppet::Transaction::Report.new }
-  let(:catalog) { Puppet::Resource::Catalog.new("tester", Puppet::Node::Environment.remote(Puppet[:environment].to_sym)) }
-  let(:resource) { Puppet::Resource.new(:notice, 'a') }
-  let(:facts) { Puppet::Node::Facts.new(Puppet[:node_name_value]) }
+  it "should include the Fact Handler module" do
+    expect(Puppet::Configurer.ancestors).to be_include(Puppet::Configurer::FactHandler)
+  end
 
   describe "when executing a pre-run hook" do
     it "should do nothing if the hook is set to an empty string" do
       Puppet.settings[:prerun_command] = ""
-      expect(Puppet::Util::Execution).not_to receive(:execute)
+      expect(Puppet::Util).not_to receive(:exec)
 
-      configurer.execute_prerun_command
+      @agent.execute_prerun_command
     end
 
     it "should execute any pre-run command provided via the 'prerun_command' setting" do
       Puppet.settings[:prerun_command] = "/my/command"
       expect(Puppet::Util::Execution).to receive(:execute).with(["/my/command"]).and_raise(Puppet::ExecutionFailure, "Failed")
 
-      configurer.execute_prerun_command
+      @agent.execute_prerun_command
     end
 
     it "should fail if the command fails" do
       Puppet.settings[:prerun_command] = "/my/command"
       expect(Puppet::Util::Execution).to receive(:execute).with(["/my/command"]).and_raise(Puppet::ExecutionFailure, "Failed")
 
-      expect(configurer.execute_prerun_command).to be_falsey
+      expect(@agent.execute_prerun_command).to be_falsey
     end
   end
 
   describe "when executing a post-run hook" do
     it "should do nothing if the hook is set to an empty string" do
       Puppet.settings[:postrun_command] = ""
-      expect(Puppet::Util::Execution).not_to receive(:execute)
+      expect(Puppet::Util).not_to receive(:exec)
 
-      configurer.execute_postrun_command
+      @agent.execute_postrun_command
     end
 
     it "should execute any post-run command provided via the 'postrun_command' setting" do
       Puppet.settings[:postrun_command] = "/my/command"
       expect(Puppet::Util::Execution).to receive(:execute).with(["/my/command"]).and_raise(Puppet::ExecutionFailure, "Failed")
 
-      configurer.execute_postrun_command
+      @agent.execute_postrun_command
     end
 
     it "should fail if the command fails" do
       Puppet.settings[:postrun_command] = "/my/command"
       expect(Puppet::Util::Execution).to receive(:execute).with(["/my/command"]).and_raise(Puppet::ExecutionFailure, "Failed")
 
-      expect(configurer.execute_postrun_command).to be_falsey
+      expect(@agent.execute_postrun_command).to be_falsey
     end
   end
 
   describe "when executing a catalog run" do
     before do
+      allow(Puppet.settings).to receive(:use).and_return(true)
+      allow(@agent).to receive(:download_plugins)
+      Puppet::Node::Facts.indirection.terminus_class = :memory
+      @facts = Puppet::Node::Facts.new(Puppet[:node_name_value])
+      Puppet::Node::Facts.indirection.save(@facts)
+
+      @catalog = Puppet::Resource::Catalog.new("tester", Puppet::Node::Environment.remote(Puppet[:environment].to_sym))
+      allow(@catalog).to receive(:to_ral).and_return(@catalog)
       Puppet::Resource::Catalog.indirection.terminus_class = :rest
-      allow(Puppet::Resource::Catalog.indirection).to receive(:find).and_return(catalog)
+      allow(Puppet::Resource::Catalog.indirection).to receive(:find).and_return(@catalog)
+      allow(@agent).to receive(:send_report)
+      allow(@agent).to receive(:save_last_run_summary)
+
+      allow(Puppet::Util::Log).to receive(:close_all)
+    end
+
+    after :all do
+      Puppet::Node::Facts.indirection.reset_terminus_class
+      Puppet::Resource::Catalog.indirection.reset_terminus_class
+    end
+
+    it "should initialize storage" do
+      expect(Puppet::Util::Storage).to receive(:load)
+      @agent.run
     end
 
     it "downloads plugins when told" do
-      expect(configurer).to receive(:download_plugins)
-      configurer.run(:pluginsync => true)
+      expect(@agent).to receive(:download_plugins)
+      @agent.run(:pluginsync => true)
     end
 
     it "does not download plugins when told" do
-      expect(configurer).not_to receive(:download_plugins)
-      configurer.run(:pluginsync => false)
+      expect(@agent).not_to receive(:download_plugins)
+      @agent.run(:pluginsync => false)
     end
 
     it "should carry on when it can't fetch its node definition" do
       error = Net::HTTPError.new(400, 'dummy server communication error')
       expect(Puppet::Node.indirection).to receive(:find).and_raise(error)
-      expect(configurer.run).to eq(0)
+      expect(@agent.run).to eq(0)
     end
 
     it "applies a cached catalog when it can't connect to the master" do
@@ -92,157 +110,200 @@ describe Puppet::Configurer do
 
       expect(Puppet::Node.indirection).to receive(:find).and_raise(error)
       expect(Puppet::Resource::Catalog.indirection).to receive(:find).with(anything, hash_including(:ignore_cache => true)).and_raise(error)
-      expect(Puppet::Resource::Catalog.indirection).to receive(:find).with(anything, hash_including(:ignore_terminus => true)).and_return(catalog)
+      expect(Puppet::Resource::Catalog.indirection).to receive(:find).with(anything, hash_including(:ignore_terminus => true)).and_return(@catalog)
 
-      expect(configurer.run).to eq(0)
+      expect(@agent.run).to eq(0)
     end
 
     it "should initialize a transaction report if one is not provided" do
+      report = Puppet::Transaction::Report.new
       expect(Puppet::Transaction::Report).to receive(:new).and_return(report)
 
-      configurer.run
+      @agent.run
     end
 
     it "should respect node_name_fact when setting the host on a report" do
       Puppet[:node_name_fact] = 'my_name_fact'
-      facts.values = {'my_name_fact' => 'node_name_from_fact'}
-      Puppet::Node::Facts.indirection.save(facts)
+      @facts.values = {'my_name_fact' => 'node_name_from_fact'}
+
+      report = Puppet::Transaction::Report.new
 
-      configurer.run(:report => report)
+      @agent.run(:report => report)
       expect(report.host).to eq('node_name_from_fact')
     end
 
-    it "creates a new report when applying the catalog" do
-      options = {}
-      configurer.run(options)
+    it "should pass the new report to the catalog" do
+      report = Puppet::Transaction::Report.new
+      allow(Puppet::Transaction::Report).to receive(:new).and_return(report)
+      expect(@catalog).to receive(:apply).with(hash_including(report: report))
+
+      @agent.run
+    end
+
+    it "should use the provided report if it was passed one" do
+      report = Puppet::Transaction::Report.new
+      expect(@catalog).to receive(:apply).with(hash_including(report: report))
+
+      @agent.run(:report => report)
+    end
+
+    it "should set the report as a log destination" do
+      report = Puppet::Transaction::Report.new
+
+      expect(report).to receive(:<<).with(instance_of(Puppet::Util::Log)).at_least(:once)
 
-      expect(options[:report].metrics['time']['catalog_application']).to be_an_instance_of(Float)
+      @agent.run(:report => report)
     end
 
-    it "uses the provided report when applying the catalog" do
-      configurer.run(:report => report)
+    it "should retrieve the catalog" do
+      expect(@agent).to receive(:retrieve_catalog)
 
-      expect(report.metrics['time']['catalog_application']).to be_an_instance_of(Float)
+      @agent.run
     end
 
     it "should log a failure and do nothing if no catalog can be retrieved" do
-      expect(configurer).to receive(:retrieve_catalog).and_return(nil)
+      expect(@agent).to receive(:retrieve_catalog).and_return(nil)
 
       expect(Puppet).to receive(:err).with("Could not retrieve catalog; skipping run")
 
-      configurer.run
+      @agent.run
     end
 
-    it "passes arbitrary options when applying the catalog" do
-      expect(catalog).to receive(:apply).with(hash_including(one: true))
+    it "should apply the catalog with all options to :run" do
+      expect(@agent).to receive(:retrieve_catalog).and_return(@catalog)
 
-      configurer.run(catalog: catalog, one: true)
+      expect(@catalog).to receive(:apply).with(hash_including(one: true))
+      @agent.run :one => true
+    end
+
+    it "should accept a catalog and use it instead of retrieving a different one" do
+      expect(@agent).not_to receive(:retrieve_catalog)
+
+      expect(@catalog).to receive(:apply)
+      @agent.run :one => true, :catalog => @catalog
     end
 
     it "should benchmark how long it takes to apply the catalog" do
-      configurer.run(report: report)
+      expect(@agent).to receive(:benchmark).with(:notice, instance_of(String))
+
+      expect(@agent).to receive(:retrieve_catalog).and_return(@catalog)
 
-      expect(report.logs).to include(an_object_having_attributes(level: :notice, message: /Applied catalog in .* seconds/))
+      expect(@catalog).not_to receive(:apply) # because we're not yielding
+      @agent.run
+    end
+
+    it "should execute post-run hooks after the run" do
+      expect(@agent).to receive(:execute_postrun_command)
+
+      @agent.run
     end
 
     it "should create report with passed transaction_uuid and job_id" do
-      configurer = Puppet::Configurer.new("test_tuuid", "test_jid")
+      @agent = Puppet::Configurer.new("test_tuuid", "test_jid")
+      allow(@agent).to receive(:init_storage)
 
       report = Puppet::Transaction::Report.new(nil, "test", "aaaa")
       expect(Puppet::Transaction::Report).to receive(:new).with(anything, anything, 'test_tuuid', 'test_jid').and_return(report)
-      expect(configurer).to receive(:send_report).with(report)
+      expect(@agent).to receive(:send_report).with(report)
 
-      configurer.run
+      @agent.run
     end
 
     it "should send the report" do
       report = Puppet::Transaction::Report.new(nil, "test", "aaaa")
       expect(Puppet::Transaction::Report).to receive(:new).and_return(report)
-      expect(configurer).to receive(:send_report).with(report)
+      expect(@agent).to receive(:send_report).with(report)
 
       expect(report.environment).to eq("test")
       expect(report.transaction_uuid).to eq("aaaa")
 
-      configurer.run
+      @agent.run
     end
 
     it "should send the transaction report even if the catalog could not be retrieved" do
-      expect(configurer).to receive(:retrieve_catalog).and_return(nil)
+      expect(@agent).to receive(:retrieve_catalog).and_return(nil)
 
       report = Puppet::Transaction::Report.new(nil, "test", "aaaa")
       expect(Puppet::Transaction::Report).to receive(:new).and_return(report)
-      expect(configurer).to receive(:send_report).with(report)
+      expect(@agent).to receive(:send_report).with(report)
 
       expect(report.environment).to eq("test")
       expect(report.transaction_uuid).to eq("aaaa")
 
-      configurer.run
+      @agent.run
     end
 
     it "should send the transaction report even if there is a failure" do
-      expect(configurer).to receive(:retrieve_catalog).and_raise("whatever")
+      expect(@agent).to receive(:retrieve_catalog).and_raise("whatever")
 
       report = Puppet::Transaction::Report.new(nil, "test", "aaaa")
       expect(Puppet::Transaction::Report).to receive(:new).and_return(report)
-      expect(configurer).to receive(:send_report).with(report)
+      expect(@agent).to receive(:send_report).with(report)
 
       expect(report.environment).to eq("test")
       expect(report.transaction_uuid).to eq("aaaa")
 
-      expect(configurer.run).to be_nil
+      expect(@agent.run).to be_nil
     end
 
     it "should remove the report as a log destination when the run is finished" do
+      report = Puppet::Transaction::Report.new
       expect(Puppet::Transaction::Report).to receive(:new).and_return(report)
 
-      configurer.run
+      @agent.run
 
       expect(Puppet::Util::Log.destinations).not_to include(report)
     end
 
     it "should return the report exit_status as the result of the run" do
+      report = Puppet::Transaction::Report.new
       expect(Puppet::Transaction::Report).to receive(:new).and_return(report)
       expect(report).to receive(:exit_status).and_return(1234)
 
-      expect(configurer.run).to eq(1234)
+      expect(@agent.run).to eq(1234)
     end
 
     it "should return nil if catalog application fails" do
-      expect_any_instance_of(Puppet::Resource::Catalog).to receive(:apply).and_raise(Puppet::Error, 'One or more resource dependency cycles detected in graph')
-      expect(configurer.run(catalog: catalog, report: report)).to be_nil
+      expect(@catalog).to receive(:apply).and_raise(Puppet::Error, 'One or more resource dependency cycles detected in graph')
+      report = Puppet::Transaction::Report.new
+      expect(@agent.run(catalog: @catalog, report: report)).to be_nil
     end
 
     it "should send the transaction report even if the pre-run command fails" do
+      report = Puppet::Transaction::Report.new
       expect(Puppet::Transaction::Report).to receive(:new).and_return(report)
 
       Puppet.settings[:prerun_command] = "/my/command"
       expect(Puppet::Util::Execution).to receive(:execute).with(["/my/command"]).and_raise(Puppet::ExecutionFailure, "Failed")
-      expect(configurer).to receive(:send_report).with(report)
+      expect(@agent).to receive(:send_report).with(report)
 
-      expect(configurer.run).to be_nil
+      expect(@agent.run).to be_nil
     end
 
     it "should include the pre-run command failure in the report" do
+      report = Puppet::Transaction::Report.new
       expect(Puppet::Transaction::Report).to receive(:new).and_return(report)
 
       Puppet.settings[:prerun_command] = "/my/command"
       expect(Puppet::Util::Execution).to receive(:execute).with(["/my/command"]).and_raise(Puppet::ExecutionFailure, "Failed")
 
-      expect(configurer.run).to be_nil
+      expect(@agent.run).to be_nil
       expect(report.logs.find { |x| x.message =~ /Could not run command from prerun_command/ }).to be
     end
 
     it "should send the transaction report even if the post-run command fails" do
+      report = Puppet::Transaction::Report.new
       expect(Puppet::Transaction::Report).to receive(:new).and_return(report)
 
       Puppet.settings[:postrun_command] = "/my/command"
       expect(Puppet::Util::Execution).to receive(:execute).with(["/my/command"]).and_raise(Puppet::ExecutionFailure, "Failed")
-      expect(configurer).to receive(:send_report).with(report)
+      expect(@agent).to receive(:send_report).with(report)
 
-      expect(configurer.run).to be_nil
+      expect(@agent.run).to be_nil
     end
 
     it "should include the post-run command failure in the report" do
+      report = Puppet::Transaction::Report.new
       expect(Puppet::Transaction::Report).to receive(:new).and_return(report)
 
       Puppet.settings[:postrun_command] = "/my/command"
@@ -250,7 +311,7 @@ describe Puppet::Configurer do
 
       expect(report).to receive(:<<) { |log, _| expect(log.message).to match(/Could not run command from postrun_command/) }.at_least(:once)
 
-      expect(configurer.run).to be_nil
+      expect(@agent.run).to be_nil
     end
 
     it "should execute post-run command even if the pre-run command fails" do
@@ -259,42 +320,47 @@ describe Puppet::Configurer do
       expect(Puppet::Util::Execution).to receive(:execute).with(["/my/precommand"]).and_raise(Puppet::ExecutionFailure, "Failed")
       expect(Puppet::Util::Execution).to receive(:execute).with(["/my/postcommand"])
 
-      expect(configurer.run).to be_nil
+      expect(@agent.run).to be_nil
     end
 
     it "should finalize the report" do
+      report = Puppet::Transaction::Report.new
       expect(Puppet::Transaction::Report).to receive(:new).and_return(report)
 
       expect(report).to receive(:finalize_report)
-      configurer.run
+      @agent.run
     end
 
     it "should not apply the catalog if the pre-run command fails" do
+      report = Puppet::Transaction::Report.new
       expect(Puppet::Transaction::Report).to receive(:new).and_return(report)
 
       Puppet.settings[:prerun_command] = "/my/command"
       expect(Puppet::Util::Execution).to receive(:execute).with(["/my/command"]).and_raise(Puppet::ExecutionFailure, "Failed")
 
-      expect_any_instance_of(Puppet::Resource::Catalog).not_to receive(:apply)
-      expect(configurer).to receive(:send_report)
+      expect(@catalog).not_to receive(:apply)
+      expect(@agent).to receive(:send_report)
 
-      expect(configurer.run).to be_nil
+      expect(@agent.run).to be_nil
     end
 
     it "should apply the catalog, send the report, and return nil if the post-run command fails" do
+      report = Puppet::Transaction::Report.new
       expect(Puppet::Transaction::Report).to receive(:new).and_return(report)
 
       Puppet.settings[:postrun_command] = "/my/command"
       expect(Puppet::Util::Execution).to receive(:execute).with(["/my/command"]).and_raise(Puppet::ExecutionFailure, "Failed")
 
-      expect_any_instance_of(Puppet::Resource::Catalog).to receive(:apply)
-      expect(configurer).to receive(:send_report)
+      expect(@catalog).to receive(:apply)
+      expect(@agent).to receive(:send_report)
 
-      expect(configurer.run).to be_nil
+      expect(@agent.run).to be_nil
     end
 
     it 'includes total time metrics in the report after successfully applying the catalog' do
-      configurer.run(report: report)
+      report = Puppet::Transaction::Report.new
+      allow(@catalog).to receive(:apply).with(hash_including(report: report))
+      @agent.run(report: report)
 
       expect(report.metrics['time']).to be
       expect(report.metrics['time']['total']).to be_a_kind_of(Numeric)
@@ -304,15 +370,17 @@ describe Puppet::Configurer do
       Puppet.settings[:prerun_command] = "/my/command"
       expect(Puppet::Util::Execution).to receive(:execute).with(["/my/command"]).and_raise(Puppet::ExecutionFailure, "Failed")
 
-      configurer.run(report: report)
+      report = Puppet::Transaction::Report.new
+      @agent.run(report: report)
 
       expect(report.metrics['time']).to be
       expect(report.metrics['time']['total']).to be_a_kind_of(Numeric)
     end
 
     it 'includes total time metrics in the report even if catalog retrieval fails' do
-      allow(configurer).to receive(:prepare_and_retrieve_catalog_from_cache).and_raise
-      configurer.run(:report => report)
+      report = Puppet::Transaction::Report.new
+      allow(@agent).to receive(:prepare_and_retrieve_catalog_from_cache).and_raise
+      @agent.run(:report => report)
 
       expect(report.metrics['time']).to be
       expect(report.metrics['time']['total']).to be_a_kind_of(Numeric)
@@ -320,146 +388,159 @@ describe Puppet::Configurer do
 
     it "should refetch the catalog if the server specifies a new environment in the catalog" do
       catalog = Puppet::Resource::Catalog.new("tester", Puppet::Node::Environment.remote('second_env'))
-      expect(configurer).to receive(:retrieve_catalog).and_return(catalog).twice
+      expect(@agent).to receive(:retrieve_catalog).and_return(catalog).twice
 
-      configurer.run
+      @agent.run
     end
 
-    it "changes the configurer's environment if the server specifies a new environment in the catalog" do
-      allow_any_instance_of(Puppet::Resource::Catalog).to receive(:environment).and_return("second_env")
+    it "should change the environment setting if the server specifies a new environment in the catalog" do
+      allow(@catalog).to receive(:environment).and_return("second_env")
 
-      configurer.run
+      @agent.run
 
-      expect(configurer.environment).to eq("second_env")
+      expect(@agent.environment).to eq("second_env")
     end
 
-    it "changes the report's environment if the server specifies a new environment in the catalog" do
-      allow_any_instance_of(Puppet::Resource::Catalog).to receive(:environment).and_return("second_env")
+    it "should fix the report if the server specifies a new environment in the catalog" do
+      report = Puppet::Transaction::Report.new(nil, "test", "aaaa")
+      expect(Puppet::Transaction::Report).to receive(:new).and_return(report)
+      expect(@agent).to receive(:send_report).with(report)
+
+      allow(@catalog).to receive(:environment).and_return("second_env")
+      allow(@agent).to receive(:retrieve_catalog).and_return(@catalog)
 
-      configurer.run(report: report)
+      @agent.run
 
       expect(report.environment).to eq("second_env")
     end
 
     it "sends the transaction uuid in a catalog request" do
-      configurer = Puppet::Configurer.new('aaa')
+      @agent.instance_variable_set(:@transaction_uuid, 'aaa')
       expect(Puppet::Resource::Catalog.indirection).to receive(:find).with(anything, hash_including(transaction_uuid: 'aaa'))
-      configurer.run
+      @agent.run
     end
 
     it "sends the transaction uuid in a catalog request" do
-      configurer = Puppet::Configurer.new('b', 'aaa')
+      @agent.instance_variable_set(:@job_id, 'aaa')
       expect(Puppet::Resource::Catalog.indirection).to receive(:find).with(anything, hash_including(job_id: 'aaa'))
-      configurer.run
+      @agent.run
     end
 
     it "sets the static_catalog query param to true in a catalog request" do
       expect(Puppet::Resource::Catalog.indirection).to receive(:find).with(anything, hash_including(static_catalog: true))
-      configurer.run
+      @agent.run
     end
 
     it "sets the checksum_type query param to the default supported_checksum_types in a catalog request" do
       expect(Puppet::Resource::Catalog.indirection).to receive(:find).with(anything,
         hash_including(checksum_type: 'md5.sha256.sha384.sha512.sha224'))
-      configurer.run
+      @agent.run
     end
 
     it "sets the checksum_type query param to the supported_checksum_types setting in a catalog request" do
-      Puppet[:supported_checksum_types] = ['sha256']
       # Regenerate the agent to pick up the new setting
-      configurer = Puppet::Configurer.new
+      Puppet[:supported_checksum_types] = ['sha256']
+      @agent = Puppet::Configurer.new
+      allow(@agent).to receive(:init_storage)
+      allow(@agent).to receive(:download_plugins)
+      allow(@agent).to receive(:send_report)
+      allow(@agent).to receive(:save_last_run_summary)
 
       expect(Puppet::Resource::Catalog.indirection).to receive(:find).with(anything, hash_including(checksum_type: 'sha256'))
-      configurer.run
+      @agent.run
     end
 
     describe "when not using a REST terminus for catalogs" do
       it "should not pass any facts when retrieving the catalog" do
-        # This is weird, we collect facts when constructing the node,
-        # but we don't send them in the indirector request. Then the compiler
-        # looks up the node, and collects its facts, which we could have sent
-        # in the first place. This seems like a bug.
         Puppet::Resource::Catalog.indirection.terminus_class = :compiler
-
+        expect(@agent).not_to receive(:facts_for_uploading)
         expect(Puppet::Resource::Catalog.indirection).to receive(:find) do |name, options|
-          expect(options[:facts]).to be_nil
-        end.and_return(catalog)
+          options[:facts].nil?
+        end.and_return(@catalog)
 
-        configurer.run
+        @agent.run
       end
     end
 
     describe "when using a REST terminus for catalogs" do
-      it "should pass the url encoded facts and facts format as arguments when retrieving the catalog" do
+      it "should pass the prepared facts and the facts format as arguments when retrieving the catalog" do
         Puppet::Resource::Catalog.indirection.terminus_class = :rest
-
-        facts.values = { 'foo' => 'bar' }
-        Puppet::Node::Facts.indirection.save(facts)
-
-        expect(
-          Puppet::Resource::Catalog.indirection
-        ).to receive(:find) do |_, options|
-          expect(options[:facts_format]).to eq("application/json")
-
-          unescaped = JSON.parse(CGI.unescape(options[:facts]))
-          expect(unescaped).to include("values" => {"foo" => "bar"})
-        end.and_return(catalog)
-
-        configurer.run
+        # the "facts_for_uploading" are prepeared by first finding facts, and then encoding them
+        # this mocks the "find" with a special value 12345, which is then expected back in the
+        # call to "encode" - the encode in turn returns mocked data that is asserted as being
+        # presented to the catalog terminus as options.
+        #
+        expect(@agent).to receive(:find_facts).and_return(12345)
+        expect(@agent).to receive(:encode_facts).with(12345).and_return(:facts => "myfacts", :facts_format => :foo)
+        expect(Puppet::Resource::Catalog.indirection).to receive(:find).with(anything, hash_including(facts: "myfacts", facts_format: :foo)).and_return(@catalog)
+
+        @agent.run
       end
     end
   end
 
+  describe "when initialized with a transaction_uuid" do
+    it "stores it" do
+      expect(SecureRandom).not_to receive(:uuid)
+      configurer = Puppet::Configurer.new('foo')
+      expect(configurer.instance_variable_get(:@transaction_uuid) == 'foo')
+    end
+  end
+
   describe "when sending a report" do
     include PuppetSpec::Files
 
     before do
+      allow(Puppet.settings).to receive(:use).and_return(true)
+      @configurer = Puppet::Configurer.new
       Puppet[:lastrunfile] = tmpfile('last_run_file')
+
+      @report = Puppet::Transaction::Report.new
       Puppet[:reports] = "none"
     end
 
     it "should print a report summary if configured to do so" do
       Puppet.settings[:summarize] = true
 
-      expect(report).to receive(:summary).and_return("stuff")
+      expect(@report).to receive(:summary).and_return("stuff")
 
-      expect(configurer).to receive(:puts).with("stuff")
-      configurer.send_report(report)
+      expect(@configurer).to receive(:puts).with("stuff")
+      @configurer.send_report(@report)
     end
 
     it "should not print a report summary if not configured to do so" do
       Puppet.settings[:summarize] = false
 
-      expect(configurer).not_to receive(:puts)
-      configurer.send_report(report)
+      expect(@configurer).not_to receive(:puts)
+      @configurer.send_report(@report)
     end
 
     it "should save the report if reporting is enabled" do
       Puppet.settings[:report] = true
 
-      expect(Puppet::Transaction::Report.indirection).to receive(:save).with(report, nil, instance_of(Hash))
-      configurer.send_report(report)
+      expect(Puppet::Transaction::Report.indirection).to receive(:save).with(@report, nil, instance_of(Hash))
+      @configurer.send_report(@report)
     end
 
     it "should not save the report if reporting is disabled" do
       Puppet.settings[:report] = false
 
-      expect(Puppet::Transaction::Report.indirection).not_to receive(:save).with(report, nil, instance_of(Hash))
-      configurer.send_report(report)
+      expect(Puppet::Transaction::Report.indirection).not_to receive(:save).with(@report, nil, instance_of(Hash))
+      @configurer.send_report(@report)
     end
 
     it "should save the last run summary if reporting is enabled" do
       Puppet.settings[:report] = true
 
-      expect(configurer).to receive(:save_last_run_summary).with(report)
-      configurer.send_report(report)
+      expect(@configurer).to receive(:save_last_run_summary).with(@report)
+      @configurer.send_report(@report)
     end
 
     it "should save the last run summary if reporting is disabled" do
       Puppet.settings[:report] = false
 
-      expect(configurer).to receive(:save_last_run_summary).with(report)
-      configurer.send_report(report)
+      expect(@configurer).to receive(:save_last_run_summary).with(@report)
+      @configurer.send_report(@report)
     end
 
     it "should log but not fail if saving the report fails" do
@@ -467,9 +548,8 @@ describe Puppet::Configurer do
 
       expect(Puppet::Transaction::Report.indirection).to receive(:save).and_raise("whatever")
 
-      configurer.send_report(report)
-
-      expect(@logs).to include(an_object_having_attributes(level: :err, message: 'Could not send report: whatever'))
+      expect(Puppet).to receive(:send_log).with(:err, 'Could not send report: whatever')
+      @configurer.send_report(@report)
     end
   end
 
@@ -477,17 +557,22 @@ describe Puppet::Configurer do
     include PuppetSpec::Files
 
     before do
+      allow(Puppet.settings).to receive(:use).and_return(true)
+      @configurer = Puppet::Configurer.new
+
+      @report = double('report', :raw_summary => {})
+
       Puppet[:lastrunfile] = tmpfile('last_run_file')
     end
 
     it "should write the last run file" do
-      configurer.save_last_run_summary(report)
+      @configurer.save_last_run_summary(@report)
       expect(Puppet::FileSystem.exist?(Puppet[:lastrunfile])).to be_truthy
     end
 
     it "should write the raw summary as yaml" do
-      expect(report).to receive(:raw_summary).and_return("summary")
-      configurer.save_last_run_summary(report)
+      expect(@report).to receive(:raw_summary).and_return("summary")
+      @configurer.save_last_run_summary(@report)
       expect(File.read(Puppet[:lastrunfile])).to eq(YAML.dump("summary"))
     end
 
@@ -502,14 +587,13 @@ describe Puppet::Configurer do
 
       expect(Puppet::Util).to receive(:replace_file).and_yield(fh)
 
-      configurer.save_last_run_summary(report)
-
-      expect(@logs).to include(an_object_having_attributes(level: :err, message: 'Could not save last run local report: failed to do print'))
+      expect(Puppet).to receive(:send_log).with(:err, 'Could not save last run local report: failed to do print')
+      @configurer.save_last_run_summary(@report)
     end
 
     it "should create the last run file with the correct mode" do
       expect(Puppet.settings.setting(:lastrunfile)).to receive(:mode).and_return('664')
-      configurer.save_last_run_summary(report)
+      @configurer.save_last_run_summary(@report)
 
       if Puppet::Util::Platform.windows?
         require 'puppet/util/windows/security'
@@ -522,28 +606,26 @@ describe Puppet::Configurer do
 
     it "should report invalid last run file permissions" do
       expect(Puppet.settings.setting(:lastrunfile)).to receive(:mode).and_return('892')
-
-      configurer.save_last_run_summary(report)
-
-      expect(@logs).to include(an_object_having_attributes(level: :err, message: /Could not save last run local report.*892 is invalid/))
+      expect(Puppet).to receive(:send_log).with(:err, /Could not save last run local report.*892 is invalid/)
+      @configurer.save_last_run_summary(@report)
     end
   end
 
   describe "when requesting a node" do
     it "uses the transaction uuid in the request" do
       expect(Puppet::Node.indirection).to receive(:find).with(anything, hash_including(transaction_uuid: anything)).twice
-      configurer.run
+      @agent.run
     end
 
     it "sends an explicitly configured environment request" do
       expect(Puppet.settings).to receive(:set_by_config?).with(:environment).and_return(true)
       expect(Puppet::Node.indirection).to receive(:find).with(anything, hash_including(configured_environment: Puppet[:environment])).twice
-      configurer.run
+      @agent.run
     end
 
     it "does not send a configured_environment when using the default" do
       expect(Puppet::Node.indirection).to receive(:find).with(anything, hash_including(configured_environment: nil)).twice
-      configurer.run
+      @agent.run
     end
   end
 
@@ -574,6 +656,14 @@ describe Puppet::Configurer do
 
   describe "when retrieving a catalog" do
     before do
+      allow(Puppet.settings).to receive(:use).and_return(true)
+      allow(@agent).to receive(:facts_for_uploading).and_return({})
+      allow(@agent).to receive(:download_plugins)
+
+      # retrieve a catalog in the current environment, so we don't try to converge unexpectedly
+      @catalog = Puppet::Resource::Catalog.new("tester", Puppet::Node::Environment.remote(Puppet[:environment].to_sym))
+
+      # this is the default when using a Configurer instance
       allow(Puppet::Resource::Catalog.indirection).to receive(:terminus_class).and_return(:rest)
     end
 
@@ -583,139 +673,100 @@ describe Puppet::Configurer do
       end
 
       it "should first look in the cache for a catalog" do
-        expects_cached_catalog_only(catalog)
+        expects_cached_catalog_only(@catalog)
 
-        configurer.run
+        expect(@agent.retrieve_catalog({})).to eq(@catalog)
       end
 
       it "should not make a node request or pluginsync when a cached catalog is successfully retrieved" do
         expect(Puppet::Node.indirection).not_to receive(:find)
-        expects_cached_catalog_only(catalog)
-        expect(configurer).not_to receive(:download_plugins)
+        expects_cached_catalog_only(@catalog)
+        expect(@agent).not_to receive(:download_plugins)
 
-        configurer.run
+        @agent.run
       end
 
       it "should make a node request and pluginsync when a cached catalog cannot be retrieved" do
         expect(Puppet::Node.indirection).to receive(:find).and_return(nil)
-        expects_fallback_to_new_catalog(catalog)
-        expect(configurer).to receive(:download_plugins)
+        expects_fallback_to_new_catalog(@catalog)
+        expect(@agent).to receive(:download_plugins)
 
-        configurer.run
+        @agent.run
       end
 
       it "should set its cached_catalog_status to 'explicitly_requested'" do
-        expects_cached_catalog_only(catalog)
+        expects_cached_catalog_only(@catalog)
 
-        options = {}
-        configurer.run(options)
-
-        expect(options[:report].cached_catalog_status).to eq('explicitly_requested')
+        @agent.retrieve_catalog({})
+        expect(@agent.instance_variable_get(:@cached_catalog_status)).to eq('explicitly_requested')
       end
 
       it "should set its cached_catalog_status to 'explicitly requested' if the cached catalog is from a different environment" do
         cached_catalog = Puppet::Resource::Catalog.new("tester", Puppet::Node::Environment.remote('second_env'))
         expects_cached_catalog_only(cached_catalog)
 
-        options = {}
-        configurer.run(options)
-
-        expect(options[:report].cached_catalog_status).to eq('explicitly_requested')
+        @agent.retrieve_catalog({})
+        expect(@agent.instance_variable_get(:@cached_catalog_status)).to eq('explicitly_requested')
       end
 
-      it "should pluginsync and compile a new catalog if none is found in the cache" do
-        expects_fallback_to_new_catalog(catalog)
-        stub_request(:get, %r{/puppet/v3/file_metadatas?/plugins}).to_return(:status => 404)
-        stub_request(:get, %r{/puppet/v3/file_metadatas?/pluginfacts}).to_return(:status => 404)
+      it "should compile a new catalog if none is found in the cache" do
+        expects_fallback_to_new_catalog(@catalog)
+
+        expect(@agent.retrieve_catalog({})).to eq(@catalog)
+      end
 
-        options = {}
-        configurer.run(options)
+      it "should set its cached_catalog_status to 'not_used' if no catalog is found in the cache" do
+        expects_fallback_to_new_catalog(@catalog)
 
-        expect(options[:report].cached_catalog_status).to eq('not_used')
+        @agent.retrieve_catalog({})
+        expect(@agent.instance_variable_get(:@cached_catalog_status)).to eq('not_used')
       end
 
       it "should not attempt to retrieve a cached catalog again if the first attempt failed" do
         expect(Puppet::Node.indirection).to receive(:find).and_return(nil)
         expects_neither_new_or_cached_catalog
 
-        # after failing to use a cached catalog, we'll need to pluginsync before getting
-        # a new catalog, which also fails.
-        stub_request(:get, %r{/puppet/v3/file_metadatas?/plugins}).to_return(:status => 404)
-        stub_request(:get, %r{/puppet/v3/file_metadatas?/pluginfacts}).to_return(:status => 404)
-
-        configurer.run
+        @agent.run
       end
 
       it "should return the cached catalog when the environment doesn't match" do
         cached_catalog = Puppet::Resource::Catalog.new("tester", Puppet::Node::Environment.remote('second_env'))
         expects_cached_catalog_only(cached_catalog)
 
-        allow(Puppet).to receive(:info)
         expect(Puppet).to receive(:info).with("Using cached catalog from environment 'second_env'")
-
-        configurer.run
-      end
-
-      it "applies the catalog passed as options when the catalog cache terminus is not set" do
-        stub_request(:get, %r{/puppet/v3/file_metadatas?/plugins}).to_return(:status => 404)
-        stub_request(:get, %r{/puppet/v3/file_metadatas?/pluginfacts}).to_return(:status => 404)
-
-        catalog.add_resource(Puppet::Resource.new('notify', 'from apply'))
-        configurer.run(catalog: catalog.to_ral)
-
-        # make sure cache class is not set to avoid surprises later
-        expect(Puppet::Resource::Catalog.indirection).to_not be_cache
-        expect(@logs).to include(an_object_having_attributes(level: :notice, message: /defined 'message' as 'from apply'/))
-      end
-
-      it "applies the cached catalog when the catalog cache terminus is set, ignoring the catalog passed as options" do
-        Puppet::Resource::Catalog.indirection.cache_class = :json
-
-        cached_catalog = Puppet::Resource::Catalog.new(Puppet[:node_name_value], Puppet[:environment])
-        cached_catalog.add_resource(Puppet::Resource.new('notify', 'from cache'))
-
-        # update cached catalog
-        Puppet.settings.use(:main, :agent)
-        path = Puppet::Resource::Catalog.indirection.cache.path(cached_catalog.name)
-        FileUtils.mkdir(File.dirname(path))
-        File.write(path, cached_catalog.render(:json))
-
-        configurer.run(catalog: catalog.to_ral)
-
-        expect(@logs).to include(an_object_having_attributes(level: :notice, message: /defined 'message' as 'from cache'/))
+        expect(@agent.retrieve_catalog({})).to eq(cached_catalog)
       end
     end
 
     describe "and strict environment mode is set" do
       before do
+        allow(@catalog).to receive(:to_ral).and_return(@catalog)
+        allow(@catalog).to receive(:write_class_file)
+        allow(@catalog).to receive(:write_resource_file)
+        allow(@agent).to receive(:send_report)
+        allow(@agent).to receive(:save_last_run_summary)
         Puppet.settings[:strict_environment_mode] = true
       end
 
       it "should not make a node request" do
-        stub_request(:get, %r{/puppet/v3/file_metadatas?/plugins}).to_return(:status => 404)
-        stub_request(:get, %r{/puppet/v3/file_metadatas?/pluginfacts}).to_return(:status => 404)
-        expects_new_catalog_only(catalog)
-        
         expect(Puppet::Node.indirection).not_to receive(:find)
 
-        configurer.run
+        @agent.run
       end
 
       it "should return nil when the catalog's environment doesn't match the agent specified environment" do
-        Puppet[:environment] = 'second_env'
-        configurer = Puppet::Configurer.new
-
-        catalog = Puppet::Resource::Catalog.new("tester", Puppet::Node::Environment.remote("production"))
-        expects_new_catalog_only(catalog)
+        @agent.instance_variable_set(:@environment, 'second_env')
+        expects_new_catalog_only(@catalog)
 
         expect(Puppet).to receive(:err).with("Not using catalog because its environment 'production' does not match agent specified environment 'second_env' and strict_environment_mode is set")
-        expect(configurer.run).to be_nil
+        expect(@agent.run).to be_nil
       end
 
-      it "should return 0 when the catalog's environment matches the agent specified environment" do
-        expects_new_catalog_only(catalog)
+      it "should not return nil when the catalog's environment matches the agent specified environment" do
+        @agent.instance_variable_set(:@environment, 'production')
+        expects_new_catalog_only(@catalog)
 
-        expect(configurer.run).to eq(0)
+        expect(@agent.run).to eq(0)
       end
 
       describe "and a cached catalog is explicitly requested" do
@@ -724,81 +775,83 @@ describe Puppet::Configurer do
         end
 
         it "should return nil when the cached catalog's environment doesn't match the agent specified environment" do
-          Puppet[:environment] = 'second_env'
-          configurer = Puppet::Configurer.new
-
-          catalog = Puppet::Resource::Catalog.new("tester", Puppet::Node::Environment.remote("production"))
-          expects_cached_catalog_only(catalog)
+          @agent.instance_variable_set(:@environment, 'second_env')
+          expects_cached_catalog_only(@catalog)
 
           expect(Puppet).to receive(:err).with("Not using catalog because its environment 'production' does not match agent specified environment 'second_env' and strict_environment_mode is set")
-          expect(configurer.run).to be_nil
+          expect(@agent.run).to be_nil
         end
 
         it "should proceed with the cached catalog if its environment matchs the local environment" do
-          expects_cached_catalog_only(catalog)
+          Puppet.settings[:use_cached_catalog] = true
+          @agent.instance_variable_set(:@environment, 'production')
+          expects_cached_catalog_only(@catalog)
 
-          expect(configurer.run).to eq(0)
+          expect(@agent.run).to eq(0)
         end
       end
     end
 
-    it "should set its cached_catalog_status to 'not_used' when downloading a new catalog" do
-      expect(Puppet::Resource::Catalog.indirection).to receive(:find).with(anything, hash_including(ignore_cache: true)).and_return(catalog)
+    it "should use the Catalog class to get its catalog" do
+      expect(Puppet::Resource::Catalog.indirection).to receive(:find).and_return(@catalog)
 
-      options = {}
-      configurer.run(options)
+      @agent.retrieve_catalog({})
+    end
 
-      expect(options[:report].cached_catalog_status).to eq('not_used')
+    it "should set its cached_catalog_status to 'not_used' when downloading a new catalog" do
+      expect(Puppet::Resource::Catalog.indirection).to receive(:find).with(anything, hash_including(ignore_cache: true)).and_return(@catalog)
+
+      @agent.retrieve_catalog({})
+      expect(@agent.instance_variable_get(:@cached_catalog_status)).to eq('not_used')
     end
 
     it "should use its node_name_value to retrieve the catalog" do
-      myhost_facts = Puppet::Node::Facts.new("myhost.domain.com")
-      Puppet::Node::Facts.indirection.save(myhost_facts)
-
+      allow(Facter).to receive(:value).and_return("eh")
       Puppet.settings[:node_name_value] = "myhost.domain.com"
-      expect(Puppet::Resource::Catalog.indirection).to receive(:find).with("myhost.domain.com", anything).and_return(catalog)
+      expect(Puppet::Resource::Catalog.indirection).to receive(:find).with("myhost.domain.com", anything).and_return(@catalog)
 
-      configurer.run
+      @agent.retrieve_catalog({})
     end
 
-    it "should log when no catalog can be retrieved from the server" do
-      expects_fallback_to_cached_catalog(catalog)
+    it "should default to returning a catalog retrieved directly from the server, skipping the cache" do
+      expect(Puppet::Resource::Catalog.indirection).to receive(:find).with(anything, hash_including(ignore_cache: true)).and_return(@catalog)
+
+      expect(@agent.retrieve_catalog({})).to eq(@catalog)
+    end
+
+    it "should log and return the cached catalog when no catalog can be retrieved from the server" do
+      expects_fallback_to_cached_catalog(@catalog)
 
-      allow(Puppet).to receive(:info)
       expect(Puppet).to receive(:info).with("Using cached catalog from environment 'production'")
-      configurer.run
+      expect(@agent.retrieve_catalog({})).to eq(@catalog)
     end
 
     it "should set its cached_catalog_status to 'on_failure' when no catalog can be retrieved from the server" do
-      expects_fallback_to_cached_catalog(catalog)
+      expects_fallback_to_cached_catalog(@catalog)
 
-      options = {}
-      configurer.run(options)
-
-      expect(options[:report].cached_catalog_status).to eq('on_failure')
+      @agent.retrieve_catalog({})
+      expect(@agent.instance_variable_get(:@cached_catalog_status)).to eq('on_failure')
     end
 
     it "should not look in the cache for a catalog if one is returned from the server" do
-      expects_new_catalog_only(catalog)
+      expects_new_catalog_only(@catalog)
 
-      configurer.run
+      expect(@agent.retrieve_catalog({})).to eq(@catalog)
     end
 
     it "should return the cached catalog when retrieving the remote catalog throws an exception" do
       expect(Puppet::Resource::Catalog.indirection).to receive(:find).with(anything, hash_including(ignore_cache: true)).and_raise("eh")
-      expect(Puppet::Resource::Catalog.indirection).to receive(:find).with(anything, hash_including(ignore_terminus: true)).and_return(catalog)
+      expect(Puppet::Resource::Catalog.indirection).to receive(:find).with(anything, hash_including(ignore_terminus: true)).and_return(@catalog)
 
-      configurer.run
+      expect(@agent.retrieve_catalog({})).to eq(@catalog)
     end
 
     it "should set its cached_catalog_status to 'on_failure' when retrieving the remote catalog throws an exception" do
       expect(Puppet::Resource::Catalog.indirection).to receive(:find).with(anything, hash_including(ignore_cache: true)).and_raise("eh")
-      expect(Puppet::Resource::Catalog.indirection).to receive(:find).with(anything, hash_including(ignore_terminus: true)).and_return(catalog)
-
-      options = {}
-      configurer.run(options)
+      expect(Puppet::Resource::Catalog.indirection).to receive(:find).with(anything, hash_including(ignore_terminus: true)).and_return(@catalog)
 
-      expect(options[:report].cached_catalog_status).to eq('on_failure')
+      @agent.retrieve_catalog({})
+      expect(@agent.instance_variable_get(:@cached_catalog_status)).to eq('on_failure')
     end
 
     it "should log and return nil if no catalog can be retrieved from the server and :usecacheonfailure is disabled" do
@@ -807,173 +860,120 @@ describe Puppet::Configurer do
 
       expect(Puppet).to receive(:warning).with('Not using cache on failed catalog')
 
-      expect(configurer.run).to be_nil
+      expect(@agent.retrieve_catalog({})).to be_nil
     end
 
     it "should set its cached_catalog_status to 'not_used' if no catalog can be retrieved from the server and :usecacheonfailure is disabled or fails to retrieve a catalog" do
       Puppet[:usecacheonfailure] = false
       expect(Puppet::Resource::Catalog.indirection).to receive(:find).with(anything, hash_including(ignore_cache: true)).and_return(nil)
 
-      options = {}
-      configurer.run(options)
-
-      expect(options[:report].cached_catalog_status).to eq('not_used')
+      @agent.retrieve_catalog({})
+      expect(@agent.instance_variable_get(:@cached_catalog_status)).to eq('not_used')
     end
 
     it "should return nil if no cached catalog is available and no catalog can be retrieved from the server" do
       expects_neither_new_or_cached_catalog
 
-      expect(configurer.run).to be_nil
+      expect(@agent.retrieve_catalog({})).to be_nil
     end
 
     it "should return nil if its cached catalog environment doesn't match server-specified environment" do
       cached_catalog = Puppet::Resource::Catalog.new("tester", Puppet::Node::Environment.remote('second_env'))
+      @agent.instance_variable_set(:@node_environment, 'production')
 
       expects_fallback_to_cached_catalog(cached_catalog)
 
-      allow(Puppet).to receive(:err)
       expect(Puppet).to receive(:err).with("Not using cached catalog because its environment 'second_env' does not match 'production'")
-      expect(configurer.run).to be_nil
+      expect(@agent.retrieve_catalog({})).to be_nil
     end
 
     it "should set its cached_catalog_status to 'not_used' if the cached catalog environment doesn't match server-specified environment" do
       cached_catalog = Puppet::Resource::Catalog.new("tester", Puppet::Node::Environment.remote('second_env'))
+      @agent.instance_variable_set(:@node_environment, 'production')
 
       expects_fallback_to_cached_catalog(cached_catalog)
 
-      options = {}
-      configurer.run(options)
-      expect(options[:report].cached_catalog_status).to eq('not_used')
+      @agent.retrieve_catalog({})
+      expect(@agent.instance_variable_get(:@cached_catalog_status)).to eq('not_used')
     end
 
-    it "should set its cached_catalog_status to 'on_failure' if the cached catalog environment matches server-specified environment" do
-      expects_fallback_to_cached_catalog(catalog)
+    it "should return its cached catalog if the environment matches the server-specified environment" do
+      cached_catalog = Puppet::Resource::Catalog.new("tester", Puppet::Node::Environment.remote(Puppet[:environment]))
+      @agent.instance_variable_set(:@node_environment, cached_catalog.environment)
 
-      options = {}
-      configurer.run(options)
-      expect(options[:report].cached_catalog_status).to eq('on_failure')
-    end
-
-    it "should not update the cached catalog in noop mode" do
-      Puppet[:noop] = true
-
-      stub_request(:get, %r{/puppet/v3/catalog}).to_return(:status => 200, :body => catalog.render(:json), :headers => {'Content-Type' => 'application/json'})
-
-      Puppet::Resource::Catalog.indirection.cache_class = :json
-      path = Puppet::Resource::Catalog.indirection.cache.path(catalog.name)
+      expects_fallback_to_cached_catalog(cached_catalog)
 
-      expect(File).to_not be_exist(path)
-      configurer.run
-      expect(File).to_not be_exist(path)
+      expect(@agent.retrieve_catalog({})).to eq(cached_catalog)
     end
 
-    it "should update the cached catalog when not in noop mode" do
-      Puppet[:noop] = false
-      Puppet[:log_level] = 'info'
-
-      stub_request(:get, %r{/puppet/v3/catalog}).to_return(:status => 200, :body => catalog.render(:json), :headers => {'Content-Type' => 'application/json'})
-
-      Puppet::Resource::Catalog.indirection.cache_class = :json
-      cache_path = Puppet::Resource::Catalog.indirection.cache.path(Puppet[:node_name_value])
+    it "should set its cached_catalog_status to 'on_failure' if the cached catalog environment matches server-specified environment" do
+      cached_catalog = Puppet::Resource::Catalog.new("tester", Puppet::Node::Environment.remote(Puppet[:environment]))
+      @agent.instance_variable_set(:@node_environment, cached_catalog.environment)
 
-      expect(File).to_not be_exist(cache_path)
-      configurer.run
-      expect(File).to be_exist(cache_path)
+      expects_fallback_to_cached_catalog(cached_catalog)
 
-      expect(@logs).to include(an_object_having_attributes(level: :info, message: "Caching catalog for #{Puppet[:node_name_value]}"))
+      @agent.retrieve_catalog({})
+      expect(@agent.instance_variable_get(:@cached_catalog_status)).to eq('on_failure')
     end
 
-    it "successfully applies the catalog without a cache" do
-      stub_request(:get, %r{/puppet/v3/catalog}).to_return(:status => 200, :body => catalog.render(:json), :headers => {'Content-Type' => 'application/json'})
-
-      Puppet::Resource::Catalog.indirection.cache_class = nil
+    it "should not update the cached catalog in noop mode" do
+      Puppet[:noop] = true
+      expect(Puppet::Resource::Catalog.indirection).to receive(:find).with(anything, hash_including(ignore_cache: true, ignore_cache_save: true)).and_return(@catalog)
 
-      expect(configurer.run).to eq(0)
+      @agent.retrieve_catalog({})
     end
 
-    it "should not update the cached catalog when running puppet apply" do
-      Puppet::Resource::Catalog.indirection.cache_class = :json
-      path = Puppet::Resource::Catalog.indirection.cache.path(catalog.name)
+    it "should update the cached catalog when not in noop mode" do
+      Puppet[:noop] = false
+      expect(Puppet::Resource::Catalog.indirection).to receive(:find).with(anything, hash_including(ignore_cache: true, ignore_cache_save: false)).and_return(@catalog)
 
-      expect(File).to_not be_exist(path)
-      configurer.run(catalog: catalog)
-      expect(File).to_not be_exist(path)
+      @agent.retrieve_catalog({})
     end
   end
 
-  describe "when converging the environment" do
-    let(:apple) { Puppet::Resource::Catalog.new(Puppet[:node_name_value], Puppet::Node::Environment.remote('apple')) }
-    let(:banana) { Puppet::Resource::Catalog.new(Puppet[:node_name_value], Puppet::Node::Environment.remote('banana')) }
+  describe "when converting the catalog" do
+    before do
+      allow(Puppet.settings).to receive(:use).and_return(true)
 
-    before :each do
-      apple.add_resource(resource)
-      banana.add_resource(resource)
+      allow(catalog).to receive(:to_ral).and_return(ral_catalog)
     end
 
-    it "converges after multiple attempts" do
-      expect(Puppet::Resource::Catalog.indirection).to receive(:find).and_return(apple, banana, banana)
-
-      allow(Puppet).to receive(:notice)
-      expect(Puppet).to receive(:notice).with("Local environment: 'production' doesn't match server specified environment 'apple', restarting agent run with environment 'apple'")
-      expect(Puppet).to receive(:notice).with("Local environment: 'apple' doesn't match server specified environment 'banana', restarting agent run with environment 'banana'")
+    let (:catalog) { Puppet::Resource::Catalog.new('tester', Puppet::Node::Environment.remote(Puppet[:environment].to_sym)) }
+    let (:ral_catalog) { Puppet::Resource::Catalog.new('tester', Puppet::Node::Environment.remote(Puppet[:environment].to_sym)) }
 
-      configurer.run
+    it "should convert the catalog to a RAL-formed catalog" do
+      expect(@agent.convert_catalog(catalog, 10)).to equal(ral_catalog)
     end
 
-    it "raises if it can't converge after 4 tries after the initial catalog request" do
-      expect(Puppet::Resource::Catalog.indirection).to receive(:find).and_return(apple, banana, apple, banana, apple)
+    it "should finalize the catalog" do
+      expect(ral_catalog).to receive(:finalize)
 
-      configurer.run
-
-      expect(@logs).to include(an_object_having_attributes(level: :err, message: "Failed to apply catalog: Catalog environment didn't stabilize after 4 fetches, aborting run"))
+      @agent.convert_catalog(catalog, 10)
     end
-  end
 
-  describe "when converting the catalog" do
-    it "converts Puppet::Resource into Puppet::Type::Notify" do
-      expect(configurer).to receive(:apply_catalog) do |ral, _|
-        expect(ral.resources).to contain(an_instance_of(Puppet::Type::Notify))
-      end
+    it "should record the passed retrieval time with the RAL catalog" do
+      expect(ral_catalog).to receive(:retrieval_duration=).with(10)
 
-      configurer.run(catalog: catalog)
+      @agent.convert_catalog(catalog, 10)
     end
 
-    it "adds default schedules" do
-      expect(configurer).to receive(:apply_catalog) do |ral, _|
-        expect(ral.resources.map(&:to_ref)).to contain(%w{Schedule[puppet] Schedule[hourly] Schedule[daily] Schedule[weekly] Schedule[monthly] Schedule[never]})
-      end
-
-      configurer.run
-    end
-
-    it "records the retrieval duration to the catalog" do
-      expect(configurer).to receive(:apply_catalog) do |ral, _|
-        expect(ral.retrieval_duration).to be_an_instance_of(Float)
-      end
+    it "should write the RAL catalog's class file" do
+      expect(ral_catalog).to receive(:write_class_file)
 
-      configurer.run
+      @agent.convert_catalog(catalog, 10)
     end
 
-    it "writes the class file containing applied settings classes" do
-      expect(File).to_not be_exist(Puppet[:classfile])
-
-      configurer.run
+    it "should write the RAL catalog's resource file" do
+      expect(ral_catalog).to receive(:write_resource_file)
 
-      expect(File.read(Puppet[:classfile]).chomp).to eq('settings')
+      @agent.convert_catalog(catalog, 10)
     end
 
-    it "writes an empty resource file since no resources are 'managed'" do
-      expect(File).to_not be_exist(Puppet[:resourcefile])
+    it "should set catalog conversion time on the report" do
+      report = Puppet::Transaction::Report.new
 
-      configurer.run
-
-      expect(File.read(Puppet[:resourcefile]).chomp).to eq("")
-    end
-
-    it "adds the conversion time to the report" do
-      configurer.run(report: report)
-
-      expect(report.metrics['time']['convert_catalog']).to be_an_instance_of(Float)
+      expect(report).to receive(:add_times).with(:convert_catalog, kind_of(Numeric))
+      @agent.convert_catalog(catalog, 10, {:report => report})
     end
   end
 
@@ -994,62 +994,66 @@ describe Puppet::Configurer do
   describe "when attempting failover" do
     it "should not failover if server_list is not set" do
       Puppet.settings[:server_list] = []
-      configurer.run
+      expect(@agent).not_to receive(:find_functional_server)
+      @agent.run
     end
 
     it "should not failover during an apply run" do
       Puppet.settings[:server_list] = ["myserver:123"]
+      expect(@agent).not_to receive(:find_functional_server)
       catalog = Puppet::Resource::Catalog.new("tester", Puppet::Node::Environment.remote(Puppet[:environment].to_sym))
-      configurer.run(catalog: catalog)
+      @agent.run :catalog => catalog
     end
 
-    it "should select a server when it receives 200 OK response" do
+    it "should select a server when provided" do
       Puppet.settings[:server_list] = ["myserver:123"]
-
-      stub_request(:get, 'https://myserver:123/status/v1/simple/master').to_return(status: 200)
+      response = Net::HTTPOK.new(nil, 200, 'OK')
+      allow(Puppet::Network::HttpPool).to receive(:connection).with('myserver', 123, anything).and_return(double('request', get: response))
+      allow(@agent).to receive(:run_internal)
 
       options = {}
-      configurer.run(options)
+      @agent.run(options)
       expect(options[:report].master_used).to eq('myserver:123')
     end
 
-    it "should report when a server is unavailable" do
+    it "queries the simple status for the 'master' service" do
       Puppet.settings[:server_list] = ["myserver:123"]
+      response = Net::HTTPOK.new(nil, 200, 'OK')
+      http = double('request')
+      expect(http).to receive(:get).with('/status/v1/simple/master').and_return(response)
+      allow(Puppet::Network::HttpPool).to receive(:connection).with('myserver', 123, anything).and_return(http)
+      allow(@agent).to receive(:run_internal)
 
-      stub_request(:get, 'https://myserver:123/status/v1/simple/master').to_return(status: [500, "Internal Server Error"])
+      @agent.run
+    end
 
-      allow(Puppet).to receive(:debug)
-      expect(Puppet).to receive(:debug).with("Puppet server myserver:123 is unavailable: 500 Internal Server Error")
+    it "should report when a server is unavailable" do
+      Puppet.settings[:server_list] = ["myserver:123"]
+      response = Net::HTTPInternalServerError.new(nil, 500, 'Internal Server Error')
+      allow(Puppet::Network::HttpPool).to receive(:connection).with('myserver', 123, anything).and_return(double('request', get: response))
 
-      expect {
-        configurer.run
-      }.to raise_error(Puppet::Error, /Could not select a functional puppet master from server_list:/)
+      expect(Puppet).to receive(:debug).with("Puppet server myserver:123 is unavailable: 500 Internal Server Error")
+      expect{ @agent.run }.to raise_error(Puppet::Error, /Could not select a functional puppet master from server_list/)
     end
 
     it "should error when no servers in 'server_list' are reachable" do
-      Puppet.settings[:server_list] = "myserver:123,someotherservername"
-
-      stub_request(:get, 'https://myserver:123/status/v1/simple/master').to_return(status: 400)
-      stub_request(:get, 'https://someotherservername:8140/status/v1/simple/master').to_return(status: 400)
+      Puppet.settings[:server_list] = ["myserver:123"]
+      error = Net::HTTPError.new(400, 'dummy server communication error')
+      allow(Puppet::Network::HttpPool).to receive(:connection).with('myserver', 123, anything).and_return(double('request', get: error))
 
-      expect{
-        configurer.run
-      }.to raise_error(Puppet::Error, /Could not select a functional puppet master from server_list: 'myserver:123,someotherservername'/)
+      options = {}
+      expect{ @agent.run(options) }.to raise_error(Puppet::Error, /Could not select a functional puppet master from server_list/)
+      expect(options[:report].master_used).to be_nil
     end
 
     it "should not make multiple node requests when the server is found" do
       Puppet.settings[:server_list] = ["myserver:123"]
+      response = Net::HTTPOK.new(nil, 200, 'OK')
 
-      Puppet::Node.indirection.terminus_class = :rest
-      Puppet::Resource::Catalog.indirection.terminus_class = :rest
-
-      stub_request(:get, 'https://myserver:123/status/v1/simple/master').to_return(status: 200)
-      stub_request(:get, %r{https://myserver:123/puppet/v3/catalog}).to_return(status: 200)
-      node_request = stub_request(:get, %r{https://myserver:123/puppet/v3/node/}).to_return(status: 200)
-
-      configurer.run
+      expect(Puppet::Network::HttpPool).to receive(:connection).with('myserver', 123, anything).and_return(double('request', get: response))
+      allow(@agent).to receive(:run_internal)
 
-      expect(node_request).to have_been_requested.once
+      @agent.run
     end
   end
 end