12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364 |
- require 'spec_helper'
- require 'puppet/util/puppetdb_validator'
- describe 'Puppet::Util::PuppetdbValidator' do
- before do
- response_ok = stub()
- response_ok.stubs(:kind_of?).with(Net::HTTPSuccess).returns(true)
- response_not_found = stub()
- response_not_found.stubs(:kind_of?).with(Net::HTTPSuccess).returns(false)
- response_not_found.stubs(:code).returns(404)
- response_not_found.stubs(:msg).returns('Not found')
- conn_ok = stub()
- conn_ok.stubs(:get).with('/pdb/meta/v1/version', {"Accept" => "application/json"}).returns(response_ok)
- conn_ok.stubs(:read_timeout=).with(2)
- conn_ok.stubs(:open_timeout=).with(2)
- conn_not_found = stub()
- conn_not_found.stubs(:get).with('/pdb/meta/v1/version', {"Accept" => "application/json"}).returns(response_not_found)
- Puppet::Network::HttpPool.stubs(:http_instance).raises('Unknown host')
- Puppet::Network::HttpPool.stubs(:http_instance).with('mypuppetdb.com', 8080, true).raises('Connection refused')
- Puppet::Network::HttpPool.stubs(:http_instance).with('mypuppetdb.com', 8080, false).returns(conn_ok)
- Puppet::Network::HttpPool.stubs(:http_instance).with('mypuppetdb.com', 8081, true).returns(conn_ok)
- Puppet::Network::HttpPool.stubs(:http_instance).with('wrongserver.com', 8081, true).returns(conn_not_found)
- end
- it 'returns true if connection succeeds' do
- validator = Puppet::Util::PuppetdbValidator.new('mypuppetdb.com', 8081)
- expect(validator.attempt_connection).to be true
- end
- it 'should still validate without ssl' do
- Puppet[:configtimeout] = 2
- validator = Puppet::Util::PuppetdbValidator.new('mypuppetdb.com', 8080, false)
- expect(validator.attempt_connection).to be true
- end
- it 'returns false and issues an appropriate notice if connection is refused' do
- puppetdb_server = 'mypuppetdb.com'
- puppetdb_port = 8080
- validator = Puppet::Util::PuppetdbValidator.new(puppetdb_server, puppetdb_port)
- Puppet.expects(:notice).with("Unable to connect to puppetdb server (https://#{puppetdb_server}:#{puppetdb_port}): Connection refused")
- expect(validator.attempt_connection).to be false
- end
- it 'returns false and issues an appropriate notice if connection succeeds but puppetdb is not available' do
- puppetdb_server = 'wrongserver.com'
- puppetdb_port = 8081
- validator = Puppet::Util::PuppetdbValidator.new(puppetdb_server, puppetdb_port)
- Puppet.expects(:notice).with("Unable to connect to puppetdb server (https://#{puppetdb_server}:#{puppetdb_port}): [404] Not found")
- expect(validator.attempt_connection).to be false
- end
- it 'returns false and issues an appropriate notice if host:port is unreachable or does not exist' do
- puppetdb_server = 'non-existing.com'
- puppetdb_port = nil
- validator = Puppet::Util::PuppetdbValidator.new(puppetdb_server, puppetdb_port)
- Puppet.expects(:notice).with("Unable to connect to puppetdb server (https://#{puppetdb_server}:#{puppetdb_port}): Unknown host")
- expect(validator.attempt_connection).to be false
- end
- end
|