mirror of
https://github.com/VictoriaMetrics/VictoriaMetrics.git
synced 2024-12-22 08:26:19 +01:00
127 lines
3.2 KiB
Go
127 lines
3.2 KiB
Go
package searchutils
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
"net/url"
|
|
"reflect"
|
|
"testing"
|
|
|
|
"github.com/VictoriaMetrics/VictoriaMetrics/lib/storage"
|
|
)
|
|
|
|
func TestGetTimeSuccess(t *testing.T) {
|
|
f := func(s string, timestampExpected int64) {
|
|
t.Helper()
|
|
urlStr := fmt.Sprintf("http://foo.bar/baz?s=%s", url.QueryEscape(s))
|
|
r, err := http.NewRequest("GET", urlStr, nil)
|
|
if err != nil {
|
|
t.Fatalf("unexpected error in NewRequest: %s", err)
|
|
}
|
|
|
|
// Verify defaultValue
|
|
ts, err := GetTime(r, "foo", 123456)
|
|
if err != nil {
|
|
t.Fatalf("unexpected error when obtaining default time from GetTime(%q): %s", s, err)
|
|
}
|
|
if ts != 123000 {
|
|
t.Fatalf("unexpected default value for GetTime(%q); got %d; want %d", s, ts, 123000)
|
|
}
|
|
|
|
// Verify timestampExpected
|
|
ts, err = GetTime(r, "s", 123)
|
|
if err != nil {
|
|
t.Fatalf("unexpected error in GetTime(%q): %s", s, err)
|
|
}
|
|
if ts != timestampExpected {
|
|
t.Fatalf("unexpected timestamp for GetTime(%q); got %d; want %d", s, ts, timestampExpected)
|
|
}
|
|
}
|
|
|
|
f("2019-07-07T20:01:02Z", 1562529662000)
|
|
f("2019-07-07T20:47:40+03:00", 1562521660000)
|
|
f("-292273086-05-16T16:47:06Z", minTimeMsecs)
|
|
f("292277025-08-18T07:12:54.999999999Z", maxTimeMsecs)
|
|
f("1562529662.324", 1562529662324)
|
|
f("-9223372036.854", minTimeMsecs)
|
|
f("-9223372036.855", minTimeMsecs)
|
|
f("9223372036.855", maxTimeMsecs)
|
|
}
|
|
|
|
func TestGetTimeError(t *testing.T) {
|
|
f := func(s string) {
|
|
t.Helper()
|
|
urlStr := fmt.Sprintf("http://foo.bar/baz?s=%s", url.QueryEscape(s))
|
|
r, err := http.NewRequest("GET", urlStr, nil)
|
|
if err != nil {
|
|
t.Fatalf("unexpected error in NewRequest: %s", err)
|
|
}
|
|
|
|
// Verify defaultValue
|
|
ts, err := GetTime(r, "foo", 123456)
|
|
if err != nil {
|
|
t.Fatalf("unexpected error when obtaining default time from GetTime(%q): %s", s, err)
|
|
}
|
|
if ts != 123000 {
|
|
t.Fatalf("unexpected default value for GetTime(%q); got %d; want %d", s, ts, 123000)
|
|
}
|
|
|
|
// Verify timestampExpected
|
|
_, err = GetTime(r, "s", 123)
|
|
if err == nil {
|
|
t.Fatalf("expecting non-nil error in GetTime(%q)", s)
|
|
}
|
|
}
|
|
|
|
f("foo")
|
|
f("2019-07-07T20:01:02Zisdf")
|
|
f("2019-07-07T20:47:40+03:00123")
|
|
f("-292273086-05-16T16:47:07Z")
|
|
f("292277025-08-18T07:12:54.999999998Z")
|
|
}
|
|
|
|
// helper for tests
|
|
func tfFromKV(k, v string) storage.TagFilter {
|
|
return storage.TagFilter{
|
|
Key: []byte(k),
|
|
Value: []byte(v),
|
|
}
|
|
}
|
|
|
|
func TestGetEnforcedTagFiltersFromRequest(t *testing.T) {
|
|
httpReqWithForm := func(tfs []string) *http.Request {
|
|
return &http.Request{
|
|
Form: map[string][]string{
|
|
"extra_label": tfs,
|
|
},
|
|
}
|
|
}
|
|
f := func(t *testing.T, r *http.Request, want []storage.TagFilter, wantErr bool) {
|
|
t.Helper()
|
|
got, err := GetEnforcedTagFiltersFromRequest(r)
|
|
if (err != nil) != wantErr {
|
|
t.Fatalf("unexpected error: %v", err)
|
|
}
|
|
if !reflect.DeepEqual(got, want) {
|
|
t.Fatalf("unxpected result for getEnforcedTagFiltersFromRequest, \ngot: %v,\n want: %v", want, got)
|
|
}
|
|
}
|
|
|
|
f(t, httpReqWithForm([]string{"label=value"}),
|
|
[]storage.TagFilter{
|
|
tfFromKV("label", "value"),
|
|
},
|
|
false)
|
|
|
|
f(t, httpReqWithForm([]string{"job=vmagent", "dc=gce"}),
|
|
[]storage.TagFilter{tfFromKV("job", "vmagent"), tfFromKV("dc", "gce")},
|
|
false,
|
|
)
|
|
f(t, httpReqWithForm([]string{"bad_filter"}),
|
|
nil,
|
|
true,
|
|
)
|
|
f(t, &http.Request{},
|
|
nil, false)
|
|
}
|