From c28702cb2d7d15fcbe814b582144a49bada874c9 Mon Sep 17 00:00:00 2001
From: Diptanu Choudhury <diptanuc@gmail.com>
Date: Thu, 5 Nov 2015 17:33:03 -0800
Subject: [PATCH] Refactored test

---
 client/restarts_test.go | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/client/restarts_test.go b/client/restarts_test.go
index e27f10390..9d5b59bb4 100644
--- a/client/restarts_test.go
+++ b/client/restarts_test.go
@@ -39,7 +39,12 @@ func TestTaskRunner_BatchRestartCounter(t *testing.T) {
 	attempts := 2
 	interval := 1 * time.Second
 	delay := 1 * time.Second
-	rt := newRestartTracker(structs.JobTypeBatch, &structs.RestartPolicy{Attempts: attempts, Interval: interval, Delay: delay})
+	rt := newRestartTracker(structs.JobTypeBatch,
+		&structs.RestartPolicy{Attempts: attempts,
+			Interval: interval,
+			Delay:    delay,
+		},
+	)
 	for i := 0; i < attempts; i++ {
 		shouldRestart, when := rt.nextRestart()
 		if !shouldRestart {