influxdb/inmem/scraper_test.go

42 lines
1.1 KiB
Go

package inmem
import (
"context"
"testing"
"github.com/influxdata/platform"
platformtesting "github.com/influxdata/platform/testing"
)
func initScraperTargetStoreService(f platformtesting.TargetFields, t *testing.T) (platform.ScraperTargetStoreService, func()) {
s := NewService()
s.IDGenerator = f.IDGenerator
ctx := context.TODO()
for _, target := range f.Targets {
if err := s.PutTarget(ctx, &target); err != nil {
t.Fatalf("failed to populate scraper targets")
}
}
return s, func() {}
}
func TestScraperTargetStoreService_AddTarget(t *testing.T) {
platformtesting.AddTarget(initScraperTargetStoreService, t)
}
func TestScraperTargetStoreService_ListTargets(t *testing.T) {
platformtesting.ListTargets(initScraperTargetStoreService, t)
}
func TestScraperTargetStoreService_RemoveTarget(t *testing.T) {
platformtesting.RemoveTarget(initScraperTargetStoreService, t)
}
func TestScraperTargetStoreService_UpdateTarget(t *testing.T) {
platformtesting.UpdateTarget(initScraperTargetStoreService, t)
}
func TestScraperTargetStoreService_GetTargetByID(t *testing.T) {
platformtesting.GetTargetByID(initScraperTargetStoreService, t)
}