internal: cleanup roundrobin_test with cleanup function (grpc#2519)

This PR refactor the code with a simple cleanup function.
zasweq · Dec 13, 2018 · 102aee9 · 102aee9
1 parent 14de8b6
commit 102aee9
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/balancer/roundrobin/roundrobin_test.go b/balancer/roundrobin/roundrobin_test.go
@@ -66,9 +66,7 @@ func startTestServers(count int) (_ *test, err error) {
 
 	defer func() {
 		if err != nil {
-			for _, s := range t.servers {
-				s.Stop()
-			}
+			t.cleanup()
 		}
 	}()
 	for i := 0; i < count; i++ {