influx/telegraf
1
0
Fork 0
Code Issues Activity

Adding upstream version 1.34.4.

Browse source 
Signed-off-by: Daniel Baumann <daniel@debian.org>
This commit is contained in:
Daniel Baumann 2025-05-24 07:26:29 +02:00
parent e393c3af3f
commit 4978089aab
Signed by: daniel
GPG key ID: FBB4F0E80A80222F
4963 changed files with 677545 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

154
plugins/inputs/nginx_upstream_check/nginx_upstream_check_test.go Normal file
View file

@ -0,0 +1,154 @@
package nginx_upstream_check
import (
"fmt"
"net/http"
"net/http/httptest"
"testing"
"github.com/stretchr/testify/require"
"github.com/influxdata/telegraf/testutil"
)
const sampleStatusResponse = `
{
"servers": {
"total": 2,
"generation": 1,
"server": [
{
"index": 0,
"upstream": "upstream-1",
"name": "127.0.0.1:8081",
"status": "up",
"rise": 1000,
"fall": 0,
"type": "http",
"port": 0
},
{
"index": 1,
"upstream": "upstream-2",
"name": "127.0.0.1:8082",
"status": "down",
"rise": 0,
"fall": 2000,
"type": "tcp",
"port": 8080
}
]
}
}`
func TestNginxUpstreamCheckData(test *testing.T) {
testServer := httptest.NewServer(http.HandlerFunc(func(responseWriter http.ResponseWriter, request *http.Request) {
if request.URL.Path != "/status" {
responseWriter.WriteHeader(http.StatusInternalServerError)
test.Errorf("Cannot handle request, expected: %q, actual: %q", "/status", request.URL.Path)
return
}
responseWriter.Header()["Content-Type"] = []string{"application/json"}
if _, err := fmt.Fprintln(responseWriter, sampleStatusResponse); err != nil {
responseWriter.WriteHeader(http.StatusInternalServerError)
test.Error(err)
return
}
}))
defer testServer.Close()
check := newNginxUpstreamCheck()
check.URL = testServer.URL + "/status"
var accumulator testutil.Accumulator
checkError := check.Gather(&accumulator)
require.NoError(test, checkError)
accumulator.AssertContainsTaggedFields(
test,
"nginx_upstream_check",
map[string]interface{}{
"status": "up",
"status_code": uint8(1),
"rise": uint64(1000),
"fall": uint64(0),
},
map[string]string{
"upstream": "upstream-1",
"type": "http",
"name": "127.0.0.1:8081",
"port": "0",
"url": testServer.URL + "/status",
})
accumulator.AssertContainsTaggedFields(
test,
"nginx_upstream_check",
map[string]interface{}{
"status": "down",
"status_code": uint8(2),
"rise": uint64(0),
"fall": uint64(2000),
},
map[string]string{
"upstream": "upstream-2",
"type": "tcp",
"name": "127.0.0.1:8082",
"port": "8080",
"url": testServer.URL + "/status",
})
}
func TestNginxUpstreamCheckRequest(test *testing.T) {
testServer := httptest.NewServer(http.HandlerFunc(func(responseWriter http.ResponseWriter, request *http.Request) {
if request.URL.Path != "/status" {
responseWriter.WriteHeader(http.StatusInternalServerError)
test.Errorf("Cannot handle request, expected: %q, actual: %q", "/status", request.URL.Path)
return
}
responseWriter.Header()["Content-Type"] = []string{"application/json"}
if _, err := fmt.Fprintln(responseWriter, sampleStatusResponse); err != nil {
responseWriter.WriteHeader(http.StatusInternalServerError)
test.Error(err)
return
}
if request.Method != "POST" {
responseWriter.WriteHeader(http.StatusInternalServerError)
test.Errorf("Not equal, expected: %q, actual: %q", "POST", request.Method)
return
}
if request.Header.Get("X-Test") != "test-value" {
responseWriter.WriteHeader(http.StatusInternalServerError)
test.Errorf("Not equal, expected: %q, actual: %q", "test-value", request.Header.Get("X-Test"))
return
}
if request.Header.Get("Authorization") != "Basic dXNlcjpwYXNzd29yZA==" {
responseWriter.WriteHeader(http.StatusInternalServerError)
test.Errorf("Not equal, expected: %q, actual: %q", "Basic dXNlcjpwYXNzd29yZA==", request.Header.Get("Authorization"))
return
}
if request.Host != "status.local" {
responseWriter.WriteHeader(http.StatusInternalServerError)
test.Errorf("Not equal, expected: %q, actual: %q", "status.local", request.Host)
return
}
}))
defer testServer.Close()
check := newNginxUpstreamCheck()
check.URL = testServer.URL + "/status"
check.Headers["X-test"] = "test-value"
check.HostHeader = "status.local"
check.Username = "user"
check.Password = "password"
check.Method = "POST"
var accumulator testutil.Accumulator
checkError := check.Gather(&accumulator)
require.NoError(test, checkError)
}