119 lines
		
	
	
		
			2.7 KiB
		
	
	
	
		
			Go
		
	
	
	
			
		
		
	
	
			119 lines
		
	
	
		
			2.7 KiB
		
	
	
	
		
			Go
		
	
	
	
| // 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 promql
 | |
| 
 | |
| import (
 | |
| 	"context"
 | |
| 	"testing"
 | |
| 
 | |
| 	"github.com/prometheus/prometheus/pkg/labels"
 | |
| 	"github.com/prometheus/prometheus/pkg/timestamp"
 | |
| 	"github.com/prometheus/prometheus/util/testutil"
 | |
| )
 | |
| 
 | |
| func BenchmarkHoltWinters4Week5Min(b *testing.B) {
 | |
| 	input := `
 | |
| clear
 | |
| load 5m
 | |
|     http_requests{path="/foo"}    0+10x8064
 | |
| 
 | |
| eval instant at 4w holt_winters(http_requests[4w], 0.3, 0.3)
 | |
|     {path="/foo"} 80640
 | |
| `
 | |
| 
 | |
| 	bench := NewBenchmark(b, input)
 | |
| 	bench.Run()
 | |
| 
 | |
| }
 | |
| 
 | |
| func BenchmarkHoltWinters1Week5Min(b *testing.B) {
 | |
| 	input := `
 | |
| clear
 | |
| load 5m
 | |
|     http_requests{path="/foo"}    0+10x2016
 | |
| 
 | |
| eval instant at 1w holt_winters(http_requests[1w], 0.3, 0.3)
 | |
|     {path="/foo"} 20160
 | |
| `
 | |
| 
 | |
| 	bench := NewBenchmark(b, input)
 | |
| 	bench.Run()
 | |
| }
 | |
| 
 | |
| func BenchmarkHoltWinters1Day1Min(b *testing.B) {
 | |
| 	input := `
 | |
| clear
 | |
| load 1m
 | |
|     http_requests{path="/foo"}    0+10x1440
 | |
| 
 | |
| eval instant at 1d holt_winters(http_requests[1d], 0.3, 0.3)
 | |
|     {path="/foo"} 14400
 | |
| `
 | |
| 
 | |
| 	bench := NewBenchmark(b, input)
 | |
| 	bench.Run()
 | |
| }
 | |
| 
 | |
| func BenchmarkChanges1Day1Min(b *testing.B) {
 | |
| 	input := `
 | |
| clear
 | |
| load 1m
 | |
|     http_requests{path="/foo"}    0+10x1440
 | |
| 
 | |
| eval instant at 1d changes(http_requests[1d])
 | |
|     {path="/foo"} 1440
 | |
| `
 | |
| 
 | |
| 	bench := NewBenchmark(b, input)
 | |
| 	bench.Run()
 | |
| }
 | |
| 
 | |
| func TestDeriv(t *testing.T) {
 | |
| 	// https://github.com/prometheus/prometheus/issues/2674#issuecomment-315439393
 | |
| 	// This requires more precision than the usual test system offers,
 | |
| 	// so we test it by hand.
 | |
| 	storage := testutil.NewStorage(t)
 | |
| 	defer storage.Close()
 | |
| 	engine := NewEngine(storage, nil)
 | |
| 
 | |
| 	a, err := storage.Appender()
 | |
| 	if err != nil {
 | |
| 		t.Fatal(err)
 | |
| 	}
 | |
| 
 | |
| 	metric := labels.FromStrings("__name__", "foo")
 | |
| 	a.Add(metric, 1493712816939, 1.0)
 | |
| 	a.Add(metric, 1493712846939, 1.0)
 | |
| 
 | |
| 	if err := a.Commit(); err != nil {
 | |
| 		t.Fatal(err)
 | |
| 	}
 | |
| 
 | |
| 	query, err := engine.NewInstantQuery("deriv(foo[30m])", timestamp.Time(1493712846939))
 | |
| 	if err != nil {
 | |
| 		t.Fatalf("Error parsing query: %s", err)
 | |
| 	}
 | |
| 	result := query.Exec(context.Background())
 | |
| 	if result.Err != nil {
 | |
| 		t.Fatalf("Error running query: %s", result.Err)
 | |
| 	}
 | |
| 	vec, _ := result.Vector()
 | |
| 	if len(vec) != 1 {
 | |
| 		t.Fatalf("Expected 1 result, got %d", len(vec))
 | |
| 	}
 | |
| 	if vec[0].V != 0.0 {
 | |
| 		t.Fatalf("Expected 0.0 as value, got %f", vec[0].V)
 | |
| 	}
 | |
| }
 |