require 'spec_helper'

describe 'haproxy', type: :class do
  let(:default_facts) do
    {
      concat_basedir: '/dne',
      ipaddress: '10.10.10.10',
    }
  end

  describe 'for OS-agnostic configuration' do
    %w[Debian RedHat Archlinux FreeBSD Gentoo].each do |osfamily|
      context "on #{osfamily} family operatingsystems" do
        let(:facts) do
          { osfamily: osfamily }.merge default_facts
        end
        let(:params) do
          {
            'service_ensure' => 'running',
            'package_ensure' => 'present',
            'service_manage' => true,
          }
        end

        it 'installs the haproxy package' do
          subject.should contain_package('haproxy').with(
            'ensure' => 'present',
          )
        end
        it 'installs the haproxy service' do
          subject.should contain_service('haproxy').with(
            'ensure' => 'running', 'enable' => 'true',
            'hasrestart' => 'true', 'hasstatus' => 'true'
          )
        end
      end
      context "on #{osfamily} specifying a package version" do
        let(:facts) do
          { osfamily: osfamily }.merge default_facts
        end
        let(:params) do
          {
            'service_ensure' => 'running',
            'package_ensure' => '1.7.9-1',
            'service_manage' => true,
          }
        end

        it 'installs the haproxy package in a specific version' do
          subject.should contain_package('haproxy').with(
            'ensure' => '1.7.9-1',
          )
        end
        it 'installs the haproxy service' do
          subject.should contain_service('haproxy').with(
            'ensure' => 'running', 'enable' => 'true',
            'hasrestart' => 'true', 'hasstatus' => 'true'
          )
        end
      end
      # C9938
      context "on #{osfamily} when specifying custom content" do
        let(:facts) do
          { osfamily: osfamily }.merge default_facts
        end
        let(:params) do
          { 'custom_fragment' => "listen stats :9090\n  mode http\n  stats uri /\n  stats auth puppet:puppet\n" }
        end

        it 'sets the haproxy package' do
          subject.should contain_concat__fragment('haproxy-haproxy-base').with_content(
            %r{listen stats :9090\n  mode http\n  stats uri /\n  stats auth puppet:puppet\n},
          )
        end
      end
    end
  end

  describe 'for linux operating systems' do
    %w[Debian RedHat Archlinux Gentoo].each do |osfamily|
      context "on #{osfamily} family operatingsystems" do
        let(:facts) do
          { osfamily: osfamily }.merge default_facts
        end

        it 'sets up /etc/haproxy/haproxy.cfg as a concat resource' do
          subject.should contain_concat('/etc/haproxy/haproxy.cfg').with(
            'owner' => '0',
            'group' => '0',
            'mode'  => '0640',
          )
        end
        it 'manages the chroot directory' do
          subject.should contain_file('/var/lib/haproxy').with(
            'ensure' => 'directory',
            'owner'  => 'haproxy',
            'group'  => 'haproxy',
          )
        end
        it 'contains a header concat fragment' do
          subject.should contain_concat__fragment('haproxy-00-header').with(
            'target'  => '/etc/haproxy/haproxy.cfg',
            'order'   => '01',
            'content' => "# This file managed by Puppet\n",
          )
        end
        it 'contains a haproxy-haproxy-base concat fragment' do
          subject.should contain_concat__fragment('haproxy-haproxy-base').with(
            'target'  => '/etc/haproxy/haproxy.cfg',
            'order'   => '10',
          )
        end
        describe 'Base concat fragment contents' do
          let(:contents) { param_value(catalogue, 'concat::fragment', 'haproxy-haproxy-base', 'content').split("\n") }

          # C9936 C9937
          it 'contains global and defaults sections' do
            contents.should include('global')
            contents.should include('defaults')
          end
          it 'logs to an ip address for local0' do
            contents.should be_any do |match|
              match =~ %r{  log  \d+(\.\d+){3} local0}
            end
          end
          it 'specifies the default chroot' do
            contents.should include('  chroot  /var/lib/haproxy')
          end
          it 'specifies the correct user' do
            contents.should include('  user  haproxy')
          end
          it 'specifies the correct group' do
            contents.should include('  group  haproxy')
          end
          it 'specifies the correct pidfile' do
            contents.should include('  pidfile  /var/run/haproxy.pid')
          end
        end
      end
      context "on #{osfamily} family operatingsystems with setting haproxy.cfg location" do
        let(:facts) do
          { osfamily: osfamily }.merge default_facts
        end
        let(:params) do
          {
            'config_file' => '/tmp/haproxy.cfg',
          }
        end

        it 'sets up /tmp/haproxy.cfg as a concat resource' do
          subject.should contain_concat('/tmp/haproxy.cfg').with(
            'owner' => '0',
            'group' => '0',
            'mode'  => '0640',
          )
        end
      end
      context "on #{osfamily} family operatingsystems without managing the service" do
        let(:facts) do
          { osfamily: osfamily }.merge default_facts
        end
        let(:params) do
          {
            'service_ensure' => true,
            'package_ensure' => 'present',
            'service_manage' => false,
          }
        end

        it 'installs the haproxy package' do
          subject.should contain_package('haproxy').with(
            'ensure' => 'present',
          )
        end
        it 'does not manage the haproxy service' do
          subject.should_not contain_service('haproxy')
        end
        it 'sets up /etc/haproxy/haproxy.cfg as a concat resource' do
          subject.should contain_concat('/etc/haproxy/haproxy.cfg').with(
            'owner' => '0',
            'group' => '0',
            'mode'  => '0640',
          )
        end
        it 'manages the chroot directory' do
          subject.should contain_file('/var/lib/haproxy').with(
            'ensure' => 'directory',
          )
        end
        it 'contains a header concat fragment' do
          subject.should contain_concat__fragment('haproxy-00-header').with(
            'target'  => '/etc/haproxy/haproxy.cfg',
            'order'   => '01',
            'content' => "# This file managed by Puppet\n",
          )
        end
        it 'contains a haproxy-base concat fragment' do
          subject.should contain_concat__fragment('haproxy-haproxy-base').with(
            'target'  => '/etc/haproxy/haproxy.cfg',
            'order'   => '10',
          )
        end
        describe 'Base concat fragment contents' do
          let(:contents) { param_value(catalogue, 'concat::fragment', 'haproxy-haproxy-base', 'content').split("\n") }

          it 'contains global and defaults sections' do
            contents.should include('global')
            contents.should include('defaults')
          end
          it 'logs to an ip address for local0' do
            contents.should be_any do |match|
              match =~ %r{  log  \d+(\.\d+){3} local0}
            end
          end
          it 'specifies the default chroot' do
            contents.should include('  chroot  /var/lib/haproxy')
          end
          it 'specifies the correct user' do
            contents.should include('  user  haproxy')
          end
          it 'specifies the correct group' do
            contents.should include('  group  haproxy')
          end
          it 'specifies the correct pidfile' do
            contents.should include('  pidfile  /var/run/haproxy.pid')
          end
        end
      end
      context "on #{osfamily} when specifying a restart_command" do
        let(:facts) do
          { osfamily: osfamily }.merge default_facts
        end
        let(:params) do
          {
            'restart_command' => '/etc/init.d/haproxy reload',
            'service_manage'  => true,
          }
        end

        it 'sets the haproxy package' do
          subject.should contain_service('haproxy').with(
            'restart' => '/etc/init.d/haproxy reload',
          )
        end
      end
    end
  end

  describe 'for freebsd' do
    context 'when on freebsd family operatingsystems' do
      let(:facts) do
        { osfamily: 'FreeBSD' }.merge default_facts
      end

      it 'sets up /usr/local/etc/haproxy.conf as a concat resource' do
        subject.should contain_concat('/usr/local/etc/haproxy.conf').with(
          'owner' => '0',
          'group' => '0',
          'mode'  => '0640',
        )
      end
      it 'manages the chroot directory' do
        subject.should contain_file('/usr/local/haproxy').with(
          'ensure' => 'directory',
        )
      end
      it 'contains a header concat fragment' do
        subject.should contain_concat__fragment('haproxy-00-header').with(
          'target'  => '/usr/local/etc/haproxy.conf',
          'order'   => '01',
          'content' => "# This file managed by Puppet\n",
        )
      end
      it 'contains a haproxy-base concat fragment' do
        subject.should contain_concat__fragment('haproxy-haproxy-base').with(
          'target'  => '/usr/local/etc/haproxy.conf',
          'order'   => '10',
        )
      end
      describe 'Base concat fragment contents' do
        let(:contents) { param_value(catalogue, 'concat::fragment', 'haproxy-haproxy-base', 'content').split("\n") }

        # C9936 C9937
        it 'contains global and defaults sections' do
          contents.should include('global')
          contents.should include('defaults')
        end
        it 'logs to an ip address for local0' do
          contents.should be_any do |match|
            match =~ %r{  log  \d+(\.\d+){3} local0}
          end
        end
        it 'specifies the default chroot' do
          contents.should include('  chroot  /usr/local/haproxy')
        end
        it 'specifies the correct pidfile' do
          contents.should include('  pidfile  /var/run/haproxy.pid')
        end
      end
    end
    context 'when on freebsd family operatingsystems without managing the service' do
      let(:facts) do
        { osfamily: 'FreeBSD' }.merge default_facts
      end
      let(:params) do
        {
          'service_ensure' => true,
          'package_ensure' => 'present',
          'service_manage' => false,
        }
      end

      it 'installs the haproxy package' do
        subject.should contain_package('haproxy').with(
          'ensure' => 'present',
        )
      end
      it 'does not manage the haproxy service' do
        subject.should_not contain_service('haproxy')
      end
      it 'sets up /usr/local/etc/haproxy.conf as a concat resource' do
        subject.should contain_concat('/usr/local/etc/haproxy.conf').with(
          'owner' => '0',
          'group' => '0',
          'mode'  => '0640',
        )
      end
      it 'manages the chroot directory' do
        subject.should contain_file('/usr/local/haproxy').with(
          'ensure' => 'directory',
        )
      end
      it 'contains a header concat fragment' do
        subject.should contain_concat__fragment('haproxy-00-header').with(
          'target'  => '/usr/local/etc/haproxy.conf',
          'order'   => '01',
          'content' => "# This file managed by Puppet\n",
        )
      end
      it 'contains a haproxy-base concat fragment' do
        subject.should contain_concat__fragment('haproxy-haproxy-base').with(
          'target'  => '/usr/local/etc/haproxy.conf',
          'order'   => '10',
        )
      end
      describe 'Base concat fragment contents' do
        let(:contents) { param_value(catalogue, 'concat::fragment', 'haproxy-haproxy-base', 'content').split("\n") }

        it 'contains global and defaults sections' do
          contents.should include('global')
          contents.should include('defaults')
        end
        it 'logs to an ip address for local0' do
          contents.should be_any do |match|
            match =~ %r{  log  \d+(\.\d+){3} local0}
          end
        end
        it 'specifies the default chroot' do
          contents.should include('  chroot  /usr/local/haproxy')
        end
        it 'specifies the correct pidfile' do
          contents.should include('  pidfile  /var/run/haproxy.pid')
        end
      end
    end
    context 'when on freebsd when specifying a restart_command' do
      let(:facts) do
        { osfamily: 'FreeBSD' }.merge default_facts
      end
      let(:params) do
        {
          'restart_command' => '/usr/local/etc/rc.d/haproxy reload',
          'service_manage'  => true,
        }
      end

      it 'sets the haproxy package' do
        subject.should contain_service('haproxy').with(
          'restart' => '/usr/local/etc/rc.d/haproxy reload',
        )
      end
    end
  end

  describe 'for OS-specific configuration' do
    context 'when only on Debian family operatingsystems' do
      let(:facts) do
        { osfamily: 'Debian' }.merge default_facts
      end

      it 'manages haproxy service defaults' do
        subject.should contain_file('/etc/default/haproxy')
        verify_contents(catalogue, '/etc/default/haproxy', ['ENABLED=1'])
      end
    end
    context 'when only on RedHat family operatingsystems' do
      let(:facts) do
        { osfamily: 'RedHat' }.merge default_facts
      end

      pending('Not yet implemented')
    end
    context 'when only on Gentoo family operatingsystems' do
      let(:facts) do
        { osfamily: 'Gentoo' }.merge default_facts
      end

      it 'creates directory /etc/haproxy' do
        subject.should contain_file('/etc/haproxy').with(
          'ensure' => 'directory',
        )
      end
    end
  end

  describe 'when merging global and defaults options with user-supplied overrides and additions' do
    # For testing the merging functionality we restrict ourselves to
    # Debian OS family so that we don't have to juggle different sets of
    # global_options and defaults_options (like for FreeBSD).
    ['Debian'].each do |osfamily|
      context "on #{osfamily} family operatingsystems" do
        let(:facts) do
          { osfamily: osfamily }.merge default_facts
        end
        let(:contents) { param_value(catalogue, 'concat::fragment', 'haproxy-haproxy-base', 'content').split("\n") }
        let(:params) do
          {
            'merge_options'   => true,
            'global_options'  => {
              'log-send-hostname' => '',
              'chroot'            => '/srv/haproxy-chroot',
              'maxconn'           => nil,
              'stats'             => [
                'socket /var/lib/haproxy/admin.sock mode 660 level admin',
                'timeout 30s',
              ],
            },
            'defaults_options' => {
              'mode'    => 'http',
              'option'  => [
                'abortonclose',
                'logasap',
                'dontlognull',
                'httplog',
                'http-server-close',
                'forwardfor except 127.0.0.1',
              ],
              'timeout' => [
                'connect 5s',
                'client 1m',
                'server 1m',
                'check 7s',
              ],
            },
          }
        end

        it 'manages a custom chroot directory' do
          subject.should contain_file('/srv/haproxy-chroot').with(
            'ensure' => 'directory',
            'owner'  => 'haproxy',
            'group'  => 'haproxy',
          )
        end
        it 'contains global and defaults sections' do
          contents.should include('global')
          contents.should include('defaults')
        end
        it 'sends hostname with log in global options' do
          contents.should include('  log-send-hostname  ')
        end
        it 'enables admin stats and stats timeout in global options' do
          contents.should include('  stats  socket /var/lib/haproxy/admin.sock mode 660 level admin')
          contents.should include('  stats  timeout 30s')
        end
        it 'logs to an ip address for local0' do
          contents.should be_any do |match|
            match =~ %r{  log  \d+(\.\d+){3} local0}
          end
        end
        it 'specifies the correct user' do
          contents.should include('  user  haproxy')
        end
        it 'specifies the correct group' do
          contents.should include('  group  haproxy')
        end
        it 'specifies the correct pidfile' do
          contents.should include('  pidfile  /var/run/haproxy.pid')
        end
        it 'sets mode http in default options' do
          contents.should include('  mode  http')
        end
        it 'does not set the global parameter "maxconn"' do
          contents.should_not include('  maxconn  4000')
        end
        expected_array_one = ['  option  abortonclose', '  option  logasap', '  option  dontlognull',
                              '  option  httplog', '  option  http-server-close', '  option  forwardfor except 127.0.0.1']
        it 'sets various options in defaults, removing the "redispatch" option' do
          contents.should_not include('  option  redispatch')
          expected_array_one.each do |expected|
            contents.should include(expected)
          end
        end
        expected_array_two = ['  timeout  connect 5s', '  timeout  check 7s', '  timeout  client 1m', '  timeout  server 1m']
        it 'sets timeouts in defaults, removing the "http-request 10s" and "queue 1m" timeout' do
          contents.should_not include('  timeout  http-request 10s')
          contents.should_not include('  timeout  queue 1m')
          expected_array_two.each do |expected|
            contents.should include(expected)
          end
        end
      end
    end
  end

  describe 'when overriding global and defaults options with user-supplied overrides and additions' do
    # For testing the merging functionality we restrict ourselves to
    # Debian OS family so that we don't have to juggle different sets of
    # global_options and defaults_options (like for FreeBSD).
    ['Debian'].each do |osfamily|
      context "on #{osfamily} family operatingsystems" do
        let(:facts) do
          { osfamily: osfamily }.merge default_facts
        end
        let(:contents) { param_value(catalogue, 'concat::fragment', 'haproxy-haproxy-base', 'content').split("\n") }
        let(:params) do
          {
            'merge_options'   => false,
            'global_options'  => {
              'log-send-hostname' => '',
              'chroot'            => '/srv/haproxy-chroot',
              'stats'             => [
                'socket /var/lib/haproxy/admin.sock mode 660 level admin',
                'timeout 30s',
              ],
            },
            'defaults_options' => {
              'mode'    => 'http',
              'option'  => [
                'abortonclose',
                'logasap',
                'dontlognull',
                'httplog',
                'http-server-close',
                'forwardfor except 127.0.0.1',
              ],
              'timeout' => [
                'connect 5s',
                'client 1m',
                'server 1m',
                'check 7s',
              ],
            },
          }
        end

        it 'manages a custom chroot directory' do
          subject.should contain_file('/srv/haproxy-chroot').with(
            'ensure' => 'directory',
          )
        end
        it 'contains global and defaults sections' do
          contents.should include('global')
          contents.should include('defaults')
        end
        it 'sends hostname with log in global options' do
          contents.should include('  log-send-hostname  ')
        end
        it 'enables admin stats and stats timeout in global options' do
          contents.should include('  stats  socket /var/lib/haproxy/admin.sock mode 660 level admin')
          contents.should include('  stats  timeout 30s')
        end
        it 'sets mode http in default options' do
          contents.should include('  mode  http')
        end
        it 'does not set the global parameter "maxconn"' do
          contents.should_not include('  maxconn  4000')
        end
        expected_array_one = ['  option  abortonclose', '  option  logasap', '  option  dontlognull', '  option  httplog',
                              '  option  http-server-close', '  option  forwardfor except 127.0.0.1']
        it 'sets various options in defaults, removing the "redispatch" option' do
          contents.should_not include('  option  redispatch')
          expected_array_one.each do |expected|
            contents.should include(expected)
          end
        end
        expected_array_two = ['  timeout  connect 5s', '  timeout  check 7s', '  timeout  client 1m', '  timeout  server 1m']
        it 'sets timeouts in defaults, removing the "http-request 10s" and "queue 1m" timeout' do
          contents.should_not include('  timeout  http-request 10s')
          contents.should_not include('  timeout  queue 1m')
          expected_array_two.each do |expected|
            contents.should include(expected)
          end
        end
      end
    end
  end

  context 'when on unsupported operatingsystems' do
    let(:facts) do
      { osfamily: 'windows' }.merge default_facts
    end

    it do
      expect {
        is_expected.to contain_service('haproxy')
      }.to raise_error(Puppet::Error, %r{operating system is not supported with the haproxy module})
    end
  end
end