mgmtfn/k8splugin: fix test setup

Signed-off-by: Cristian Staretu <[email protected]>
contiv · Jul 5, 2017 · fccea5c · fccea5c
1 parent a9dde45
commit fccea5c
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/mgmtfn/k8splugin/driver_test.go b/mgmtfn/k8splugin/driver_test.go
@@ -55,7 +55,10 @@ func (s *NetSetup) SetUpTest(c *C) {
 	}
 	s.globalNS = globalNS
 	defer func() {
-		netns.Set(globalNS)
+		currNS, _ := netns.Get()
+		if !globalNS.Equal(currNS) {
+			netns.Set(globalNS)
+		}
 	}()
 
 	newNS, err := netns.New()