mudhorn/prometheus
1
0
Fork 0
mirror of https://github.com/prometheus/prometheus.git synced 2025-03-05 20:59:13 -08:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

Fix retrieval unit tests.

Browse source 
Change-Id: I299b71406b59539230e5182ccc37bc8a83af60b3
This commit is contained in:
Bjoern Rabenstein 2014-10-10 15:24:38 +02:00
parent b3ed9aa7a2
commit 4fc8ad6677
1 changed files with 32 additions and 24 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

56
retrieval/targetpool_test.go
View file

@ -14,8 +14,11 @@
package retrieval package retrieval
import ( import (
"net/http"
"testing" "testing"
"time" "time"
clientmodel "github.com/prometheus/client_golang/model"
) )
func testTargetPool(t testing.TB) { func testTargetPool(t testing.TB) {
@ -46,12 +49,12 @@ func testTargetPool(t testing.TB) {
name: "single element", name: "single element",
inputs: []input{ inputs: []input{
{ {
address: "http://single.com", address: "single1",
}, },
}, },
outputs: []output{ outputs: []output{
{ {
address: "http://single.com", address: "single1",
}, },
}, },
}, },
@ -59,18 +62,18 @@ func testTargetPool(t testing.TB) {
name: "plural schedules", name: "plural schedules",
inputs: []input{ inputs: []input{
{ {
address: "http://plural.net", address: "plural1",
}, },
{ {
address: "http://plural.com", address: "plural2",
}, },
}, },
outputs: []output{ outputs: []output{
{ {
address: "http://plural.net", address: "plural1",
}, },
{ {
address: "http://plural.com", address: "plural2",
}, },
}, },
}, },
@ -81,9 +84,10 @@ func testTargetPool(t testing.TB) {
for _, input := range scenario.inputs { for _, input := range scenario.inputs {
target := target{ target := target{
address: input.address, address: input.address,
newBaseLabels: make(chan clientmodel.LabelSet, 1),
httpClient: &http.Client{},
} }
pool.addTarget(&target) pool.addTarget(&target)
} }
@ -110,30 +114,34 @@ func TestTargetPool(t *testing.T) {
func TestTargetPoolReplaceTargets(t *testing.T) { func TestTargetPoolReplaceTargets(t *testing.T) {
pool := NewTargetPool(nil, nil, nopIngester{}, time.Duration(1)) pool := NewTargetPool(nil, nil, nopIngester{}, time.Duration(1))
oldTarget1 := &target{ oldTarget1 := &target{
address: "example1", address: "example1",
state: UNREACHABLE, state: UNREACHABLE,
stopScraper: make(chan bool, 1), stopScraper: make(chan bool, 1),
newBaseLabels: make(chan clientmodel.LabelSet, 1),
httpClient: &http.Client{},
} }
oldTarget2 := &target{ oldTarget2 := &target{
address: "example2", address: "example2",
state: UNREACHABLE, state: UNREACHABLE,
stopScraper: make(chan bool, 1), stopScraper: make(chan bool, 1),
newBaseLabels: make(chan clientmodel.LabelSet, 1),
httpClient: &http.Client{},
} }
newTarget1 := &target{ newTarget1 := &target{
address: "example1", address: "example1",
state: ALIVE, state: ALIVE,
stopScraper: make(chan bool, 1), stopScraper: make(chan bool, 1),
newBaseLabels: make(chan clientmodel.LabelSet, 1),
httpClient: &http.Client{},
} }
newTarget2 := &target{ newTarget2 := &target{
address: "example3", address: "example3",
state: ALIVE, state: ALIVE,
stopScraper: make(chan bool, 1), stopScraper: make(chan bool, 1),
newBaseLabels: make(chan clientmodel.LabelSet, 1),
httpClient: &http.Client{},
} }
oldTarget1.StopScraper()
oldTarget2.StopScraper()
newTarget2.StopScraper()
pool.addTarget(oldTarget1) pool.addTarget(oldTarget1)
pool.addTarget(oldTarget2) pool.addTarget(oldTarget2)