puppet nftables

# frozen_string_literal: true

require 'spec_helper'

describe 'nftables' do

  let(:pre_condition) { 'Exec{path => "/bin"}' }

  on_supported_os.each do |os, os_facts|

    context "on #{os}" do

      let(:facts) { os_facts }

      it { is_expected.to compile }

      it { is_expected.to contain_package('nftables') }

      it {

        is_expected.to contain_file('/etc/nftables').with(

          ensure: 'directory',

          owner: 'root',

          group: 'root',

          mode: '0750'

        )

      }

      it {

        expect(subject).to contain_file('/etc/nftables/puppet.nft').with(

          ensure: 'file',

          owner: 'root',

          group: 'root',

          mode: '0640',

          content: %r{flush ruleset}

        )

      }

      it {

        expect(subject).to contain_file('/etc/nftables/puppet').with(

          ensure: 'directory',

          owner: 'root',

          group: 'root',

          mode: '0750',

          purge: true,

          force: true,

          recurse: true

        )

      }

      it {

        expect(subject).to contain_file('/etc/nftables/puppet-preflight.nft').with(

          ensure: 'file',

          owner: 'root',

          group: 'root',

          mode: '0640',

          content: %r{flush ruleset}

        )

      }

      it {

        expect(subject).to contain_file('/etc/nftables/puppet-preflight').with(

          ensure: 'directory',

          owner: 'root',

          group: 'root',

          mode: '0750',

          purge: true,

          force: true,

          recurse: true

        )

      }

      it {

        expect(subject).to contain_exec('nft validate').with(

          refreshonly: true,

          command: %r{^/usr/sbin/nft -I /etc/nftables/puppet-preflight -c -f /etc/nftables/puppet-preflight.nft.*}

        )

      }

      it {

        expect(subject).to contain_service('nftables').with(

          ensure: 'running',

          enable: true,

          hasrestart: true,

          restart: %r{/usr/bin/systemctl reload nft.*}

        )

      }

      if os_facts[:os]['family'] == 'Archlinux'

        it {

          expect(subject).to contain_systemd__dropin_file('puppet_nft.conf').with(

            content: %r{^ExecReload=/sbin/nft -I /etc/nftables/puppet -f /etc/nftables.conf$}

          )

        }

        it {

          expect(subject).to contain_service('firewalld').with(

            ensure: 'stopped',

            enable: false

          )

        }

      else

        it {

          expect(subject).to contain_systemd__dropin_file('puppet_nft.conf').with(

            content: %r{^ExecReload=/sbin/nft -I /etc/nftables/puppet -f /etc/sysconfig/nftables.conf$}

          )

        }

        it {

          expect(subject).to contain_service('firewalld').with(

            ensure: 'stopped',

            enable: 'mask'

          )

        }

      end

      it { is_expected.to contain_class('nftables::inet_filter') }

      it { is_expected.to contain_class('nftables::ip_nat') }

      it { is_expected.to contain_class('nftables::rules::out::http') }

      it { is_expected.to contain_class('nftables::rules::out::https') }

      it { is_expected.to contain_class('nftables::rules::out::dns') }

      it { is_expected.to contain_class('nftables::rules::out::chrony') }

      it { is_expected.not_to contain_class('nftables::rules::out::all') }

      it { is_expected.not_to contain_nftables__rule('default_out-all') }

      context 'with out_all set true' do

        let(:params) do

          {

            out_all: true,

          }

        end

        it { is_expected.to contain_class('nftables::rules::out::all') }

        it { is_expected.not_to contain_class('nftables::rules::out::http') }

        it { is_expected.not_to contain_class('nftables::rules::out::https') }

        it { is_expected.not_to contain_class('nftables::rules::out::dns') }

        it { is_expected.not_to contain_class('nftables::rules::out::chrony') }

        it { is_expected.to contain_nftables__rule('default_out-all').with_content('accept') }

        it { is_expected.to contain_nftables__rule('default_out-all').with_order('90') }

      end

      context 'with custom rules' do

        let(:params) do

          {

            rules: {

              'INPUT-web_accept' => {

                order: '50',

                content: 'iifname eth0 tcp dport { 80, 443 } accept',

              },

            },

          }

        end

        it {

          expect(subject).to contain_concat__fragment('nftables-inet-filter-chain-INPUT-rule-web_accept').with(

            target: 'nftables-inet-filter-chain-INPUT',

            content: %r{^  iifname eth0 tcp dport \{ 80, 443 \} accept$},

            order: '50-nftables-inet-filter-chain-INPUT-rule-web_accept-b'

          )

        }

      end

      context 'with custom sets' do

        let(:params) do

          {

            sets: {

              'testset1' => {

                type: 'ipv4_addr',

                gc_interval: 2,

              },

              'testset2' => {

                type: 'ipv6_addr',

                elements: ['2a02:62:c601::dead:beef'],

              },

            },

          }

        end

        it {

          expect(subject).to contain_nftables__set('testset1').with(

            type: 'ipv4_addr',

            gc_interval: 2,

            table: 'inet-filter'

          )

        }

        it {

          expect(subject).to contain_nftables__set('testset2').with(

            type: 'ipv6_addr',

            elements: ['2a02:62:c601::dead:beef'],

            table: 'inet-filter'

          )

        }

      end

      context 'without masking firewalld' do

        let(:params) do

          {

            'firewalld_enable' => false,

          }

        end

        it {

          expect(subject).to contain_service('firewalld').with(

            ensure: 'stopped',

            enable: false

          )

        }

      end

      context 'with no default filtering rules' do

        let(:params) do

          {

            'inet_filter' => false,

          }

        end

        it { is_expected.to contain_class('nftables::ip_nat') }

        it { is_expected.not_to contain_class('nftables::inet_filter') }

      end

      context 'with no default tables, chains or rules' do

        let(:params) do

          {

            'inet_filter' => false,

            'nat' => false,

          }

        end

        it { is_expected.not_to contain_class('nftables::ip_nat') }

        it { is_expected.not_to contain_class('nftables::inet_filter') }

        it { is_expected.to have_nftables__config_resource_count(0) }

        it { is_expected.to have_nftables__chain_resource_count(0) }

        it { is_expected.to have_nftables__rule_resource_count(0) }

        it { is_expected.to have_nftables__set_resource_count(0) }

      end

      context 'with with noflush_tables parameter' do

        let(:params) do

          {

            noflush_tables: ['inet-f2b-table'],

          }

        end

        context 'with no nftables fact' do

          it { is_expected.to contain_file('/etc/nftables/puppet-preflight.nft').with_content(%r{^flush ruleset$}) }

        end

        context 'with nftables fact matching' do

          let(:facts) do

            super().merge(nftables: { tables: %w[inet-abc inet-f2b-table] })

          end

          it {

            expect(subject).to contain_file('/etc/nftables/puppet-preflight.nft').

              with_content(%r{^table inet abc \{\}$})

          }

          it {

            expect(subject).to contain_file('/etc/nftables/puppet-preflight.nft').

              with_content(%r{^flush table inet abc$})

          }

        end

        context 'with nftables fact not matching' do

          let(:facts) do

            super().merge(nftables: { tables: %w[inet-abc inet-ijk] })

          end

          it {

            expect(subject).to contain_file('/etc/nftables/puppet-preflight.nft').

              with_content(%r{^table inet abc \{\}$})

          }

          it {

            expect(subject).to contain_file('/etc/nftables/puppet-preflight.nft').

              with_content(%r{^flush table inet abc$})

          }

          it {

            expect(subject).to contain_file('/etc/nftables/puppet-preflight.nft').

              with_content(%r{^table inet ijk \{\}$})

          }

          it {

            expect(subject).to contain_file('/etc/nftables/puppet-preflight.nft').

              with_content(%r{^flush table inet ijk$})

          }

        end

      end

    end

  end

end