puppet 7.28.0-universal-darwin → 7.29.0-universal-darwin

data/spec/unit/ssl/state_machine_spec.rb

@@ -1,982 +0,0 @@
-require 'spec_helper'
-require 'puppet_spec/files'
-
-require 'puppet/ssl'
-
-describe Puppet::SSL::StateMachine, unless: Puppet::Util::Platform.jruby? do
-  include PuppetSpec::Files
-
-  let(:privatekeydir) { tmpdir('privatekeydir') }
-  let(:certdir) { tmpdir('certdir') }
-  let(:requestdir) { tmpdir('requestdir') }
-  let(:machine) { described_class.new }
-  let(:cert_provider) { Puppet::X509::CertProvider.new(privatekeydir: privatekeydir, certdir: certdir, requestdir: requestdir) }
-  let(:ssl_provider) { Puppet::SSL::SSLProvider.new }
-  let(:machine) { described_class.new(cert_provider: cert_provider, ssl_provider: ssl_provider) }
-
-  let(:cacert_pem) { cacert.to_pem }
-  let(:cacert) { cert_fixture('ca.pem') }
-  let(:cacerts) { [cacert, cert_fixture('intermediate.pem')] }
-
-  let(:crl_pem) { crl.to_pem }
-  let(:crl) { crl_fixture('crl.pem') }
-  let(:crls) { [crl, crl_fixture('intermediate-crl.pem')] }
-  let(:private_key) { key_fixture('signed-key.pem') }
-  let(:client_cert) { cert_fixture('signed.pem') }
-
-  let(:refused_message) { %r{Connection refused|No connection could be made because the target machine actively refused it} }
-
-  before(:each) do
-    Puppet[:daemonize] = false
-    Puppet[:ssl_lockfile] = tmpfile('ssllock')
-    allow(Kernel).to receive(:sleep)
-  end
-
-  def expected_digest(name, content)
-    OpenSSL::Digest.new(name).hexdigest(content)
-  end
-
-  def to_fingerprint(digest)
-    digest.scan(/../).join(':').upcase
-  end
-
-  context 'when passing keyword arguments' do
-    it "accepts digest" do
-      expect(described_class.new(digest: 'SHA512').digest).to eq('SHA512')
-    end
-
-    it "accepts ca_fingerprint" do
-      expect(described_class.new(ca_fingerprint: 'CAFE').ca_fingerprint).to eq('CAFE')
-    end
-  end
-
-  context 'when ensuring CA certs and CRLs' do
-    it 'returns an SSLContext with the loaded CA certs and CRLs' do
-      allow(cert_provider).to receive(:load_cacerts).and_return(cacerts)
-      allow(cert_provider).to receive(:load_crls).and_return(crls)
-
-      ssl_context = machine.ensure_ca_certificates
-
-      expect(ssl_context[:cacerts]).to eq(cacerts)
-      expect(ssl_context[:crls]).to eq(crls)
-      expect(ssl_context[:verify_peer]).to eq(true)
-    end
-
-    context 'when exceptions occur' do
-      it 'raises in onetime mode' do
-        stub_request(:get, %r{puppet-ca/v1/certificate/ca})
-          .to_raise(Errno::ECONNREFUSED)
-
-        machine = described_class.new(cert_provider: cert_provider, ssl_provider: ssl_provider, onetime: true)
-        expect {
-          machine.ensure_ca_certificates
-        }.to raise_error(Puppet::Error, refused_message)
-      end
-
-      it 'retries CA cert download' do
-        # allow cert to be saved to disk
-        FileUtils.mkdir_p(Puppet[:certdir])
-        allow(cert_provider).to receive(:load_crls).and_return(crls)
-
-        req = stub_request(:get, %r{puppet-ca/v1/certificate/ca})
-                .to_raise(Errno::ECONNREFUSED).then
-                .to_return(status: 200, body: cacert_pem)
-
-        machine.ensure_ca_certificates
-
-        expect(req).to have_been_made.twice
-        expect(@logs).to include(an_object_having_attributes(message: refused_message))
-      end
-
-      it 'retries CRL download' do
-        # allow crl to be saved to disk
-        FileUtils.mkdir_p(Puppet[:ssldir])
-        allow(cert_provider).to receive(:load_cacerts).and_return(cacerts)
-
-        req = stub_request(:get, %r{puppet-ca/v1/certificate_revocation_list/ca})
-                .to_raise(Errno::ECONNREFUSED).then
-                .to_return(status: 200, body: crl_pem)
-
-        machine.ensure_ca_certificates
-
-        expect(req).to have_been_made.twice
-        expect(@logs).to include(an_object_having_attributes(message: refused_message))
-      end
-    end
-  end
-
-  context 'when ensuring a client cert' do
-    it 'returns an SSLContext with the loaded CA certs, CRLs, private key and client cert' do
-      allow(cert_provider).to receive(:load_cacerts).and_return(cacerts)
-      allow(cert_provider).to receive(:load_crls).and_return(crls)
-      allow(cert_provider).to receive(:load_private_key).and_return(private_key)
-      allow(cert_provider).to receive(:load_client_cert).and_return(client_cert)
-
-      ssl_context = machine.ensure_client_certificate
-
-      expect(ssl_context[:cacerts]).to eq(cacerts)
-      expect(ssl_context[:crls]).to eq(crls)
-      expect(ssl_context[:verify_peer]).to eq(true)
-      expect(ssl_context[:private_key]).to eq(private_key)
-      expect(ssl_context[:client_cert]).to eq(client_cert)
-    end
-
-    it 'uses the specified digest to log the cert chain fingerprints' do
-      allow(cert_provider).to receive(:load_cacerts).and_return(cacerts)
-      allow(cert_provider).to receive(:load_crls).and_return(crls)
-      allow(cert_provider).to receive(:load_private_key).and_return(private_key)
-      allow(cert_provider).to receive(:load_client_cert).and_return(client_cert)
-
-      Puppet[:log_level] = :debug
-      machine = described_class.new(cert_provider: cert_provider, digest: 'SHA512')
-      machine.ensure_client_certificate
-
-      expect(@logs).to include(
-        an_object_having_attributes(message: /Verified CA certificate 'CN=Test CA' fingerprint \(SHA512\)/),
-        an_object_having_attributes(message: /Verified CA certificate 'CN=Test CA Subauthority' fingerprint \(SHA512\)/),
-        an_object_having_attributes(message: /Verified client certificate 'CN=signed' fingerprint \(SHA512\)/)
-      )
-    end
-
-    context 'when exceptions occur' do
-      before :each do
-        allow(cert_provider).to receive(:load_cacerts).and_return(cacerts)
-        allow(cert_provider).to receive(:load_crls).and_return(crls)
-      end
-
-      it 'retries CSR submission' do
-        allow(cert_provider).to receive(:load_private_key).and_return(private_key)
-        allow($stdout).to receive(:puts).with(/Couldn't fetch certificate from CA server; you might still need to sign this agent's certificate/)
-
-        stub_request(:get, %r{puppet-ca/v1/certificate/#{Puppet[:certname]}})
-          .to_return(status: 200, body: client_cert.to_pem)
-        # first request raises, second succeeds
-        req = stub_request(:put, %r{puppet-ca/v1/certificate_request/#{Puppet[:certname]}})
-                .to_raise(Errno::ECONNREFUSED).then
-                .to_return(status: 200)
-
-        machine.ensure_client_certificate
-
-        expect(req).to have_been_made.twice
-        expect(@logs).to include(an_object_having_attributes(message: refused_message))
-      end
-
-      it 'retries client cert download' do
-        allow(cert_provider).to receive(:load_private_key).and_return(private_key)
-
-        # first request raises, second succeeds
-        req = stub_request(:get, %r{puppet-ca/v1/certificate/#{Puppet[:certname]}})
-                .to_raise(Errno::ECONNREFUSED).then
-                .to_return(status: 200, body: client_cert.to_pem)
-        stub_request(:put, %r{puppet-ca/v1/certificate_request/#{Puppet[:certname]}}).to_return(status: 200)
-
-        machine.ensure_client_certificate
-
-        expect(req).to have_been_made.twice
-        expect(@logs).to include(an_object_having_attributes(message: refused_message))
-      end
-
-      it 'retries when client cert and private key are mismatched' do
-        allow(cert_provider).to receive(:load_private_key).and_return(private_key)
-
-        # return mismatched cert the first time, correct cert second time
-        req = stub_request(:get, %r{puppet-ca/v1/certificate/#{Puppet[:certname]}})
-                .to_return(status: 200, body: cert_fixture('pluto.pem').to_pem)
-                .to_return(status: 200, body: client_cert.to_pem)
-        stub_request(:put, %r{puppet-ca/v1/certificate_request/#{Puppet[:certname]}}).to_return(status: 200)
-
-        machine.ensure_client_certificate
-
-        expect(req).to have_been_made.twice
-        expect(@logs).to include(an_object_having_attributes(message: %r{The certificate for 'CN=pluto' does not match its private key}))
-      end
-
-      it 'raises in onetime mode' do
-        stub_request(:get, %r{puppet-ca/v1/certificate/#{Puppet[:certname]}})
-          .to_raise(Errno::ECONNREFUSED)
-        stub_request(:put, %r{puppet-ca/v1/certificate_request/#{Puppet[:certname]}})
-          .to_return(status: 200)
-
-        machine = described_class.new(cert_provider: cert_provider, ssl_provider: ssl_provider, onetime: true)
-        expect {
-          machine.ensure_client_certificate
-        }.to raise_error(Puppet::Error, refused_message)
-      end
-    end
-  end
-
-  context 'when locking' do
-    let(:lockfile) { Puppet::Util::Pidlock.new(Puppet[:ssl_lockfile]) }
-    let(:machine) { described_class.new(cert_provider: cert_provider, ssl_provider: ssl_provider, lockfile: lockfile) }
-
-    # lockfile is deleted before `ensure_ca_certificates` returns, so
-    # verify lockfile contents while state machine is running
-    def expect_lockfile_to_contain(pid)
-      allow(cert_provider).to receive(:load_cacerts) do
-        expect(File.read(Puppet[:ssl_lockfile])).to eq(pid.to_s)
-      end.and_return(cacerts)
-      allow(cert_provider).to receive(:load_crls).and_return(crls)
-    end
-
-    it 'locks the file prior to running the state machine and unlocks when done' do
-      expect(lockfile).to receive(:lock).and_call_original.ordered
-      expect(cert_provider).to receive(:load_cacerts).and_return(cacerts).ordered
-      expect(cert_provider).to receive(:load_crls).and_return(crls).ordered
-      expect(lockfile).to receive(:unlock).ordered
-
-      machine.ensure_ca_certificates
-    end
-
-    it 'deletes the lockfile when finished' do
-      allow(cert_provider).to receive(:load_cacerts).and_return(cacerts)
-      allow(cert_provider).to receive(:load_crls).and_return(crls)
-
-      machine = described_class.new(cert_provider: cert_provider, ssl_provider: ssl_provider)
-      machine.ensure_ca_certificates
-
-      expect(File).to_not be_exist(Puppet[:ssl_lockfile])
-    end
-
-    it 'acquires an empty lockfile' do
-      Puppet::FileSystem.touch(Puppet[:ssl_lockfile])
-
-      expect_lockfile_to_contain(Process.pid)
-
-      machine = described_class.new(cert_provider: cert_provider, ssl_provider: ssl_provider)
-      machine.ensure_ca_certificates
-    end
-
-    it 'acquires its own lockfile' do
-      File.write(Puppet[:ssl_lockfile], Process.pid.to_s)
-
-      expect_lockfile_to_contain(Process.pid)
-
-      machine = described_class.new(cert_provider: cert_provider, ssl_provider: ssl_provider)
-      machine.ensure_ca_certificates
-    end
-
-    it 'overwrites a stale lockfile' do
-      # 2**31 - 1 chosen to not conflict with existing pid
-      File.write(Puppet[:ssl_lockfile], "2147483647")
-
-      expect_lockfile_to_contain(Process.pid)
-
-      machine = described_class.new(cert_provider: cert_provider, ssl_provider: ssl_provider)
-      machine.ensure_ca_certificates
-    end
-
-    context 'and another puppet process is running' do
-      let(:now) { Time.now }
-      let(:future) { now + (5 * 60)} # 5 mins in the future
-
-      before :each do
-        allow(lockfile).to receive(:lock).and_return(false)
-      end
-
-      it 'raises a puppet exception' do
-        expect {
-          machine.ensure_ca_certificates
-        }.to raise_error(Puppet::Error, /Another puppet instance is already running and the waitforlock setting is set to 0; exiting/)
-      end
-
-      it 'sleeps and retries successfully' do
-        machine = described_class.new(lockfile: lockfile, cert_provider: cert_provider, waitforlock: 1, maxwaitforlock: 10)
-        allow(cert_provider).to receive(:load_cacerts).and_return(cacerts)
-        allow(cert_provider).to receive(:load_crls).and_return(crls)
-        allow(Time).to receive(:now).and_return(now, future)
-
-        expect(Kernel).to receive(:sleep).with(1)
-        expect(Puppet).to receive(:info).with("Another puppet instance is already running; waiting for it to finish")
-        expect(Puppet).to receive(:info).with("Will try again in 1 seconds.")
-
-        allow(lockfile).to receive(:lock).and_return(false, true)
-
-        expect(machine.ensure_ca_certificates).to be_an_instance_of(Puppet::SSL::SSLContext)
-      end
-
-      it 'sleeps and retries unsuccessfully until the deadline is exceeded' do
-        machine = described_class.new(lockfile: lockfile, waitforlock: 1, maxwaitforlock: 10)
-        allow(Time).to receive(:now).and_return(now, future)
-
-        expect(Kernel).to receive(:sleep).with(1)
-        expect(Puppet).to receive(:info).with("Another puppet instance is already running; waiting for it to finish")
-        expect(Puppet).to receive(:info).with("Will try again in 1 seconds.")
-
-        allow(lockfile).to receive(:lock).and_return(false)
-        expect {
-          machine.ensure_ca_certificates
-        }.to raise_error(Puppet::Error, /Another puppet instance is already running and the maxwaitforlock timeout has been exceeded; exiting/)
-      end
-
-      it 'defaults the waitlock deadline to 60 seconds' do
-        allow(Time).to receive(:now).and_return(now)
-
-        machine = described_class.new
-        expect(machine.waitlock_deadline).to eq(now.to_i + 60)
-      end
-    end
-  end
-
-  context 'NeedCACerts' do
-    let(:state) { Puppet::SSL::StateMachine::NeedCACerts.new(machine) }
-
-    before :each do
-      Puppet[:localcacert] = tmpfile('needcacerts')
-    end
-
-    it 'transitions to NeedCRLs state' do
-      allow(cert_provider).to receive(:load_cacerts).and_return(cacerts)
-
-      expect(state.next_state).to be_an_instance_of(Puppet::SSL::StateMachine::NeedCRLs)
-    end
-
-    it 'loads existing CA certs' do
-      allow(cert_provider).to receive(:load_cacerts).and_return(cacerts)
-
-      st = state.next_state
-      expect(st.ssl_context[:cacerts]).to eq(cacerts)
-    end
-
-    it 'fetches and saves CA certs' do
-      allow(cert_provider).to receive(:load_cacerts).and_return(nil)
-      stub_request(:get, %r{puppet-ca/v1/certificate/ca}).to_return(status: 200, body: cacert_pem)
-
-      st = state.next_state
-      expect(st.ssl_context[:cacerts].map(&:to_pem)).to eq([cacert_pem])
-      expect(File).to be_exist(Puppet[:localcacert])
-    end
-
-    it "does not verify the server's cert if there are no local CA certs" do
-      allow(cert_provider).to receive(:load_cacerts).and_return(nil)
-      stub_request(:get, %r{puppet-ca/v1/certificate/ca}).to_return(status: 200, body: cacert_pem)
-      allow(cert_provider).to receive(:save_cacerts)
-
-      receive_count = 0
-      allow_any_instance_of(Net::HTTP).to receive(:verify_mode=).with(OpenSSL::SSL::VERIFY_NONE) { receive_count += 1 }
-
-      state.next_state
-
-      expect(receive_count).to eq(2)
-    end
-
-    it 'returns an Error if the server returns 404' do
-      stub_request(:get, %r{puppet-ca/v1/certificate/ca}).to_return(status: 404)
-
-      st = state.next_state
-      expect(st).to be_an_instance_of(Puppet::SSL::StateMachine::Error)
-      expect(st.message).to eq("CA certificate is missing from the server")
-    end
-
-    it 'returns an Error if there is a different exception' do
-      stub_request(:get, %r{puppet-ca/v1/certificate/ca}).to_return(status: [500, 'Internal Server Error'])
-
-      st = state.next_state
-      expect(st).to be_an_instance_of(Puppet::SSL::StateMachine::Error)
-      expect(st.message).to eq("Could not download CA certificate: Internal Server Error")
-    end
-
-    it 'returns an Error if CA certs are invalid' do
-      allow(cert_provider).to receive(:load_cacerts).and_return(nil)
-      stub_request(:get, %r{puppet-ca/v1/certificate/ca}).to_return(status: 200, body: '')
-
-      st = state.next_state
-      expect(st).to be_an_instance_of(Puppet::SSL::StateMachine::Error)
-      expect(st.error).to be_an_instance_of(OpenSSL::X509::CertificateError)
-    end
-
-    it 'does not save invalid CA certs' do
-      stub_request(:get, %r{puppet-ca/v1/certificate/ca}).to_return(status: 200, body: <<~END)
-        -----BEGIN CERTIFICATE-----
-        MIIBpDCCAQ2gAwIBAgIBAjANBgkqhkiG9w0BAQsFADAfMR0wGwYDVQQDDBRUZXN0
-      END
-
-      state.next_state rescue OpenSSL::X509::CertificateError
-
-      expect(File).to_not exist(Puppet[:localcacert])
-    end
-
-    context 'when verifying CA cert bundle' do
-      before :each do
-        allow(cert_provider).to receive(:load_cacerts).and_return(nil)
-        stub_request(:get, %r{puppet-ca/v1/certificate/ca}).to_return(status: 200, body: cacert_pem)
-        allow(cert_provider).to receive(:save_cacerts)
-      end
-
-      it 'verifies CA cert bundle if a ca_fingerprint is given case-insensitively' do
-        Puppet[:log_level] = :info
-
-        digest = expected_digest('SHA256', cacert_pem)
-        fingerprint = to_fingerprint(digest)
-        machine = described_class.new(digest: 'SHA256', ca_fingerprint: digest.downcase)
-        state = Puppet::SSL::StateMachine::NeedCACerts.new(machine)
-        state.next_state
-
-        expect(@logs).to include(an_object_having_attributes(message: "Verified CA bundle with digest (SHA256) #{fingerprint}"))
-      end
-
-      it 'verifies CA cert bundle using non-default fingerprint' do
-        Puppet[:log_level] = :info
-
-        digest = expected_digest('SHA512', cacert_pem)
-        machine = described_class.new(digest: 'SHA512', ca_fingerprint: digest)
-        state = Puppet::SSL::StateMachine::NeedCACerts.new(machine)
-        state.next_state
-
-        expect(@logs).to include(an_object_having_attributes(message: "Verified CA bundle with digest (SHA512) #{to_fingerprint(digest)}"))
-      end
-
-      it 'returns an error if verification fails' do
-        machine = described_class.new(digest: 'SHA256', ca_fingerprint: 'wrong!')
-        state = Puppet::SSL::StateMachine::NeedCACerts.new(machine)
-
-        fingerprint = to_fingerprint(expected_digest('SHA256', cacert_pem))
-        st = state.next_state
-        expect(st).to be_an_instance_of(Puppet::SSL::StateMachine::Error)
-        expect(st.message).to eq("CA bundle with digest (SHA256) #{fingerprint} did not match expected digest WR:ON:G!")
-      end
-    end
-  end
-
-  context 'NeedCRLs' do
-    let(:ssl_context) { Puppet::SSL::SSLContext.new(cacerts: cacerts)}
-    let(:state) { Puppet::SSL::StateMachine::NeedCRLs.new(machine, ssl_context) }
-
-    before :each do
-      Puppet[:hostcrl] = tmpfile('needcrls')
-    end
-
-    it 'transitions to NeedKey state' do
-      allow(cert_provider).to receive(:load_crls).and_return(crls)
-
-      expect(state.next_state).to be_an_instance_of(Puppet::SSL::StateMachine::NeedKey)
-    end
-
-    it 'loads existing CRLs' do
-      allow(cert_provider).to receive(:load_crls).and_return(crls)
-
-      st = state.next_state
-      expect(st.ssl_context[:crls]).to eq(crls)
-    end
-
-    it 'fetches and saves CRLs' do
-      allow(cert_provider).to receive(:load_crls).and_return(nil)
-      stub_request(:get, %r{puppet-ca/v1/certificate_revocation_list/ca}).to_return(status: 200, body: crl_pem)
-
-      st = state.next_state
-      expect(st.ssl_context[:crls].map(&:to_pem)).to eq([crl_pem])
-      expect(File).to be_exist(Puppet[:hostcrl])
-    end
-
-    it "verifies the server's certificate when fetching the CRL" do
-      allow(cert_provider).to receive(:load_crls).and_return(nil)
-      stub_request(:get, %r{puppet-ca/v1/certificate_revocation_list/ca}).to_return(status: 200, body: crl_pem)
-      allow(cert_provider).to receive(:save_crls)
-
-      receive_count = 0
-      allow_any_instance_of(Net::HTTP).to receive(:verify_mode=).with(OpenSSL::SSL::VERIFY_PEER) { receive_count += 1 }
-
-      state.next_state
-
-      expect(receive_count).to eq(2)
-    end
-
-    it 'returns an Error if the server returns 404' do
-      stub_request(:get, %r{puppet-ca/v1/certificate_revocation_list/ca}).to_return(status: 404)
-
-      st = state.next_state
-      expect(st).to be_an_instance_of(Puppet::SSL::StateMachine::Error)
-      expect(st.message).to eq("CRL is missing from the server")
-    end
-
-    it 'returns an Error if there is a different exception' do
-      stub_request(:get, %r{puppet-ca/v1/certificate_revocation_list/ca}).to_return(status: [500, 'Internal Server Error'])
-
-      st = state.next_state
-      expect(st).to be_an_instance_of(Puppet::SSL::StateMachine::Error)
-      expect(st.message).to eq("Could not download CRLs: Internal Server Error")
-    end
-
-    it 'returns an Error if CRLs are invalid' do
-      allow(cert_provider).to receive(:load_crls).and_return(nil)
-      stub_request(:get, %r{puppet-ca/v1/certificate_revocation_list/ca}).to_return(status: 200, body: '')
-
-      st = state.next_state
-      expect(st).to be_an_instance_of(Puppet::SSL::StateMachine::Error)
-      expect(st.error).to be_an_instance_of(OpenSSL::X509::CRLError)
-    end
-
-    it 'does not save invalid CRLs' do
-      stub_request(:get, %r{puppet-ca/v1/certificate_revocation_list/ca}).to_return(status: 200, body: <<~END)
-        -----BEGIN X509 CRL-----
-        MIIBCjB1AgEBMA0GCSqGSIb3DQEBCwUAMBIxEDAOBgNVBAMMB1Rlc3QgQ0EXDTcw
-      END
-
-      state.next_state rescue OpenSSL::X509::CRLError
-
-      expect(File).to_not exist(Puppet[:hostcrl])
-    end
-
-    it 'skips CRL download when revocation is disabled' do
-      Puppet[:certificate_revocation] = false
-
-      expect(cert_provider).not_to receive(:load_crls)
-
-      state.next_state
-
-      expect(File).to_not exist(Puppet[:hostcrl])
-    end
-
-    it 'skips CRL refresh by default' do
-      allow_any_instance_of(Puppet::X509::CertProvider).to receive(:load_crls).and_return(crls)
-
-      state.next_state
-    end
-
-    it 'skips CRL refresh if it has not expired' do
-      Puppet[:crl_refresh_interval] = '1y'
-      Puppet::FileSystem.touch(Puppet[:hostcrl], mtime: Time.now)
-
-      allow_any_instance_of(Puppet::X509::CertProvider).to receive(:load_crls).and_return(crls)
-
-      state.next_state
-    end
-
-    context 'when refreshing a CRL' do
-      before :each do
-        Puppet[:crl_refresh_interval] = '1s'
-        allow_any_instance_of(Puppet::X509::CertProvider).to receive(:load_crls).and_return(crls)
-
-        yesterday = Time.now - (24 * 60 * 60)
-        allow_any_instance_of(Puppet::X509::CertProvider).to receive(:crl_last_update).and_return(yesterday)
-      end
-
-      let(:new_crl_bundle) do
-        # add intermediate crl to the bundle
-        int_crl = crl_fixture('intermediate-crl.pem')
-        [crl, int_crl].map(&:to_pem)
-      end
-
-      it 'uses the local crl if it has not been modified' do
-        stub_request(:get, %r{puppet-ca/v1/certificate_revocation_list/ca}).to_return(status: 304)
-
-        expect(state.next_state.ssl_context.crls).to eq(crls)
-      end
-
-      it 'uses the local crl if refreshing fails in HTTP layer' do
-        stub_request(:get, %r{puppet-ca/v1/certificate_revocation_list/ca}).to_return(status: 503)
-
-        expect(state.next_state.ssl_context.crls).to eq(crls)
-      end
-
-      it 'uses the local crl if refreshing fails in TCP layer' do
-        stub_request(:get, %r{puppet-ca/v1/certificate_revocation_list/ca}).to_raise(Errno::ECONNREFUSED)
-
-        expect(state.next_state.ssl_context.crls).to eq(crls)
-      end
-
-      it 'uses the updated crl for the future requests' do
-        stub_request(:get, %r{puppet-ca/v1/certificate_revocation_list/ca}).to_return(status: 200, body: new_crl_bundle.join)
-
-        expect(state.next_state.ssl_context.crls.map(&:to_pem)).to eq(new_crl_bundle)
-      end
-
-      it 'updates the `last_update` time' do
-        stub_request(:get, %r{puppet-ca/v1/certificate_revocation_list/ca}).to_return(status: 200, body: new_crl_bundle.join)
-
-        expect_any_instance_of(Puppet::X509::CertProvider).to receive(:crl_last_update=).with(be_within(60).of(Time.now))
-
-        state.next_state
-      end
-    end
-  end
-
-  context 'when ensuring a client cert' do
-    context 'in state NeedKey' do
-      let(:ssl_context) { Puppet::SSL::SSLContext.new(cacerts: cacerts, crls: crls)}
-      let(:state) { Puppet::SSL::StateMachine::NeedKey.new(machine, ssl_context) }
-
-      it 'loads an existing private key and passes it to the next state' do
-        allow(cert_provider).to receive(:load_private_key).and_return(private_key)
-
-        st = state.next_state
-        expect(st).to be_instance_of(Puppet::SSL::StateMachine::NeedSubmitCSR)
-        expect(st.private_key).to eq(private_key)
-      end
-
-      it 'loads a matching private key and cert' do
-        allow(cert_provider).to receive(:load_private_key).and_return(private_key)
-        allow(cert_provider).to receive(:load_client_cert).and_return(client_cert)
-
-        st = state.next_state
-        expect(st).to be_instance_of(Puppet::SSL::StateMachine::Done)
-      end
-
-      it 'raises if the client cert is mismatched' do
-        allow(cert_provider).to receive(:load_private_key).and_return(private_key)
-        allow(cert_provider).to receive(:load_client_cert).and_return(cert_fixture('tampered-cert.pem'))
-
-        ssl_context = Puppet::SSL::SSLContext.new(cacerts: [cacert], crls: [crl])
-        state = Puppet::SSL::StateMachine::NeedKey.new(machine, ssl_context)
-        expect {
-          state.next_state
-        }.to raise_error(Puppet::SSL::SSLError, %r{The certificate for 'CN=signed' does not match its private key})
-      end
-
-      it 'generates a new RSA private key, saves it and passes it to the next state' do
-        allow(cert_provider).to receive(:load_private_key).and_return(nil)
-        expect(cert_provider).to receive(:save_private_key)
-
-        st = state.next_state
-        expect(st).to be_instance_of(Puppet::SSL::StateMachine::NeedSubmitCSR)
-        expect(st.private_key).to be_instance_of(OpenSSL::PKey::RSA)
-        expect(st.private_key).to be_private
-      end
-
-      it 'generates a new EC private key, saves it and passes it to the next state' do
-        Puppet[:key_type] = 'ec'
-        allow(cert_provider).to receive(:load_private_key).and_return(nil)
-        expect(cert_provider).to receive(:save_private_key)
-
-        st = state.next_state
-        expect(st).to be_instance_of(Puppet::SSL::StateMachine::NeedSubmitCSR)
-        expect(st.private_key).to be_instance_of(OpenSSL::PKey::EC)
-        expect(st.private_key).to be_private
-        expect(st.private_key.group.curve_name).to eq('prime256v1')
-      end
-
-      it 'generates a new EC private key with curve `secp384r1`, saves it and passes it to the next state' do
-        Puppet[:key_type] = 'ec'
-        Puppet[:named_curve] = 'secp384r1'
-        allow(cert_provider).to receive(:load_private_key).and_return(nil)
-        expect(cert_provider).to receive(:save_private_key)
-
-        st = state.next_state
-        expect(st).to be_instance_of(Puppet::SSL::StateMachine::NeedSubmitCSR)
-        expect(st.private_key).to be_instance_of(OpenSSL::PKey::EC)
-        expect(st.private_key).to be_private
-        expect(st.private_key.group.curve_name).to eq('secp384r1')
-      end
-
-      it 'raises if the named curve is unsupported' do
-        Puppet[:key_type] = 'ec'
-        Puppet[:named_curve] = 'infiniteloop'
-        allow(cert_provider).to receive(:load_private_key).and_return(nil)
-
-        expect {
-          state.next_state
-        }.to raise_error(OpenSSL::PKey::ECError, /(invalid|unknown) curve name/)
-      end
-
-      it 'raises an error if it fails to load the key' do
-        allow(cert_provider).to receive(:load_private_key).and_raise(OpenSSL::PKey::RSAError)
-
-        expect {
-          state.next_state
-        }.to raise_error(OpenSSL::PKey::RSAError)
-      end
-    end
-
-    context 'in state NeedSubmitCSR' do
-      let(:ssl_context) { Puppet::SSL::SSLContext.new(cacerts: cacerts, crls: crls)}
-      let(:state) { Puppet::SSL::StateMachine::NeedSubmitCSR.new(machine, ssl_context, private_key) }
-
-      def write_csr_attributes(data)
-        file_containing('state_machine_csr', YAML.dump(data))
-      end
-
-      before :each do
-        allow(cert_provider).to receive(:save_request)
-      end
-
-      it 'submits the CSR and transitions to NeedCert' do
-        stub_request(:put, %r{puppet-ca/v1/certificate_request/#{Puppet[:certname]}}).to_return(status: 200)
-
-        expect(state.next_state).to be_an_instance_of(Puppet::SSL::StateMachine::NeedCert)
-      end
-
-      it 'saves the CSR and transitions to NeedCert' do
-        stub_request(:put, %r{puppet-ca/v1/certificate_request/#{Puppet[:certname]}}).to_return(status: 200)
-
-        expect(cert_provider).to receive(:save_request).with(Puppet[:certname], instance_of(OpenSSL::X509::Request))
-
-        state.next_state
-      end
-
-      it 'includes DNS alt names' do
-        Puppet[:dns_alt_names] = "one,IP:192.168.0.1,DNS:two.com"
-
-        stub_request(:put, %r{puppet-ca/v1/certificate_request/#{Puppet[:certname]}}).with do |request|
-          csr = Puppet::SSL::CertificateRequest.from_instance(OpenSSL::X509::Request.new(request.body))
-          expect(
-            csr.subject_alt_names
-          ).to contain_exactly('DNS:one', 'IP Address:192.168.0.1', 'DNS:two.com', "DNS:#{Puppet[:certname]}")
-        end.to_return(status: 200)
-
-        state.next_state
-      end
-
-      it 'includes CSR attributes' do
-        Puppet[:csr_attributes] = write_csr_attributes(
-          'custom_attributes' => {
-              '1.3.6.1.4.1.34380.1.2.1' => 'CSR specific info',
-              '1.3.6.1.4.1.34380.1.2.2' => 'more CSR specific info'
-            }
-        )
-
-        stub_request(:put, %r{puppet-ca/v1/certificate_request/#{Puppet[:certname]}}).with do |request|
-          csr = Puppet::SSL::CertificateRequest.from_instance(OpenSSL::X509::Request.new(request.body))
-          expect(
-            csr.custom_attributes
-          ).to contain_exactly(
-                 {'oid' => '1.3.6.1.4.1.34380.1.2.1', 'value' => 'CSR specific info'},
-                 {'oid' => '1.3.6.1.4.1.34380.1.2.2', 'value' => 'more CSR specific info'}
-               )
-        end.to_return(status: 200)
-
-        state.next_state
-      end
-
-      it 'includes CSR extension requests' do
-        Puppet[:csr_attributes] = write_csr_attributes(
-          {
-            'extension_requests' => {
-              '1.3.6.1.4.1.34380.1.1.31415' => 'pi',
-              '1.3.6.1.4.1.34380.1.1.2718'  => 'e',
-            }
-          }
-        )
-
-        stub_request(:put, %r{puppet-ca/v1/certificate_request/#{Puppet[:certname]}}).with do |request|
-          csr = Puppet::SSL::CertificateRequest.from_instance(OpenSSL::X509::Request.new(request.body))
-          expect(
-            csr.request_extensions
-          ).to contain_exactly(
-                 {'oid' => '1.3.6.1.4.1.34380.1.1.31415', 'value' => 'pi'},
-                 {'oid' => '1.3.6.1.4.1.34380.1.1.2718', 'value' => 'e'}
-               )
-        end.to_return(status: 200)
-
-        state.next_state
-      end
-
-      it 'transitions to NeedCert if the server has a requested certificate' do
-        stub_request(:put, %r{puppet-ca/v1/certificate_request/#{Puppet[:certname]}}).to_return(status: 400, body: "#{Puppet[:certname]} already has a requested certificate")
-
-        expect(state.next_state).to be_an_instance_of(Puppet::SSL::StateMachine::NeedCert)
-      end
-
-      it 'transitions to NeedCert if the server has a signed certificate' do
-        stub_request(:put, %r{puppet-ca/v1/certificate_request/#{Puppet[:certname]}}).to_return(status: 400, body: "#{Puppet[:certname]} already has a signed certificate")
-
-        expect(state.next_state).to be_an_instance_of(Puppet::SSL::StateMachine::NeedCert)
-      end
-
-      it 'transitions to NeedCert if the server has a revoked certificate' do
-        stub_request(:put, %r{puppet-ca/v1/certificate_request/#{Puppet[:certname]}}).to_return(status: 400, body: "#{Puppet[:certname]} already has a revoked certificate")
-
-        expect(state.next_state).to be_an_instance_of(Puppet::SSL::StateMachine::NeedCert)
-      end
-
-      it 'raises if the server errors' do
-        stub_request(:put, %r{puppet-ca/v1/certificate_request/#{Puppet[:certname]}}).to_return(status: 500)
-
-        st = state.next_state
-        expect(st).to be_an_instance_of(Puppet::SSL::StateMachine::Error)
-        expect(st.message).to eq("Failed to submit the CSR, HTTP response was 500")
-      end
-
-      it "verifies the server's certificate when submitting the CSR" do
-        stub_request(:put, %r{puppet-ca/v1/certificate_request/#{Puppet[:certname]}}).to_return(status: 200)
-
-        receive_count = 0
-        allow_any_instance_of(Net::HTTP).to receive(:verify_mode=).with(OpenSSL::SSL::VERIFY_PEER) { receive_count += 1 }
-
-        state.next_state
-
-        expect(receive_count).to eq(2)
-      end
-    end
-
-    context 'in state NeedCert' do
-      let(:ca_chain) { [cert_fixture('ca.pem'), cert_fixture('intermediate.pem')] }
-      let(:crl_chain) { [crl_fixture('crl.pem'), crl_fixture('intermediate-crl.pem')] }
-      let(:ssl_context) { Puppet::SSL::SSLContext.new(cacerts: ca_chain, crls: crl_chain)}
-      let(:state) { Puppet::SSL::StateMachine::NeedCert.new(machine, ssl_context, private_key) }
-
-      it 'transitions to Done if the cert is signed and matches our private key' do
-        allow(cert_provider).to receive(:save_client_cert)
-        allow(cert_provider).to receive(:save_request)
-
-        stub_request(:get, %r{puppet-ca/v1/certificate/#{Puppet[:certname]}}).to_return(status: 200, body: client_cert.to_pem)
-
-        expect(state.next_state).to be_an_instance_of(Puppet::SSL::StateMachine::Done)
-      end
-
-      it "prints a message if the cert isn't signed yet" do
-        stub_request(:get, %r{puppet-ca/v1/certificate/#{Puppet[:certname]}}).to_return(status: 404)
-
-        expect {
-          state.next_state
-        }.to output(/Couldn't fetch certificate from CA server; you might still need to sign this agent's certificate \(#{Puppet[:certname]}\)/).to_stdout
-      end
-
-      it 'transitions to Error if the cert does not match our private key' do
-        wrong_cert = cert_fixture('127.0.0.1.pem')
-        stub_request(:get, %r{puppet-ca/v1/certificate/#{Puppet[:certname]}}).to_return(status: 200, body: wrong_cert.to_pem)
-
-        st = state.next_state
-        expect(st).to be_an_instance_of(Puppet::SSL::StateMachine::Error)
-        expect(st.message).to eq("The certificate for 'CN=127.0.0.1' does not match its private key")
-      end
-
-      it 'transitions to Wait if the server returns non-200' do
-        stub_request(:get, %r{puppet-ca/v1/certificate/#{Puppet[:certname]}}).to_return(status: 404)
-
-        allow($stdout).to receive(:puts).with(/Couldn't fetch certificate from CA server; you might still need to sign this agent's certificate/)
-        expect(state.next_state).to be_an_instance_of(Puppet::SSL::StateMachine::Wait)
-      end
-
-      it "verifies the server's certificate when getting the client cert" do
-        stub_request(:get, %r{puppet-ca/v1/certificate/#{Puppet[:certname]}}).to_return(status: 200, body: client_cert.to_pem)
-        allow(cert_provider).to receive(:save_client_cert)
-        allow(cert_provider).to receive(:save_request)
-
-        receive_count = 0
-        allow_any_instance_of(Net::HTTP).to receive(:verify_mode=).with(OpenSSL::SSL::VERIFY_PEER) { receive_count += 1 }
-
-        state.next_state
-
-        expect(receive_count).to eq(2)
-      end
-
-      it 'does not save an invalid client cert' do
-        stub_request(:get, %r{puppet-ca/v1/certificate/#{Puppet[:certname]}}).to_return(status: 200, body: <<~END)
-          -----BEGIN CERTIFICATE-----
-          MIIBpDCCAQ2gAwIBAgIBAjANBgkqhkiG9w0BAQsFADAfMR0wGwYDVQQDDBRUZXN0
-        END
-
-        st = state.next_state
-        expect(st).to be_an_instance_of(Puppet::SSL::StateMachine::Error)
-        expect(st.message).to match(/Failed to parse certificate:/)
-        expect(File).to_not exist(Puppet[:hostcert])
-      end
-
-      it 'does not save a mismatched client cert' do
-        wrong_cert = cert_fixture('127.0.0.1.pem').to_pem
-        stub_request(:get, %r{puppet-ca/v1/certificate/#{Puppet[:certname]}}).to_return(status: 200, body: wrong_cert)
-
-        st = state.next_state
-        expect(st).to be_an_instance_of(Puppet::SSL::StateMachine::Error)
-        expect(st.message).to eq("The certificate for 'CN=127.0.0.1' does not match its private key")
-        expect(File).to_not exist(Puppet[:hostcert])
-      end
-
-      it 'does not save a revoked client cert' do
-        revoked_cert = cert_fixture('revoked.pem').to_pem
-        stub_request(:get, %r{puppet-ca/v1/certificate/#{Puppet[:certname]}}).to_return(status: 200, body: revoked_cert)
-
-        st = state.next_state
-        expect(st).to be_an_instance_of(Puppet::SSL::StateMachine::Error)
-        expect(st.message).to eq("Certificate 'CN=revoked' is revoked")
-        expect(File).to_not exist(Puppet[:hostcert])
-      end
-    end
-
-    context 'in state Wait' do
-      let(:ssl_context) { Puppet::SSL::SSLContext.new(cacerts: cacerts, crls: crls)}
-
-      it 'exits with 1 if waitforcert is 0' do
-        machine = described_class.new(waitforcert: 0)
-
-        expect {
-          expect {
-            Puppet::SSL::StateMachine::Wait.new(machine).next_state
-          }.to exit_with(1)
-        }.to output(/Exiting now because the waitforcert setting is set to 0./).to_stdout
-      end
-
-      it 'sleeps and transitions to NeedLock' do
-        machine = described_class.new(waitforcert: 15)
-
-        state = Puppet::SSL::StateMachine::Wait.new(machine)
-        expect(Kernel).to receive(:sleep).with(15)
-
-        expect(Puppet).to receive(:info).with(/Will try again in 15 seconds./)
-
-        expect(state.next_state).to be_an_instance_of(Puppet::SSL::StateMachine::NeedLock)
-      end
-
-      it 'sleeps and transitions to NeedLock when maxwaitforcert is set' do
-        machine = described_class.new(waitforcert: 15, maxwaitforcert: 30)
-
-        state = Puppet::SSL::StateMachine::Wait.new(machine)
-        expect(Kernel).to receive(:sleep).with(15)
-
-        expect(Puppet).to receive(:info).with(/Will try again in 15 seconds./)
-
-        expect(state.next_state).to be_an_instance_of(Puppet::SSL::StateMachine::NeedLock)
-      end
-
-      it 'waits indefinitely by default' do
-        machine = described_class.new
-        expect(machine.wait_deadline).to eq(Float::INFINITY)
-      end
-
-      it 'exits with 1 if maxwaitforcert is exceeded' do
-        machine = described_class.new(maxwaitforcert: 1)
-
-        # 5 minutes in the future
-        future = Time.now + (5 * 60)
-        allow(Time).to receive(:now).and_return(future)
-
-        expect {
-          expect {
-            Puppet::SSL::StateMachine::Wait.new(machine).next_state
-          }.to exit_with(1)
-        }.to output(/Couldn't fetch certificate from CA server; you might still need to sign this agent's certificate \(.*\). Exiting now because the maxwaitforcert timeout has been exceeded./).to_stdout
-      end
-
-      it 'closes the pool before sleeping' do
-        machine = described_class.new(waitforcert: 15)
-
-        state = Puppet::SSL::StateMachine::Wait.new(machine)
-        expect(Puppet.runtime[:http].pool).to receive(:close).and_call_original
-        expect(Kernel).to receive(:sleep).with(15).ordered
-
-        state.next_state
-      end
-
-      it 'releases the lock while sleeping' do
-        lockfile = Puppet::Util::Pidlock.new(Puppet[:ssl_lockfile])
-        machine = described_class.new(lockfile: lockfile)
-        state = Puppet::SSL::StateMachine::Wait.new(machine)
-
-        # pidlock should be unlocked while sleeping
-        allow(Kernel).to receive(:sleep) do
-          expect(lockfile).to_not be_locked
-        end
-
-        # lock before running the state
-        lockfile.lock
-        state.next_state
-      end
-    end
-
-    context 'in state NeedLock' do
-      let(:ssl_context) { Puppet::SSL::SSLContext.new(cacerts: []) }
-      let(:lockfile) { Puppet::Util::Pidlock.new(Puppet[:ssl_lockfile]) }
-      let(:machine) { described_class.new(lockfile: lockfile) }
-      let(:state) { Puppet::SSL::StateMachine::NeedLock.new(machine) }
-
-      it 'acquires the lock and transitions to NeedCACerts' do
-        expect(state.next_state).to be_an_instance_of(Puppet::SSL::StateMachine::NeedCACerts)
-        expect(lockfile).to be_locked
-      end
-
-      it 'transitions to LockFailure if it fails to acquire the lock' do
-        expect(lockfile).to receive(:lock).and_return(false)
-        expect(state.next_state).to be_an_instance_of(Puppet::SSL::StateMachine::LockFailure)
-      end
-    end
-  end
-end