mirror of https://github.com/prometheus/prometheus
Julius Volz
10 years ago
commit
7793651b7d
@ -1,42 +0,0 @@
|
||||
// Copyright 2013 The Prometheus Authors
|
||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||
// you may not use this file except in compliance with the License.
|
||||
// You may obtain a copy of the License at
|
||||
//
|
||||
// http://www.apache.org/licenses/LICENSE-2.0
|
||||
//
|
||||
// Unless required by applicable law or agreed to in writing, software
|
||||
// distributed under the License is distributed on an "AS IS" BASIS,
|
||||
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
// See the License for the specific language governing permissions and
|
||||
// limitations under the License.
|
||||
|
||||
package test
|
||||
|
||||
import (
|
||||
"github.com/prometheus/prometheus/utility"
|
||||
"time"
|
||||
)
|
||||
|
||||
type instantProvider struct {
|
||||
index int
|
||||
timeQueue []time.Time
|
||||
}
|
||||
|
||||
func (t *instantProvider) Now() (time time.Time) {
|
||||
time = t.timeQueue[t.index]
|
||||
|
||||
t.index++
|
||||
|
||||
return
|
||||
}
|
||||
|
||||
// NewInstantProvider furnishes an InstantProvider with prerecorded responses
|
||||
// for calls made against it. It has no validation behaviors of its own and
|
||||
// will panic if times are requested more than available pre-recorded behaviors.
|
||||
func NewInstantProvider(times []time.Time) utility.InstantProvider {
|
||||
return &instantProvider{
|
||||
index: 0,
|
||||
timeQueue: times,
|
||||
}
|
||||
}
|
||||
@ -1,45 +0,0 @@
|
||||
// Copyright 2013 The Prometheus Authors
|
||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||
// you may not use this file except in compliance with the License.
|
||||
// You may obtain a copy of the License at
|
||||
//
|
||||
// http://www.apache.org/licenses/LICENSE-2.0
|
||||
//
|
||||
// Unless required by applicable law or agreed to in writing, software
|
||||
// distributed under the License is distributed on an "AS IS" BASIS,
|
||||
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
// See the License for the specific language governing permissions and
|
||||
// limitations under the License.
|
||||
|
||||
package utility
|
||||
|
||||
import (
|
||||
"time"
|
||||
)
|
||||
|
||||
// InstantProvider is a basic interface only useful in testing contexts for
|
||||
// dispensing the time in a controlled manner.
|
||||
type InstantProvider interface {
|
||||
// The current instant.
|
||||
Now() time.Time
|
||||
}
|
||||
|
||||
// Time is a simple means for fluently wrapping around standard Go timekeeping
|
||||
// mechanisms to enhance testability without compromising code readability.
|
||||
//
|
||||
// It is sufficient for use on bare initialization. A provider should be
|
||||
// set only for test contexts. When not provided, it emits the current
|
||||
// system time.
|
||||
type Time struct {
|
||||
// The underlying means through which time is provided, if supplied.
|
||||
Provider InstantProvider
|
||||
}
|
||||
|
||||
// Now emits the current instant.
|
||||
func (t Time) Now() time.Time {
|
||||
if t.Provider == nil {
|
||||
return time.Now()
|
||||
}
|
||||
|
||||
return t.Provider.Now()
|
||||
}
|
||||
@ -0,0 +1,134 @@
|
||||
// Copyright 2015 The Prometheus Authors
|
||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||
// you may not use this file except in compliance with the License.
|
||||
// You may obtain a copy of the License at
|
||||
//
|
||||
// http://www.apache.org/licenses/LICENSE-2.0
|
||||
//
|
||||
// Unless required by applicable law or agreed to in writing, software
|
||||
// distributed under the License is distributed on an "AS IS" BASIS,
|
||||
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
// See the License for the specific language governing permissions and
|
||||
// limitations under the License.
|
||||
|
||||
package api
|
||||
|
||||
import (
|
||||
"io/ioutil"
|
||||
"net/http"
|
||||
"net/http/httptest"
|
||||
"regexp"
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
clientmodel "github.com/prometheus/client_golang/model"
|
||||
|
||||
"github.com/prometheus/prometheus/storage/local"
|
||||
)
|
||||
|
||||
// This is a bit annoying. On one hand, we have to choose a current timestamp
|
||||
// because the storage doesn't have a mocked-out time yet and would otherwise
|
||||
// immediately throw away "old" samples. On the other hand, we have to make
|
||||
// sure that the float value survives the parsing and re-formatting in the
|
||||
// query layer precisely without any change. Thus we round to seconds and then
|
||||
// add known-good digits after the decimal point which behave well in
|
||||
// parsing/re-formatting.
|
||||
var testTimestamp = clientmodel.TimestampFromTime(time.Now().Round(time.Second)).Add(124 * time.Millisecond)
|
||||
|
||||
func testNow() clientmodel.Timestamp {
|
||||
return testTimestamp
|
||||
}
|
||||
|
||||
func TestQuery(t *testing.T) {
|
||||
scenarios := []struct {
|
||||
// URL query string.
|
||||
queryStr string
|
||||
// Expected HTTP response status code.
|
||||
status int
|
||||
// Regex to match against response body.
|
||||
bodyRe string
|
||||
}{
|
||||
{
|
||||
queryStr: "",
|
||||
status: http.StatusOK,
|
||||
bodyRe: "syntax error",
|
||||
},
|
||||
{
|
||||
queryStr: "expr=testmetric",
|
||||
status: http.StatusOK,
|
||||
bodyRe: `{"type":"vector","value":\[\{"metric":{"__name__":"testmetric"},"value":"0","timestamp":\d+\.\d+}\],"version":1\}`,
|
||||
},
|
||||
{
|
||||
queryStr: "expr=testmetric×tamp=" + testTimestamp.String(),
|
||||
status: http.StatusOK,
|
||||
bodyRe: `{"type":"vector","value":\[\{"metric":{"__name__":"testmetric"},"value":"0","timestamp":` + testTimestamp.String() + `}\],"version":1\}`,
|
||||
},
|
||||
{
|
||||
queryStr: "expr=testmetric×tamp=" + testTimestamp.Add(-time.Hour).String(),
|
||||
status: http.StatusOK,
|
||||
bodyRe: `{"type":"vector","value":\[\],"version":1\}`,
|
||||
},
|
||||
{
|
||||
queryStr: "timestamp=invalid",
|
||||
status: http.StatusBadRequest,
|
||||
bodyRe: "invalid query timestamp",
|
||||
},
|
||||
{
|
||||
queryStr: "expr=(badexpression",
|
||||
status: http.StatusOK,
|
||||
bodyRe: "syntax error",
|
||||
},
|
||||
}
|
||||
|
||||
storage, closer := local.NewTestStorage(t, 1)
|
||||
defer closer.Close()
|
||||
storage.Append(&clientmodel.Sample{
|
||||
Metric: clientmodel.Metric{
|
||||
clientmodel.MetricNameLabel: "testmetric",
|
||||
},
|
||||
Timestamp: testTimestamp,
|
||||
Value: 0,
|
||||
})
|
||||
storage.WaitForIndexing()
|
||||
|
||||
api := MetricsService{
|
||||
Now: testNow,
|
||||
Storage: storage,
|
||||
}
|
||||
api.RegisterHandler()
|
||||
|
||||
server := httptest.NewServer(http.DefaultServeMux)
|
||||
defer server.Close()
|
||||
|
||||
for i, s := range scenarios {
|
||||
// Do query.
|
||||
resp, err := http.Get(server.URL + "/api/query?" + s.queryStr)
|
||||
if err != nil {
|
||||
t.Fatalf("%d. Error querying API: %s", i, err)
|
||||
}
|
||||
|
||||
// Check status code.
|
||||
if resp.StatusCode != s.status {
|
||||
t.Fatalf("%d. Unexpected status code; got %d, want %d", i, resp.StatusCode, s.status)
|
||||
}
|
||||
|
||||
// Check response headers.
|
||||
ct := resp.Header["Content-Type"]
|
||||
if len(ct) != 1 {
|
||||
t.Fatalf("%d. Unexpected number of 'Content-Type' headers; got %d, want 1", i, len(ct))
|
||||
}
|
||||
if ct[0] != "application/json" {
|
||||
t.Fatalf("%d. Unexpected 'Content-Type' header; got %s; want %s", i, ct[0], "application/json")
|
||||
}
|
||||
|
||||
// Check body.
|
||||
b, err := ioutil.ReadAll(resp.Body)
|
||||
if err != nil {
|
||||
t.Fatalf("%d. Error reading response body: %s", i, err)
|
||||
}
|
||||
re := regexp.MustCompile(s.bodyRe)
|
||||
if !re.Match(b) {
|
||||
t.Fatalf("%d. Body didn't match '%s'. Body: %s", i, s.bodyRe, string(b))
|
||||
}
|
||||
}
|
||||
}
|
||||
Loading…
Reference in new issue