Bootstrap agent2 support

* Remove some parameters in zabbix agent configuration if it's zabbix agent2 package * LogRemoteCommands * StartAgents * MaxLinesPerSecond * AllowRoot * User * LoadModulePath * Fix EnableRemoteCommands still use on zabbix agentd > 5 but remove on agent2 Conflicts: manifests/agent.pp spec/classes/agent_spec.rb spec/acceptance/agent_spec.rb templates/zabbix_agentd.conf.erb
voxpupuli · Mar 11, 2024 · 93c7965 · 93c7965
1 parent 53f5969
commit 93c7965
Show file tree

Hide file tree

Showing 7 changed files with 220 additions and 38 deletions.
diff --git a/README.md b/README.md
@@ -22,6 +22,7 @@
 5. [Usage - Configuration options and additional functionality](#usage)
     * [zabbix-server](#usage-zabbix-server)
     * [zabbix-agent](#usage-zabbix-agent)
+    * [zabbix-agent2](#usage-zabbix-agent2)
     * [zabbix-proxy](#usage-zabbix-proxy)
     * [zabbix-javagateway](#usage-zabbix-javagateway)
     * [zabbix-sender](#usage-zabbix-sender)
@@ -200,6 +201,21 @@ class { 'zabbix::agent':
 }
 ```
 
+### Usage zabbix-agent2
+
+Basic one way of setup, wheter it is monitored by zabbix-server or zabbix-proxy:
+```ruby
+class { 'zabbix::agent':
+  agent_configfile_path => '/etc/zabbix/zabbix_agent2.conf',
+  include_dir           => '/etc/zabbix/zabbix_agent2.d',
+  include_dir_purge     => false,
+  zabbix_package_agent  => 'zabbix-agent2',
+  servicename           => 'zabbix-agent2',
+  manage_startup_script => false,
+  server                => '192.168.20.11',
+}
+```
+
 ### Usage zabbix-proxy
 
 Like the zabbix-server, the zabbix-proxy can also be used in 2 ways:

diff --git a/REFERENCE.md b/REFERENCE.md
@@ -1342,6 +1342,19 @@ class { 'zabbix::agent':
 }
 ```
 
+##### Using Zabbix Agent 2
+
+```puppet
+class { 'zabbix::agent':
+  agent_configfile_path => '/etc/zabbix/zabbix_agent2.conf',
+  include_dir           => '/etc/zabbix/zabbix_agent2.d',
+  include_dir_purge     => false,
+  zabbix_package_agent  => 'zabbix-agent2',
+  servicename           => 'zabbix-agent2',
+  manage_startup_script => false,
+}
+```
+
 #### Parameters
 
 The following parameters are available in the `zabbix::agent` class:

diff --git a/manifests/agent.pp b/manifests/agent.pp
@@ -134,6 +134,17 @@
 #    monitored_by_proxy => 'my_proxy_host',
 #    server             => '192.168.1.1',
 #  }
+#
+# @example Using Zabbix Agent 2
+#   class { 'zabbix::agent':
+#     agent_configfile_path => '/etc/zabbix/zabbix_agent2.conf',
+#     include_dir           => '/etc/zabbix/zabbix_agent2.d',
+#     include_dir_purge     => false,
+#     zabbix_package_agent  => 'zabbix-agent2',
+#     servicename           => 'zabbix-agent2',
+#     manage_startup_script => false,
+#   }
+#
 # @author Werner Dijkerman [email protected]
 class zabbix::agent (
   $zabbix_version                                      = $zabbix::params::zabbix_version,
@@ -219,6 +230,8 @@
   String $service_type                                 = $zabbix::params::service_type,
   Boolean $manage_startup_script                       = $zabbix::params::manage_startup_script,
 ) inherits zabbix::params {
+  $agent2 = $zabbix_package_agent == 'zabbix-agent2'
+
   # Find if listenip is set. If not, we can set to specific ip or
   # to network name. If more than 1 interfaces are available, we
   # can find the ipaddress of this specific interface if listenip

diff --git a/spec/acceptance/agent_spec.rb b/spec/acceptance/agent_spec.rb
@@ -7,34 +7,36 @@
   next if version < '6.0' && default[:platform] =~ %r{debian-12}
 
   describe "zabbix::agent class with zabbix_version #{version}" do
-    it 'works idempotently with no errors' do
-      pp = <<-EOS
-      class { 'zabbix::agent':
-        server               => '192.168.20.11',
-        zabbix_package_state => 'latest',
-        zabbix_version       => '#{version}',
-      }
-      EOS
-
-      prepare_host
-
-      # Run it twice and test for idempotency
-      apply_manifest(pp, catch_failures: true)
-      apply_manifest(pp, catch_changes: true)
-    end
+    context 'With minimal parameter' do
+      it 'works idempotently with no errors' do
+        pp = <<-EOS
+        class { 'zabbix::agent':
+          server               => '192.168.20.11',
+          zabbix_package_state => 'latest',
+          zabbix_version       => '#{version}',
+        }
+        EOS
 
-    # do some basic checks
-    describe package('zabbix-agent') do
-      it { is_expected.to be_installed }
-    end
+        prepare_host
 
-    describe service('zabbix-agent') do
-      it { is_expected.to be_running }
-      it { is_expected.to be_enabled }
-    end
+        # Run it twice and test for idempotency
+        apply_manifest(pp, catch_failures: true)
+        apply_manifest(pp, catch_changes: true)
+      end
+
+      # do some basic checks
+      describe package('zabbix-agent') do
+        it { is_expected.to be_installed }
+      end
+
+      describe service('zabbix-agent') do
+        it { is_expected.to be_running }
+        it { is_expected.to be_enabled }
+      end
 
-    describe file('/etc/zabbix/zabbix_agentd.conf') do
-      its(:content) { is_expected.not_to match %r{ListenIP=} }
+      describe file('/etc/zabbix/zabbix_agentd.conf') do
+        its(:content) { is_expected.not_to match %r{ListenIP=} }
+      end
     end
 
     context 'With ListenIP set to an IP-Address' do
@@ -90,4 +92,101 @@ class { 'zabbix::agent':
       end
     end
   end
+
+  describe "zabbix::agent class with agent2 and zabbix_version #{version}" do
+    # <6.0 agent2 packages are not available for ubuntu 22.04
+    next if version < '6.0' && default[:platform] =~ %r{ubuntu-22}
+
+    before(:all) do
+      pp = <<-PUPPET
+        service {'zabbix-agent': ensure => stopped }
+        -> package {'zabbix-agent': ensure => purged}
+      PUPPET
+      apply_manifest(pp)
+    end
+
+    context 'With minimal parameter' do
+      it_behaves_like 'an idempotent resource' do
+        let(:manifest) do
+          <<-PUPPET
+          class { 'zabbix::agent':
+            agent_configfile_path => '/etc/zabbix/zabbix_agent2.conf',
+            include_dir           => '/etc/zabbix/zabbix_agent2.d',
+            include_dir_purge     => false,
+            zabbix_package_agent  => 'zabbix-agent2',
+            servicename           => 'zabbix-agent2',
+            manage_startup_script => false,
+            server                => '192.168.20.11',
+            zabbix_package_state  => 'latest',
+            zabbix_version        => '#{version}',
+          }
+          PUPPET
+        end
+      end
+
+      # do some basic checks
+      describe package('zabbix-agent2') do
+        it { is_expected.to be_installed }
+      end
+
+      describe service('zabbix-agent2') do
+        it { is_expected.to be_running }
+        it { is_expected.to be_enabled }
+      end
+
+      describe file('/etc/zabbix/zabbix_agentd2.conf') do
+        its(:content) { is_expected.not_to match %r{ListenIP=} }
+      end
+    end
+
+    context 'With ListenIP set to an IP-Address' do
+      it_behaves_like 'an idempotent resource' do
+        let(:manifest) do
+          <<-PUPPET
+          class { 'zabbix::agent':
+            agent_configfile_path => '/etc/zabbix/zabbix_agent2.conf',
+            include_dir           => '/etc/zabbix/zabbix_agent2.d',
+            include_dir_purge     => false,
+            zabbix_package_agent  => 'zabbix-agent2',
+            servicename           => 'zabbix-agent2',
+            manage_startup_script => false,
+            server                => '192.168.20.11',
+            zabbix_package_state  => 'latest',
+            listenip              => '127.0.0.1',
+            zabbix_version        => '#{version}',
+          }
+          PUPPET
+        end
+      end
+
+      describe file('/etc/zabbix/zabbix_agent2.conf') do
+        its(:content) { is_expected.to match %r{ListenIP=127.0.0.1} }
+      end
+    end
+
+    context 'With ListenIP set to lo' do
+      it_behaves_like 'an idempotent resource' do
+        let(:manifest) do
+          <<-PUPPET
+          class { 'zabbix::agent':
+            agent_configfile_path => '/etc/zabbix/zabbix_agent2.conf',
+            include_dir           => '/etc/zabbix/zabbix_agent2.d',
+            include_dir_purge     => false,
+            zabbix_package_agent  => 'zabbix-agent2',
+            servicename           => 'zabbix-agent2',
+            manage_startup_script => false,
+            server                => '192.168.20.11',
+            zabbix_package_state  => 'latest',
+            listenip              => 'lo',
+            zabbix_version        => '#{version}',
+          }
+          PUPPET
+        end
+
+        describe file('/etc/zabbix/zabbix_agent2.conf') do
+          its(:content) { is_expected.to match %r{ListenIP=127.0.0.1} }
+        end
+      end
+    end
+  end
 end
diff --git a/spec/classes/agent_spec.rb b/spec/classes/agent_spec.rb
@@ -445,6 +445,28 @@
           it { is_expected.to contain_systemd__unit_file('zabbix-agent.service') }
         end
       end
+
+      context 'when zabbix_package_agent is zabbix-agent2' do
+        let :params do
+          {
+            zabbix_package_agent: 'zabbix-agent2', startagents: 1,
+            maxlinespersecond: 1, allowroot: 1, zabbix_user: 'root',
+            loadmodulepath: '/tmp', allowkey: 'system.run[*]',
+            denykey: 'system.run[*]', enableremotecommands: 1,
+            logremotecommands: 1
+          }
+        end
+
+        it { is_expected.to contain_package('zabbix-agent2') }
+
+        it do
+          is_expected.not_to contain_file(config_path).with_content(
+            %r{^(LogRemoteCommands|StartAgents|MaxLinesPerSecond
+                 |AllowRoot|User|LoadModulePath|AllowKey|DenyKey|
+                 EnableRemoteCommands|LogRemoteCommands)}
+          )
+        end
+      end
     end
   end
 end
diff --git a/spec/support/acceptance/prepare_host.rb b/spec/support/acceptance/prepare_host.rb
@@ -5,21 +5,27 @@ def prepare_host
 
   apply_manifest <<~PUPPET
     $services = $facts['os']['family'] ? {
-      'RedHat' => ['zabbix-server', 'httpd'],
-      'Debian' => ['zabbix-server', 'apache2'],
-      default  => [],
+      'RedHat' => ['zabbix-server', 'httpd', 'zabbix-agentd', 'zabbix-agent', 'zabbix-agent2'],
+      'Debian' => ['zabbix-server', 'apache2', 'zabbix-agentd', 'zabbix-agent', 'zabbix-agent2'],
+      default  => ['zabbix-agentd', 'zabbix-agent', 'zabbix-agent2'],
     }
     service { $services:
       ensure => stopped
     }
 
     $packages = $facts['os']['family'] ? {
-      'RedHat' => ['zabbix-server-pgsql', 'zabbix-server-pgsql-scl', 'zabbix-web', 'zabbix-web-pgsql', 'zabbix-web-pgsql-scl', 'zabbix-frontend-php', 'zabbix-sql-scripts'],
-      'Debian' => ['zabbix-server-pgsql', 'zabbix-web-pgsql', 'zabbix-frontend-php', 'zabbix-sql-scripts'],
-      default  => [],
+      'RedHat' => ['zabbix-server-pgsql', 'zabbix-server-pgsql-scl', 'zabbix-web', 'zabbix-web-pgsql', 'zabbix-web-pgsql-scl', 'zabbix-frontend-php', 'zabbix-sql-scripts', 'zabbix-agent', 'zabbix-agent2'],
+      'Debian' => ['zabbix-server-pgsql', 'zabbix-web-pgsql', 'zabbix-frontend-php', 'zabbix-sql-scripts', 'zabbix-agent', 'zabbix-agent2'],
+      default  => ['zabbix-agent', 'zabbix-agent2'],
     }
+
+    $pkg_ensure = $facts['os']['family'] ? {
+      'Archlinux' => absent,
+      default     => purged,
+    }
+
     package { $packages:
-      ensure => purged
+      ensure => $pkg_ensure
     }
   PUPPET