Use the product token for the user agent in more locations (#7378)

This commit is contained in:
Daniel Nelson
2020-04-21 10:02:18 -07:00
committed by GitHub
parent c9a3b697b8
commit a4eb9c2205
7 changed files with 15 additions and 10 deletions

View File

@@ -129,7 +129,7 @@ func NewHTTPClient(config HTTPConfig) (*httpClient, error) {
userAgent := config.UserAgent
if userAgent == "" {
userAgent = "Telegraf/" + internal.Version()
userAgent = internal.ProductToken()
}
if config.Headers == nil {

View File

@@ -248,8 +248,6 @@ func TestHTTP_Write(t *testing.T) {
u, err := url.Parse(fmt.Sprintf("http://%s", ts.Listener.Addr().String()))
require.NoError(t, err)
internal.SetVersion("1.2.3")
tests := []struct {
name string
config influxdb.HTTPConfig
@@ -310,7 +308,7 @@ func TestHTTP_Write(t *testing.T) {
Log: testutil.Logger{},
},
queryHandlerFunc: func(t *testing.T, w http.ResponseWriter, r *http.Request) {
require.Equal(t, r.Header.Get("User-Agent"), "Telegraf/1.2.3")
require.Equal(t, internal.ProductToken(), r.Header.Get("User-Agent"))
w.WriteHeader(http.StatusNoContent)
},
},