1 files changed, 147 insertions, 0 deletions
diff --git a/src/go/collectors/go.d.plugin/modules/tengine/tengine_test.go b/src/go/collectors/go.d.plugin/modules/tengine/tengine_test.go
new file mode 100644
index 000000000..d8b8ec997
--- /dev/null
+++ b/src/go/collectors/go.d.plugin/modules/tengine/tengine_test.go
@@ -0,0 +1,147 @@
+// SPDX-License-Identifier: GPL-3.0-or-later
+
+package tengine
+
+import (
+	"net/http"
+	"net/http/httptest"
+	"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")
+
+	dataStatusMetrics, _ = os.ReadFile("testdata/status.txt")
+)
+
+func Test_testDataIsValid(t *testing.T) {
+	for name, data := range map[string][]byte{
+		"dataConfigJSON":    dataConfigJSON,
+		"dataConfigYAML":    dataConfigYAML,
+		"dataStatusMetrics": dataStatusMetrics,
+	} {
+		require.NotNil(t, data, name)
+	}
+}
+
+func TestTengine_ConfigurationSerialize(t *testing.T) {
+	module.TestConfigurationSerialize(t, &Tengine{}, dataConfigJSON, dataConfigYAML)
+}
+
+func TestTengine_Cleanup(t *testing.T) {
+	New().Cleanup()
+}
+
+func TestTengine_Init(t *testing.T) {
+	job := New()
+
+	require.NoError(t, job.Init())
+	assert.NotNil(t, job.apiClient)
+}
+
+func TestTengine_Check(t *testing.T) {
+	ts := httptest.NewServer(
+		http.HandlerFunc(
+			func(w http.ResponseWriter, r *http.Request) {
+				_, _ = w.Write(dataStatusMetrics)
+			}))
+	defer ts.Close()
+
+	job := New()
+	job.URL = ts.URL
+	require.NoError(t, job.Init())
+	assert.NoError(t, job.Check())
+}
+
+func TestTengine_CheckNG(t *testing.T) {
+	job := New()
+
+	job.URL = "http://127.0.0.1:38001/us"
+	require.NoError(t, job.Init())
+	assert.Error(t, job.Check())
+}
+
+func TestTengine_Charts(t *testing.T) { assert.NotNil(t, New().Charts()) }
+
+func TestTengine_Collect(t *testing.T) {
+	ts := httptest.NewServer(
+		http.HandlerFunc(
+			func(w http.ResponseWriter, r *http.Request) {
+				_, _ = w.Write(dataStatusMetrics)
+			}))
+	defer ts.Close()
+
+	job := New()
+	job.URL = ts.URL
+	require.NoError(t, job.Init())
+	require.NoError(t, job.Check())
+
+	expected := map[string]int64{
+		"bytes_in":                 5944,
+		"bytes_out":                20483,
+		"conn_total":               354,
+		"http_200":                 1536,
+		"http_206":                 0,
+		"http_2xx":                 1536,
+		"http_302":                 43,
+		"http_304":                 0,
+		"http_3xx":                 50,
+		"http_403":                 1,
+		"http_404":                 75,
+		"http_416":                 0,
+		"http_499":                 0,
+		"http_4xx":                 80,
+		"http_500":                 0,
+		"http_502":                 1,
+		"http_503":                 0,
+		"http_504":                 0,
+		"http_508":                 0,
+		"http_5xx":                 1,
+		"http_other_detail_status": 11,
+		"http_other_status":        0,
+		"http_ups_4xx":             26,
+		"http_ups_5xx":             1,
+		"req_total":                1672,
+		"rt":                       1339,
+		"ups_req":                  268,
+		"ups_rt":                   644,
+		"ups_tries":                268,
+	}
+
+	assert.Equal(t, expected, job.Collect())
+}
+
+func TestTengine_InvalidData(t *testing.T) {
+	ts := httptest.NewServer(
+		http.HandlerFunc(
+			func(w http.ResponseWriter, r *http.Request) {
+				_, _ = w.Write([]byte("hello and goodbye"))
+			}))
+	defer ts.Close()
+
+	job := New()
+	job.URL = ts.URL
+	require.NoError(t, job.Init())
+	assert.Error(t, job.Check())
+}
+
+func TestTengine_404(t *testing.T) {
+	ts := httptest.NewServer(
+		http.HandlerFunc(
+			func(w http.ResponseWriter, r *http.Request) {
+				w.WriteHeader(http.StatusNotFound)
+			}))
+	defer ts.Close()
+
+	job := New()
+	job.URL = ts.URL
+	require.NoError(t, job.Init())
+	assert.Error(t, job.Check())
+}