269 lines
7.3 KiB
Go
269 lines
7.3 KiB
Go
package jolokia
|
|
|
|
import (
|
|
_ "fmt"
|
|
"io/ioutil"
|
|
"net/http"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/influxdata/telegraf/testutil"
|
|
"github.com/stretchr/testify/assert"
|
|
_ "github.com/stretchr/testify/require"
|
|
)
|
|
|
|
const validThreeLevelMultiValueJSON = `
|
|
[
|
|
{
|
|
"request":{
|
|
"mbean":"java.lang:type=*",
|
|
"type":"read"
|
|
},
|
|
"value":{
|
|
"java.lang:type=Memory":{
|
|
"ObjectPendingFinalizationCount":0,
|
|
"Verbose":false,
|
|
"HeapMemoryUsage":{
|
|
"init":134217728,
|
|
"committed":173015040,
|
|
"max":1908932608,
|
|
"used":16840016
|
|
},
|
|
"NonHeapMemoryUsage":{
|
|
"init":2555904,
|
|
"committed":51380224,
|
|
"max":-1,
|
|
"used":49944048
|
|
},
|
|
"ObjectName":{
|
|
"objectName":"java.lang:type=Memory"
|
|
}
|
|
}
|
|
},
|
|
"timestamp":1446129191,
|
|
"status":200
|
|
}
|
|
]`
|
|
|
|
const validBulkResponseJSON = `
|
|
[
|
|
{
|
|
"request":{
|
|
"mbean":"java.lang:type=Memory",
|
|
"attribute":"HeapMemoryUsage",
|
|
"type":"read"
|
|
},
|
|
"value":{
|
|
"init":67108864,
|
|
"committed":456130560,
|
|
"max":477626368,
|
|
"used":203288528
|
|
},
|
|
"timestamp":1446129191,
|
|
"status":200
|
|
},
|
|
{
|
|
"request":{
|
|
"mbean":"java.lang:type=Memory",
|
|
"attribute":"NonHeapMemoryUsage",
|
|
"type":"read"
|
|
},
|
|
"value":{
|
|
"init":2555904,
|
|
"committed":51380224,
|
|
"max":-1,
|
|
"used":49944048
|
|
},
|
|
"timestamp":1446129191,
|
|
"status":200
|
|
}
|
|
]`
|
|
|
|
const validMultiValueJSON = `
|
|
[
|
|
{
|
|
"request":{
|
|
"mbean":"java.lang:type=Memory",
|
|
"attribute":"HeapMemoryUsage",
|
|
"type":"read"
|
|
},
|
|
"value":{
|
|
"init":67108864,
|
|
"committed":456130560,
|
|
"max":477626368,
|
|
"used":203288528
|
|
},
|
|
"timestamp":1446129191,
|
|
"status":200
|
|
}
|
|
]`
|
|
|
|
const validSingleValueJSON = `
|
|
[
|
|
{
|
|
"request":{
|
|
"path":"used",
|
|
"mbean":"java.lang:type=Memory",
|
|
"attribute":"HeapMemoryUsage",
|
|
"type":"read"
|
|
},
|
|
"value":209274376,
|
|
"timestamp":1446129256,
|
|
"status":200
|
|
}
|
|
]`
|
|
|
|
const invalidJSON = "I don't think this is JSON"
|
|
|
|
const empty = ""
|
|
|
|
var Servers = []Server{{Name: "as1", Host: "127.0.0.1", Port: "8080"}}
|
|
var HeapMetric = Metric{Name: "heap_memory_usage",
|
|
Mbean: "java.lang:type=Memory", Attribute: "HeapMemoryUsage"}
|
|
var UsedHeapMetric = Metric{Name: "heap_memory_usage",
|
|
Mbean: "java.lang:type=Memory", Attribute: "HeapMemoryUsage"}
|
|
var NonHeapMetric = Metric{Name: "non_heap_memory_usage",
|
|
Mbean: "java.lang:type=Memory", Attribute: "NonHeapMemoryUsage"}
|
|
|
|
type jolokiaClientStub struct {
|
|
responseBody string
|
|
statusCode int
|
|
}
|
|
|
|
func (c jolokiaClientStub) MakeRequest(req *http.Request) (*http.Response, error) {
|
|
resp := http.Response{}
|
|
resp.StatusCode = c.statusCode
|
|
resp.Body = ioutil.NopCloser(strings.NewReader(c.responseBody))
|
|
return &resp, nil
|
|
}
|
|
|
|
// Generates a pointer to an HttpJson object that uses a mock HTTP client.
|
|
// Parameters:
|
|
// response : Body of the response that the mock HTTP client should return
|
|
// statusCode: HTTP status code the mock HTTP client should return
|
|
//
|
|
// Returns:
|
|
// *HttpJson: Pointer to an HttpJson object that uses the generated mock HTTP client
|
|
func genJolokiaClientStub(response string, statusCode int, servers []Server, metrics []Metric) *Jolokia {
|
|
return &Jolokia{
|
|
jClient: jolokiaClientStub{responseBody: response, statusCode: statusCode},
|
|
Servers: servers,
|
|
Metrics: metrics,
|
|
Delimiter: "_",
|
|
}
|
|
}
|
|
|
|
// Test that the proper values are ignored or collected
|
|
func TestHttpJsonMultiValue(t *testing.T) {
|
|
jolokia := genJolokiaClientStub(validMultiValueJSON, 200, Servers, []Metric{HeapMetric})
|
|
|
|
var acc testutil.Accumulator
|
|
err := acc.GatherError(jolokia.Gather)
|
|
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, 1, len(acc.Metrics))
|
|
|
|
fields := map[string]interface{}{
|
|
"heap_memory_usage_init": 67108864.0,
|
|
"heap_memory_usage_committed": 456130560.0,
|
|
"heap_memory_usage_max": 477626368.0,
|
|
"heap_memory_usage_used": 203288528.0,
|
|
}
|
|
tags := map[string]string{
|
|
"jolokia_host": "127.0.0.1",
|
|
"jolokia_port": "8080",
|
|
"jolokia_name": "as1",
|
|
}
|
|
acc.AssertContainsTaggedFields(t, "jolokia", fields, tags)
|
|
}
|
|
|
|
// Test that bulk responses are handled
|
|
func TestHttpJsonBulkResponse(t *testing.T) {
|
|
jolokia := genJolokiaClientStub(validBulkResponseJSON, 200, Servers, []Metric{HeapMetric, NonHeapMetric})
|
|
|
|
var acc testutil.Accumulator
|
|
err := jolokia.Gather(&acc)
|
|
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, 1, len(acc.Metrics))
|
|
|
|
fields := map[string]interface{}{
|
|
"heap_memory_usage_init": 67108864.0,
|
|
"heap_memory_usage_committed": 456130560.0,
|
|
"heap_memory_usage_max": 477626368.0,
|
|
"heap_memory_usage_used": 203288528.0,
|
|
"non_heap_memory_usage_init": 2555904.0,
|
|
"non_heap_memory_usage_committed": 51380224.0,
|
|
"non_heap_memory_usage_max": -1.0,
|
|
"non_heap_memory_usage_used": 49944048.0,
|
|
}
|
|
tags := map[string]string{
|
|
"jolokia_host": "127.0.0.1",
|
|
"jolokia_port": "8080",
|
|
"jolokia_name": "as1",
|
|
}
|
|
acc.AssertContainsTaggedFields(t, "jolokia", fields, tags)
|
|
}
|
|
|
|
// Test that the proper values are ignored or collected
|
|
func TestHttpJsonThreeLevelMultiValue(t *testing.T) {
|
|
jolokia := genJolokiaClientStub(validThreeLevelMultiValueJSON, 200, Servers, []Metric{HeapMetric})
|
|
|
|
var acc testutil.Accumulator
|
|
err := acc.GatherError(jolokia.Gather)
|
|
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, 1, len(acc.Metrics))
|
|
|
|
fields := map[string]interface{}{
|
|
"heap_memory_usage_java.lang:type=Memory_ObjectPendingFinalizationCount": 0.0,
|
|
"heap_memory_usage_java.lang:type=Memory_Verbose": false,
|
|
"heap_memory_usage_java.lang:type=Memory_HeapMemoryUsage_init": 134217728.0,
|
|
"heap_memory_usage_java.lang:type=Memory_HeapMemoryUsage_max": 1908932608.0,
|
|
"heap_memory_usage_java.lang:type=Memory_HeapMemoryUsage_used": 16840016.0,
|
|
"heap_memory_usage_java.lang:type=Memory_HeapMemoryUsage_committed": 173015040.0,
|
|
"heap_memory_usage_java.lang:type=Memory_NonHeapMemoryUsage_init": 2555904.0,
|
|
"heap_memory_usage_java.lang:type=Memory_NonHeapMemoryUsage_committed": 51380224.0,
|
|
"heap_memory_usage_java.lang:type=Memory_NonHeapMemoryUsage_max": -1.0,
|
|
"heap_memory_usage_java.lang:type=Memory_NonHeapMemoryUsage_used": 49944048.0,
|
|
"heap_memory_usage_java.lang:type=Memory_ObjectName_objectName": "java.lang:type=Memory",
|
|
}
|
|
|
|
tags := map[string]string{
|
|
"jolokia_host": "127.0.0.1",
|
|
"jolokia_port": "8080",
|
|
"jolokia_name": "as1",
|
|
}
|
|
acc.AssertContainsTaggedFields(t, "jolokia", fields, tags)
|
|
}
|
|
|
|
// Test that the proper values are ignored or collected
|
|
func TestHttp404(t *testing.T) {
|
|
|
|
jolokia := genJolokiaClientStub(invalidJSON, 404, Servers,
|
|
[]Metric{UsedHeapMetric})
|
|
|
|
var acc testutil.Accumulator
|
|
acc.SetDebug(true)
|
|
err := acc.GatherError(jolokia.Gather)
|
|
|
|
assert.Error(t, err)
|
|
assert.Equal(t, 0, len(acc.Metrics))
|
|
assert.Contains(t, err.Error(), "has status code 404")
|
|
}
|
|
|
|
// Test that the proper values are ignored or collected
|
|
func TestHttpInvalidJson(t *testing.T) {
|
|
|
|
jolokia := genJolokiaClientStub(invalidJSON, 200, Servers,
|
|
[]Metric{UsedHeapMetric})
|
|
|
|
var acc testutil.Accumulator
|
|
acc.SetDebug(true)
|
|
err := acc.GatherError(jolokia.Gather)
|
|
|
|
assert.Error(t, err)
|
|
assert.Equal(t, 0, len(acc.Metrics))
|
|
assert.Contains(t, err.Error(), "Error decoding JSON response")
|
|
}
|