mirror of
https://github.com/VictoriaMetrics/VictoriaMetrics.git
synced 2024-11-21 17:32:05 +01:00
2febd00bb3
* make test suite responisble for stopping apps
* reuse test suite fields to simplify function signatures
---------
Signed-off-by: hagen1778 <roman@victoriametrics.com>
(cherry picked from commit e60cce54a8
)
Signed-off-by: hagen1778 <roman@victoriametrics.com>
113 lines
3.8 KiB
Go
113 lines
3.8 KiB
Go
package apptest
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
"net/url"
|
|
"regexp"
|
|
"testing"
|
|
)
|
|
|
|
// Vmselect holds the state of a vmselect app and provides vmselect-specific
|
|
// functions.
|
|
type Vmselect struct {
|
|
*app
|
|
*ServesMetrics
|
|
|
|
httpListenAddr string
|
|
clusternativeListenAddr string
|
|
cli *Client
|
|
}
|
|
|
|
// StartVmselect starts an instance of vmselect with the given flags. It also
|
|
// sets the default flags and populates the app instance state with runtime
|
|
// values extracted from the application log (such as httpListenAddr)
|
|
func StartVmselect(instance string, flags []string, cli *Client) (*Vmselect, error) {
|
|
app, stderrExtracts, err := startApp(instance, "../../bin/vmselect", flags, &appOptions{
|
|
defaultFlags: map[string]string{
|
|
"-httpListenAddr": "127.0.0.1:0",
|
|
"-clusternativeListenAddr": "127.0.0.1:0",
|
|
},
|
|
extractREs: []*regexp.Regexp{
|
|
httpListenAddrRE,
|
|
vmselectAddrRE,
|
|
},
|
|
})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &Vmselect{
|
|
app: app,
|
|
ServesMetrics: &ServesMetrics{
|
|
metricsURL: fmt.Sprintf("http://%s/metrics", stderrExtracts[0]),
|
|
cli: cli,
|
|
},
|
|
httpListenAddr: stderrExtracts[0],
|
|
clusternativeListenAddr: stderrExtracts[1],
|
|
cli: cli,
|
|
}, nil
|
|
}
|
|
|
|
// ClusternativeListenAddr returns the address at which the vmselect process is
|
|
// listening for connections from other vmselect apps.
|
|
func (app *Vmselect) ClusternativeListenAddr() string {
|
|
return app.clusternativeListenAddr
|
|
}
|
|
|
|
// PrometheusAPIV1Query is a test helper function that performs PromQL/MetricsQL
|
|
// instant query by sending a HTTP POST request to /prometheus/api/v1/query
|
|
// vmsingle endpoint.
|
|
//
|
|
// See https://docs.victoriametrics.com/url-examples/#apiv1query
|
|
func (app *Vmselect) PrometheusAPIV1Query(t *testing.T, query, time, step string, opts QueryOpts) *PrometheusAPIV1QueryResponse {
|
|
t.Helper()
|
|
|
|
queryURL := fmt.Sprintf("http://%s/select/%s/prometheus/api/v1/query", app.httpListenAddr, opts.Tenant)
|
|
values := url.Values{}
|
|
values.Add("query", query)
|
|
values.Add("time", time)
|
|
values.Add("step", step)
|
|
values.Add("timeout", opts.Timeout)
|
|
res := app.cli.PostForm(t, queryURL, values, http.StatusOK)
|
|
return NewPrometheusAPIV1QueryResponse(t, res)
|
|
}
|
|
|
|
// PrometheusAPIV1QueryRange is a test helper function that performs
|
|
// PromQL/MetricsQL range query by sending a HTTP POST request to
|
|
// /prometheus/api/v1/query_range vmsingle endpoint.
|
|
//
|
|
// See https://docs.victoriametrics.com/url-examples/#apiv1query_range
|
|
func (app *Vmselect) PrometheusAPIV1QueryRange(t *testing.T, query, start, end, step string, opts QueryOpts) *PrometheusAPIV1QueryResponse {
|
|
t.Helper()
|
|
|
|
queryURL := fmt.Sprintf("http://%s/select/%s/prometheus/api/v1/query_range", app.httpListenAddr, opts.Tenant)
|
|
values := url.Values{}
|
|
values.Add("query", query)
|
|
values.Add("start", start)
|
|
values.Add("end", end)
|
|
values.Add("step", step)
|
|
values.Add("timeout", opts.Timeout)
|
|
res := app.cli.PostForm(t, queryURL, values, http.StatusOK)
|
|
return NewPrometheusAPIV1QueryResponse(t, res)
|
|
}
|
|
|
|
// PrometheusAPIV1Series sends a query to a /prometheus/api/v1/series endpoint
|
|
// and returns the list of time series that match the query.
|
|
//
|
|
// See https://docs.victoriametrics.com/url-examples/#apiv1series
|
|
func (app *Vmselect) PrometheusAPIV1Series(t *testing.T, matchQuery string, opts QueryOpts) *PrometheusAPIV1SeriesResponse {
|
|
t.Helper()
|
|
|
|
seriesURL := fmt.Sprintf("http://%s/select/%s/prometheus/api/v1/series", app.httpListenAddr, opts.Tenant)
|
|
values := url.Values{}
|
|
values.Add("match[]", matchQuery)
|
|
res := app.cli.PostForm(t, seriesURL, values, http.StatusOK)
|
|
return NewPrometheusAPIV1SeriesResponse(t, res)
|
|
}
|
|
|
|
// String returns the string representation of the vmselect app state.
|
|
func (app *Vmselect) String() string {
|
|
return fmt.Sprintf("{app: %s httpListenAddr: %q}", app.app, app.httpListenAddr)
|
|
}
|