Merge pull request #135 from mavenugo/epleave

Cleaning up the ActiveContainer test properly

libnetwork/libnetwork_test.go

@@ -606,10 +606,12 @@ func TestEndpointJoin(t *testing.T) {
 		t.Fatal(err)
 	}
 
-	err = ep.Leave(containerID)
-	if err != nil {
-		t.Fatal(err)
-	}
+	defer func() {
+		err = ep.Leave(containerID)
+		if err != nil {
+			t.Fatal(err)
+		}
+	}()
 }
 
 func TestEndpointJoinInvalidContainerId(t *testing.T) {
@@ -655,6 +657,17 @@ func TestEndpointDeleteWithActiveContainer(t *testing.T) {
 	if err != nil {
 		t.Fatal(err)
 	}
+	defer func() {
+		err = ep.Leave(containerID)
+		if err != nil {
+			t.Fatal(err)
+		}
+
+		err = ep.Delete()
+		if err != nil {
+			t.Fatal(err)
+		}
+	}()
 
 	err = ep.Delete()
 	if err == nil {
@@ -687,6 +700,12 @@ func TestEndpointMultipleJoins(t *testing.T) {
 	if err != nil {
 		t.Fatal(err)
 	}
+	defer func() {
+		err = ep.Leave(containerID)
+		if err != nil {
+			t.Fatal(err)
+		}
+	}()
 
 	_, err = ep.Join("container2")
 	if err == nil {
@@ -696,11 +715,6 @@ func TestEndpointMultipleJoins(t *testing.T) {
 	if err != libnetwork.ErrInvalidJoin {
 		t.Fatalf("Failed for unexpected reason: %v", err)
 	}
-
-	err = ep.Leave(containerID)
-	if err != nil {
-		t.Fatal(err)
-	}
 }
 
 func TestEndpointInvalidLeave(t *testing.T) {
@@ -733,6 +747,12 @@ func TestEndpointInvalidLeave(t *testing.T) {
 	if err != nil {
 		t.Fatal(err)
 	}
+	defer func() {
+		err = ep.Leave(containerID)
+		if err != nil {
+			t.Fatal(err)
+		}
+	}()
 
 	err = ep.Leave("")
 	if err == nil {
@@ -752,10 +772,6 @@ func TestEndpointInvalidLeave(t *testing.T) {
 		t.Fatalf("Failed for unexpected reason: %v", err)
 	}
 
-	err = ep.Leave(containerID)
-	if err != nil {
-		t.Fatal(err)
-	}
 }
 
 func TestEndpointUpdateParent(t *testing.T) {
@@ -779,6 +795,12 @@ func TestEndpointUpdateParent(t *testing.T) {
 	if err != nil {
 		t.Fatal(err)
 	}
+	defer func() {
+		err = ep1.Leave(containerID)
+		if err != nil {
+			t.Fatal(err)
+		}
+	}()
 
 	ep2, err := n.CreateEndpoint("ep2", nil)
 	if err != nil {
@@ -795,15 +817,13 @@ func TestEndpointUpdateParent(t *testing.T) {
 		t.Fatal(err)
 	}
 
-	err = ep2.Leave("container2")
-	if err != nil {
-		t.Fatal(err)
-	}
+	defer func() {
+		err = ep2.Leave("container2")
+		if err != nil {
+			t.Fatal(err)
+		}
+	}()
 
-	err = ep1.Leave(containerID)
-	if err != nil {
-		t.Fatal(err)
-	}
 }
 
 func TestEnableIPv6(t *testing.T) {
@@ -845,6 +865,16 @@ func TestEnableIPv6(t *testing.T) {
 	_, err = ep1.Join(containerID,
 		libnetwork.JoinOptionResolvConfPath(resolvConfPath))
 
+	if err != nil {
+		t.Fatal(err)
+	}
+	defer func() {
+		err = ep1.Leave(containerID)
+		if err != nil {
+			t.Fatal(err)
+		}
+	}()
+
 	content, err := ioutil.ReadFile(resolvConfPath)
 	if err != nil {
 		t.Fatal(err)
@@ -894,6 +924,15 @@ func TestNoEnableIPv6(t *testing.T) {
 
 	_, err = ep1.Join(containerID,
 		libnetwork.JoinOptionResolvConfPath(resolvConfPath))
+	if err != nil {
+		t.Fatal(err)
+	}
+	defer func() {
+		err = ep1.Leave(containerID)
+		if err != nil {
+			t.Fatal(err)
+		}
+	}()
 
 	content, err := ioutil.ReadFile(resolvConfPath)
 	if err != nil {