104 lines
3.0 KiB
Go
104 lines
3.0 KiB
Go
package solr
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/influxdata/telegraf/testutil"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestGatherStats(t *testing.T) {
|
|
ts := createMockServer()
|
|
solr := NewSolr()
|
|
solr.Servers = []string{ts.URL}
|
|
var acc testutil.Accumulator
|
|
require.NoError(t, solr.Gather(&acc))
|
|
|
|
acc.AssertContainsTaggedFields(t, "solr_admin",
|
|
solrAdminMainCoreStatusExpected,
|
|
map[string]string{"core": "main"})
|
|
|
|
acc.AssertContainsTaggedFields(t, "solr_admin",
|
|
solrAdminCore1StatusExpected,
|
|
map[string]string{"core": "core1"})
|
|
|
|
acc.AssertContainsTaggedFields(t, "solr_core",
|
|
solrCoreExpected,
|
|
map[string]string{"core": "main", "handler": "searcher"})
|
|
|
|
acc.AssertContainsTaggedFields(t, "solr_queryhandler",
|
|
solrQueryHandlerExpected,
|
|
map[string]string{"core": "main", "handler": "org.apache.solr.handler.component.SearchHandler"})
|
|
|
|
acc.AssertContainsTaggedFields(t, "solr_updatehandler",
|
|
solrUpdateHandlerExpected,
|
|
map[string]string{"core": "main", "handler": "updateHandler"})
|
|
|
|
acc.AssertContainsTaggedFields(t, "solr_cache",
|
|
solrCacheExpected,
|
|
map[string]string{"core": "main", "handler": "filterCache"})
|
|
}
|
|
|
|
func createMockServer() *httptest.Server {
|
|
return httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
if strings.Contains(r.URL.Path, "/solr/admin/cores") {
|
|
w.WriteHeader(http.StatusOK)
|
|
fmt.Fprintln(w, statusResponse)
|
|
} else if strings.Contains(r.URL.Path, "solr/main/admin") {
|
|
w.WriteHeader(http.StatusOK)
|
|
fmt.Fprintln(w, mBeansMainResponse)
|
|
} else if strings.Contains(r.URL.Path, "solr/core1/admin") {
|
|
w.WriteHeader(http.StatusOK)
|
|
fmt.Fprintln(w, mBeansCore1Response)
|
|
} else {
|
|
w.WriteHeader(http.StatusNotFound)
|
|
fmt.Fprintln(w, "nope")
|
|
}
|
|
}))
|
|
}
|
|
|
|
func TestNoCoreDataHandling(t *testing.T) {
|
|
ts := createMockNoCoreDataServer()
|
|
solr := NewSolr()
|
|
solr.Servers = []string{ts.URL}
|
|
var acc testutil.Accumulator
|
|
require.NoError(t, solr.Gather(&acc))
|
|
|
|
acc.AssertContainsTaggedFields(t, "solr_admin",
|
|
solrAdminMainCoreStatusExpected,
|
|
map[string]string{"core": "main"})
|
|
|
|
acc.AssertContainsTaggedFields(t, "solr_admin",
|
|
solrAdminCore1StatusExpected,
|
|
map[string]string{"core": "core1"})
|
|
|
|
acc.AssertDoesNotContainMeasurement(t, "solr_core")
|
|
acc.AssertDoesNotContainMeasurement(t, "solr_queryhandler")
|
|
acc.AssertDoesNotContainMeasurement(t, "solr_updatehandler")
|
|
acc.AssertDoesNotContainMeasurement(t, "solr_handler")
|
|
|
|
}
|
|
|
|
func createMockNoCoreDataServer() *httptest.Server {
|
|
var nodata string
|
|
return httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
if strings.Contains(r.URL.Path, "/solr/admin/cores") {
|
|
w.WriteHeader(http.StatusOK)
|
|
fmt.Fprintln(w, statusResponse)
|
|
} else if strings.Contains(r.URL.Path, "solr/main/admin") {
|
|
w.WriteHeader(http.StatusOK)
|
|
fmt.Fprintln(w, nodata)
|
|
} else if strings.Contains(r.URL.Path, "solr/core1/admin") {
|
|
w.WriteHeader(http.StatusOK)
|
|
fmt.Fprintln(w, nodata)
|
|
} else {
|
|
w.WriteHeader(http.StatusNotFound)
|
|
fmt.Fprintln(w, "nope")
|
|
}
|
|
}))
|
|
}
|