132 lines
3.3 KiB
Go
132 lines
3.3 KiB
Go
// Copyright 2013 Prometheus Team
|
|
// 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 tiered
|
|
|
|
import (
|
|
"fmt"
|
|
"testing"
|
|
"time"
|
|
|
|
clientmodel "github.com/prometheus/client_golang/model"
|
|
|
|
"github.com/prometheus/prometheus/storage/metric"
|
|
"github.com/prometheus/prometheus/utility/test"
|
|
)
|
|
|
|
var (
|
|
// ``hg clone https://code.google.com/p/go ; cd go ; hg log | tail -n 20``
|
|
usEastern, _ = time.LoadLocation("US/Eastern")
|
|
testInstant = clientmodel.TimestampFromTime(time.Date(1972, 7, 18, 19, 5, 45, 0, usEastern).In(time.UTC))
|
|
)
|
|
|
|
func testAppendSamples(p metric.Persistence, s *clientmodel.Sample, t testing.TB) {
|
|
err := p.AppendSamples(clientmodel.Samples{s})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
|
|
func buildLevelDBTestPersistencesMaker(name string, t testing.TB) func() (metric.Persistence, test.Closer) {
|
|
return func() (metric.Persistence, test.Closer) {
|
|
temporaryDirectory := test.NewTemporaryDirectory("get_value_at_time", t)
|
|
|
|
p, err := NewLevelDBPersistence(temporaryDirectory.Path())
|
|
if err != nil {
|
|
t.Errorf("Could not start up LevelDB: %q\n", err)
|
|
}
|
|
|
|
return p, temporaryDirectory
|
|
}
|
|
}
|
|
|
|
func buildLevelDBTestPersistence(name string, f func(p metric.Persistence, t testing.TB)) func(t testing.TB) {
|
|
return func(t testing.TB) {
|
|
|
|
temporaryDirectory := test.NewTemporaryDirectory(fmt.Sprintf("test_leveldb_%s", name), t)
|
|
defer temporaryDirectory.Close()
|
|
|
|
p, err := NewLevelDBPersistence(temporaryDirectory.Path())
|
|
|
|
if err != nil {
|
|
t.Errorf("Could not create LevelDB Metric Persistence: %q\n", err)
|
|
}
|
|
|
|
defer p.Close()
|
|
|
|
f(p, t)
|
|
}
|
|
}
|
|
|
|
func buildMemoryTestPersistence(f func(p metric.Persistence, t testing.TB)) func(t testing.TB) {
|
|
return func(t testing.TB) {
|
|
|
|
p := NewMemorySeriesStorage(MemorySeriesOptions{})
|
|
|
|
defer p.Close()
|
|
|
|
f(p, t)
|
|
}
|
|
}
|
|
|
|
type testTieredStorageCloser struct {
|
|
storage *TieredStorage
|
|
directory test.Closer
|
|
}
|
|
|
|
func (t *testTieredStorageCloser) Close() {
|
|
t.storage.Close()
|
|
t.directory.Close()
|
|
}
|
|
|
|
func NewTestTieredStorage(t testing.TB) (*TieredStorage, test.Closer) {
|
|
directory := test.NewTemporaryDirectory("test_tiered_storage", t)
|
|
storage, err := NewTieredStorage(2500, 1000, 5*time.Second, 0, directory.Path())
|
|
|
|
if err != nil {
|
|
if storage != nil {
|
|
storage.Close()
|
|
}
|
|
directory.Close()
|
|
t.Fatalf("Error creating storage: %s", err)
|
|
}
|
|
|
|
if storage == nil {
|
|
directory.Close()
|
|
t.Fatalf("storage == nil")
|
|
}
|
|
|
|
started := make(chan bool)
|
|
go storage.Serve(started)
|
|
<-started
|
|
|
|
closer := &testTieredStorageCloser{
|
|
storage: storage,
|
|
directory: directory,
|
|
}
|
|
|
|
return storage, closer
|
|
}
|
|
|
|
func labelMatchersFromLabelSet(l clientmodel.LabelSet) metric.LabelMatchers {
|
|
m := make(metric.LabelMatchers, 0, len(l))
|
|
for k, v := range l {
|
|
m = append(m, &metric.LabelMatcher{
|
|
Type: metric.Equal,
|
|
Name: k,
|
|
Value: v,
|
|
})
|
|
}
|
|
return m
|
|
}
|