Merge pull request #2186 from prometheus/fixes

Test fixes
This commit is contained in:
Fabian Reinartz 2016-11-14 09:52:15 +01:00 committed by GitHub
commit fa82c65d15
2 changed files with 9 additions and 5 deletions

View file

@ -506,7 +506,7 @@ func TestTargetScrapeScrapeCancel(t *testing.T) {
} }
ctx, cancel := context.WithCancel(context.Background()) ctx, cancel := context.WithCancel(context.Background())
done := make(chan struct{}) errc := make(chan error)
go func() { go func() {
time.Sleep(1 * time.Second) time.Sleep(1 * time.Second)
@ -515,15 +515,18 @@ func TestTargetScrapeScrapeCancel(t *testing.T) {
go func() { go func() {
if _, err := ts.scrape(ctx, time.Now()); err != context.Canceled { if _, err := ts.scrape(ctx, time.Now()); err != context.Canceled {
t.Fatalf("Expected context cancelation error but got: %s", err) errc <- fmt.Errorf("Expected context cancelation error but got: %s", err)
} }
close(done) close(errc)
}() }()
select { select {
case <-time.After(5 * time.Second): case <-time.After(5 * time.Second):
t.Fatalf("Scrape function did not return unexpectedly") t.Fatalf("Scrape function did not return unexpectedly")
case <-done: case err := <-errc:
if err != nil {
t.Fatalf(err.Error())
}
} }
// If this is closed in a defer above the function the test server // If this is closed in a defer above the function the test server
// does not terminate and the test doens't complete. // does not terminate and the test doens't complete.

View file

@ -1097,11 +1097,12 @@ func TestQuranatineSeriesFile(t *testing.T) {
t.Errorf("Unexpected state of quarantined file %q. Expected it to exist: %t. os.Stat returned: %s.", quarantinedFile, seriesFileShouldExist, err) t.Errorf("Unexpected state of quarantined file %q. Expected it to exist: %t. os.Stat returned: %s.", quarantinedFile, seriesFileShouldExist, err)
} }
f, err := os.Open(hintFile) f, err := os.Open(hintFile)
defer f.Close()
if err != nil { if err != nil {
t.Errorf("Could not open hint file %q: %s", hintFile, err) t.Errorf("Could not open hint file %q: %s", hintFile, err)
return return
} }
defer f.Close()
scanner := bufio.NewScanner(f) scanner := bufio.NewScanner(f)
for _, want := range contentHintFile { for _, want := range contentHintFile {
if !scanner.Scan() { if !scanner.Scan() {