From 535e0dff59ed82abe72ae6174f68af76f48c7943 Mon Sep 17 00:00:00 2001
From: Marius <maerious@gmail.com>
Date: Wed, 12 Apr 2017 12:11:11 +0200
Subject: [PATCH] Use new consul test server functions

---
 consullocker/consullocker_test.go | 16 +++++++++++-----
 1 file changed, 11 insertions(+), 5 deletions(-)

diff --git a/consullocker/consullocker_test.go b/consullocker/consullocker_test.go
index 0eb90e5..19dbed6 100644
--- a/consullocker/consullocker_test.go
+++ b/consullocker/consullocker_test.go
@@ -14,12 +14,15 @@ import (
 func TestConsulLocker(t *testing.T) {
 	a := assert.New(t)
 
-	server := consultestutil.NewTestServer(t)
+	server, err := consultestutil.NewTestServer()
+	if err != nil {
+		t.Fatal(err)
+	}
 	defer server.Stop()
 
-	client, err := consul.NewClient(&consul.Config{
-		Address: server.HTTPAddr,
-	})
+	conf := consul.DefaultConfig()
+	conf.Address = server.HTTPAddr
+	client, err := consul.NewClient(conf)
 	a.NoError(err)
 
 	locker := New(client)
@@ -38,7 +41,10 @@ func TestLockLost(t *testing.T) {
 
 	a := assert.New(t)
 
-	server := consultestutil.NewTestServer(t)
+	server, err := consultestutil.NewTestServer()
+	if err != nil {
+		t.Fatal(err)
+	}
 
 	client, err := consul.NewClient(&consul.Config{
 		Address: server.HTTPAddr,