1 files changed, 149 insertions, 0 deletions
diff --git a/src/go/collectors/go.d.plugin/modules/dnsmasq_dhcp/dhcp_test.go b/src/go/collectors/go.d.plugin/modules/dnsmasq_dhcp/dhcp_test.go
new file mode 100644
index 000000000..b83b6a3f5
--- /dev/null
+++ b/src/go/collectors/go.d.plugin/modules/dnsmasq_dhcp/dhcp_test.go
@@ -0,0 +1,149 @@
+// SPDX-License-Identifier: GPL-3.0-or-later
+
+package dnsmasq_dhcp
+
+import (
+	"os"
+	"testing"
+
+	"github.com/netdata/netdata/go/go.d.plugin/agent/module"
+
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+)
+
+var (
+	dataConfigJSON, _ = os.ReadFile("testdata/config.json")
+	dataConfigYAML, _ = os.ReadFile("testdata/config.yaml")
+)
+
+const (
+	testLeasesPath = "testdata/dnsmasq.leases"
+	testConfPath   = "testdata/dnsmasq.conf"
+	testConfDir    = "testdata/dnsmasq.d"
+)
+
+func Test_testDataIsValid(t *testing.T) {
+	for name, data := range map[string][]byte{
+		"dataConfigJSON": dataConfigJSON,
+		"dataConfigYAML": dataConfigYAML,
+	} {
+		require.NotNil(t, data, name)
+	}
+}
+
+func TestDnsmasqDHCP_ConfigurationSerialize(t *testing.T) {
+	module.TestConfigurationSerialize(t, &DnsmasqDHCP{}, dataConfigJSON, dataConfigYAML)
+}
+
+func TestDnsmasqDHCP_Init(t *testing.T) {
+	job := New()
+	job.LeasesPath = testLeasesPath
+	job.ConfPath = testConfPath
+	job.ConfDir = testConfDir
+
+	assert.NoError(t, job.Init())
+}
+
+func TestDnsmasqDHCP_InitEmptyLeasesPath(t *testing.T) {
+	job := New()
+	job.LeasesPath = ""
+
+	assert.Error(t, job.Init())
+}
+
+func TestDnsmasqDHCP_InitInvalidLeasesPath(t *testing.T) {
+	job := New()
+	job.LeasesPath = testLeasesPath
+	job.LeasesPath += "!"
+
+	assert.Error(t, job.Init())
+}
+
+func TestDnsmasqDHCP_InitZeroDHCPRanges(t *testing.T) {
+	job := New()
+	job.LeasesPath = testLeasesPath
+	job.ConfPath = "testdata/dnsmasq3.conf"
+	job.ConfDir = ""
+
+	assert.NoError(t, job.Init())
+}
+
+func TestDnsmasqDHCP_Check(t *testing.T) {
+	job := New()
+	job.LeasesPath = testLeasesPath
+	job.ConfPath = testConfPath
+	job.ConfDir = testConfDir
+
+	require.NoError(t, job.Init())
+	assert.NoError(t, job.Check())
+}
+
+func TestDnsmasqDHCP_Charts(t *testing.T) {
+	job := New()
+	job.LeasesPath = testLeasesPath
+	job.ConfPath = testConfPath
+	job.ConfDir = testConfDir
+
+	require.NoError(t, job.Init())
+
+	assert.NotNil(t, job.Charts())
+}
+
+func TestDnsmasqDHCP_Cleanup(t *testing.T) {
+	assert.NotPanics(t, New().Cleanup)
+}
+
+func TestDnsmasqDHCP_Collect(t *testing.T) {
+	job := New()
+	job.LeasesPath = testLeasesPath
+	job.ConfPath = testConfPath
+	job.ConfDir = testConfDir
+
+	require.NoError(t, job.Init())
+	require.NoError(t, job.Check())
+
+	expected := map[string]int64{
+		"dhcp_range_1230::1-1230::64_allocated_leases":              7,
+		"dhcp_range_1230::1-1230::64_utilization":                   7,
+		"dhcp_range_1231::1-1231::64_allocated_leases":              1,
+		"dhcp_range_1231::1-1231::64_utilization":                   1,
+		"dhcp_range_1232::1-1232::64_allocated_leases":              1,
+		"dhcp_range_1232::1-1232::64_utilization":                   1,
+		"dhcp_range_1233::1-1233::64_allocated_leases":              1,
+		"dhcp_range_1233::1-1233::64_utilization":                   1,
+		"dhcp_range_1234::1-1234::64_allocated_leases":              1,
+		"dhcp_range_1234::1-1234::64_utilization":                   1,
+		"dhcp_range_192.168.0.1-192.168.0.100_allocated_leases":     6,
+		"dhcp_range_192.168.0.1-192.168.0.100_utilization":          6,
+		"dhcp_range_192.168.1.1-192.168.1.100_allocated_leases":     5,
+		"dhcp_range_192.168.1.1-192.168.1.100_utilization":          5,
+		"dhcp_range_192.168.2.1-192.168.2.100_allocated_leases":     4,
+		"dhcp_range_192.168.2.1-192.168.2.100_utilization":          4,
+		"dhcp_range_192.168.200.1-192.168.200.100_allocated_leases": 1,
+		"dhcp_range_192.168.200.1-192.168.200.100_utilization":      1,
+		"dhcp_range_192.168.3.1-192.168.3.100_allocated_leases":     1,
+		"dhcp_range_192.168.3.1-192.168.3.100_utilization":          1,
+		"dhcp_range_192.168.4.1-192.168.4.100_allocated_leases":     1,
+		"dhcp_range_192.168.4.1-192.168.4.100_utilization":          1,
+		"ipv4_dhcp_hosts":  6,
+		"ipv4_dhcp_ranges": 6,
+		"ipv6_dhcp_hosts":  5,
+		"ipv6_dhcp_ranges": 5,
+	}
+
+	assert.Equal(t, expected, job.Collect())
+}
+
+func TestDnsmasqDHCP_CollectFailedToOpenLeasesPath(t *testing.T) {
+	job := New()
+	job.LeasesPath = testLeasesPath
+	job.ConfPath = testConfPath
+	job.ConfDir = testConfDir
+
+	require.NoError(t, job.Init())
+	require.NoError(t, job.Check())
+
+	job.LeasesPath = ""
+	assert.Nil(t, job.Collect())
+}