From bf7d87aed278a32eb440e4b7337b99d43a635d4c Mon Sep 17 00:00:00 2001
From: Brian Brazil <brian.brazil@robustperception.io>
Date: Thu, 8 Mar 2018 16:48:11 +0000
Subject: [PATCH] Cleanup storage from all tests.

Fixed #3299
---
 promql/bench_test.go  | 1 +
 promql/engine_test.go | 2 ++
 2 files changed, 3 insertions(+)

diff --git a/promql/bench_test.go b/promql/bench_test.go
index b628aa2a8..cd9185451 100644
--- a/promql/bench_test.go
+++ b/promql/bench_test.go
@@ -36,6 +36,7 @@ func NewBenchmark(b *testing.B, input string) *Benchmark {
 
 // Run runs the benchmark.
 func (b *Benchmark) Run() {
+	defer b.t.Close()
 	b.b.ReportAllocs()
 	b.b.ResetTimer()
 	for i := 0; i < b.b.N; i++ {
diff --git a/promql/engine_test.go b/promql/engine_test.go
index eeebe9f5a..c2e1c7753 100644
--- a/promql/engine_test.go
+++ b/promql/engine_test.go
@@ -204,6 +204,8 @@ load 10s
 	if err != nil {
 		t.Fatalf("unexpected error creating test: %q", err)
 	}
+	defer test.Close()
+
 	err = test.Run()
 	if err != nil {
 		t.Fatalf("unexpected error initializing test: %q", err)