package influxdb_test import ( "context" "net/http" "testing" "time" "github.com/influxdata/telegraf" "github.com/influxdata/telegraf/internal" "github.com/influxdata/telegraf/internal/tls" "github.com/influxdata/telegraf/metric" "github.com/influxdata/telegraf/plugins/outputs/influxdb" "github.com/stretchr/testify/require" ) type MockClient struct { URLF func() string WriteF func(context.Context, []telegraf.Metric) error CreateDatabaseF func(ctx context.Context, database string) error DatabaseF func() string CloseF func() } func (c *MockClient) URL() string { return c.URLF() } func (c *MockClient) Write(ctx context.Context, metrics []telegraf.Metric) error { return c.WriteF(ctx, metrics) } func (c *MockClient) CreateDatabase(ctx context.Context, database string) error { return c.CreateDatabaseF(ctx, database) } func (c *MockClient) Database() string { return c.DatabaseF() } func (c *MockClient) Close() { c.CloseF() } func TestDeprecatedURLSupport(t *testing.T) { var actual *influxdb.UDPConfig output := influxdb.InfluxDB{ URL: "udp://localhost:8089", CreateUDPClientF: func(config *influxdb.UDPConfig) (influxdb.Client, error) { actual = config return &MockClient{}, nil }, } err := output.Connect() require.NoError(t, err) require.Equal(t, "udp://localhost:8089", actual.URL.String()) } func TestDefaultURL(t *testing.T) { var actual *influxdb.HTTPConfig output := influxdb.InfluxDB{ CreateHTTPClientF: func(config *influxdb.HTTPConfig) (influxdb.Client, error) { actual = config return &MockClient{ DatabaseF: func() string { return "telegraf" }, CreateDatabaseF: func(ctx context.Context, database string) error { return nil }, }, nil }, } err := output.Connect() require.NoError(t, err) require.Equal(t, "http://localhost:8086", actual.URL.String()) } func TestConnectUDPConfig(t *testing.T) { var actual *influxdb.UDPConfig output := influxdb.InfluxDB{ URLs: []string{"udp://localhost:8089"}, UDPPayload: internal.Size{Size: 42}, CreateUDPClientF: func(config *influxdb.UDPConfig) (influxdb.Client, error) { actual = config return &MockClient{}, nil }, } err := output.Connect() require.NoError(t, err) require.Equal(t, "udp://localhost:8089", actual.URL.String()) require.Equal(t, 42, actual.MaxPayloadSize) require.NotNil(t, actual.Serializer) } func TestConnectHTTPConfig(t *testing.T) { var actual *influxdb.HTTPConfig output := influxdb.InfluxDB{ URLs: []string{"http://localhost:8086"}, Database: "telegraf", RetentionPolicy: "default", WriteConsistency: "any", Timeout: internal.Duration{Duration: 5 * time.Second}, Username: "guy", Password: "smiley", UserAgent: "telegraf", HTTPProxy: "http://localhost:8086", HTTPHeaders: map[string]string{ "x": "y", }, ContentEncoding: "gzip", ClientConfig: tls.ClientConfig{ InsecureSkipVerify: true, }, CreateHTTPClientF: func(config *influxdb.HTTPConfig) (influxdb.Client, error) { actual = config return &MockClient{ DatabaseF: func() string { return "telegraf" }, CreateDatabaseF: func(ctx context.Context, database string) error { return nil }, }, nil }, } err := output.Connect() require.NoError(t, err) require.Equal(t, output.URLs[0], actual.URL.String()) require.Equal(t, output.UserAgent, actual.UserAgent) require.Equal(t, output.Timeout.Duration, actual.Timeout) require.Equal(t, output.Username, actual.Username) require.Equal(t, output.Password, actual.Password) require.Equal(t, output.HTTPProxy, actual.Proxy.String()) require.Equal(t, output.HTTPHeaders, actual.Headers) require.Equal(t, output.ContentEncoding, actual.ContentEncoding) require.Equal(t, output.Database, actual.Database) require.Equal(t, output.RetentionPolicy, actual.RetentionPolicy) require.Equal(t, output.WriteConsistency, actual.Consistency) require.NotNil(t, actual.TLSConfig) require.NotNil(t, actual.Serializer) require.Equal(t, output.Database, actual.Database) } func TestWriteRecreateDatabaseIfDatabaseNotFound(t *testing.T) { output := influxdb.InfluxDB{ URLs: []string{"http://localhost:8086"}, CreateHTTPClientF: func(config *influxdb.HTTPConfig) (influxdb.Client, error) { return &MockClient{ DatabaseF: func() string { return "telegraf" }, CreateDatabaseF: func(ctx context.Context, database string) error { return nil }, WriteF: func(ctx context.Context, metrics []telegraf.Metric) error { return &influxdb.DatabaseNotFoundError{ APIError: influxdb.APIError{ StatusCode: http.StatusNotFound, Title: "404 Not Found", Description: `database not found "telegraf"`, }, } }, URLF: func() string { return "http://localhost:8086" }, }, nil }, } err := output.Connect() require.NoError(t, err) m, err := metric.New( "cpu", map[string]string{}, map[string]interface{}{ "value": 42.0, }, time.Unix(0, 0), ) require.NoError(t, err) metrics := []telegraf.Metric{m} err = output.Write(metrics) // We only have one URL, so we expect an error require.Error(t, err) }