diff options
Diffstat (limited to 'src/go/collectors/go.d.plugin/modules/fluentd/fluentd_test.go')
-rw-r--r-- | src/go/collectors/go.d.plugin/modules/fluentd/fluentd_test.go | 127 |
1 files changed, 127 insertions, 0 deletions
diff --git a/src/go/collectors/go.d.plugin/modules/fluentd/fluentd_test.go b/src/go/collectors/go.d.plugin/modules/fluentd/fluentd_test.go new file mode 100644 index 000000000..01c4f9636 --- /dev/null +++ b/src/go/collectors/go.d.plugin/modules/fluentd/fluentd_test.go @@ -0,0 +1,127 @@ +// SPDX-License-Identifier: GPL-3.0-or-later + +package fluentd + +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") + + dataPluginsMetrics, _ = os.ReadFile("testdata/plugins.json") +) + +func Test_testDataIsValid(t *testing.T) { + for name, data := range map[string][]byte{ + "dataConfigJSON": dataConfigJSON, + "dataConfigYAML": dataConfigYAML, + "dataPluginsMetrics": dataPluginsMetrics, + } { + require.NotNil(t, data, name) + } +} + +func TestFluentd_ConfigurationSerialize(t *testing.T) { + module.TestConfigurationSerialize(t, &Fluentd{}, dataConfigJSON, dataConfigYAML) +} + +func TestFluentd_Init(t *testing.T) { + // OK + job := New() + assert.NoError(t, job.Init()) + assert.NotNil(t, job.apiClient) + + //NG + job = New() + job.URL = "" + assert.Error(t, job.Init()) +} + +func TestFluentd_Check(t *testing.T) { + ts := httptest.NewServer(http.HandlerFunc( + func(w http.ResponseWriter, r *http.Request) { + _, _ = w.Write(dataPluginsMetrics) + })) + defer ts.Close() + + // OK + job := New() + job.URL = ts.URL + require.NoError(t, job.Init()) + require.NoError(t, job.Check()) + + // NG + job = New() + job.URL = "http://127.0.0.1:38001/api/plugins.json" + require.NoError(t, job.Init()) + require.Error(t, job.Check()) +} + +func TestFluentd_Charts(t *testing.T) { + assert.NotNil(t, New().Charts()) +} + +func TestFluentd_Cleanup(t *testing.T) { + New().Cleanup() +} + +func TestFluentd_Collect(t *testing.T) { + ts := httptest.NewServer(http.HandlerFunc( + func(w http.ResponseWriter, r *http.Request) { + _, _ = w.Write(dataPluginsMetrics) + })) + defer ts.Close() + + job := New() + job.URL = ts.URL + + require.NoError(t, job.Init()) + require.NoError(t, job.Check()) + + expected := map[string]int64{ + "output_stdout_stdout_output_retry_count": 0, + "output_td_tdlog_output_retry_count": 0, + "output_td_tdlog_output_buffer_queue_length": 0, + "output_td_tdlog_output_buffer_total_queued_size": 0, + } + assert.Equal(t, expected, job.Collect()) + assert.Len(t, job.charts.Get("retry_count").Dims, 2) + assert.Len(t, job.charts.Get("buffer_queue_length").Dims, 1) + assert.Len(t, job.charts.Get("buffer_total_queued_size").Dims, 1) +} + +func TestFluentd_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 TestFluentd_404(t *testing.T) { + ts := httptest.NewServer(http.HandlerFunc( + func(w http.ResponseWriter, r *http.Request) { + w.WriteHeader(404) + })) + defer ts.Close() + + job := New() + job.URL = ts.URL + require.NoError(t, job.Init()) + assert.Error(t, job.Check()) +} |