Fix tests

alvaroaleman · Jun 21, 2020 · 4da76a6 · 4da76a6
1 parent 0928844
commit 4da76a6
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/pkg/apis/machine/v1beta1/machineset_types_test.go b/pkg/apis/machine/v1beta1/machineset_types_test.go
@@ -58,8 +58,8 @@ func TestStorageMachineSet(t *testing.T) {
 	done := make(chan struct{})
 	stopped := make(chan struct{})
 	go func() {
+		defer close(stopped)
 		gs.Expect(mgr.Start(done)).To(Succeed())
-		close(stopped)
 	}()
 	defer func() {
 		close(done)

diff --git a/pkg/apis/machine/v1beta1/machineset_webhook_test.go b/pkg/apis/machine/v1beta1/machineset_webhook_test.go
@@ -574,8 +574,8 @@ func TestMachineSetUpdate(t *testing.T) {
 			done := make(chan struct{})
 			stopped := make(chan struct{})
 			go func() {
+				defer close(stopped)
 				gs.Expect(mgr.Start(done)).To(Succeed())
-				close(stopped)
 			}()
 			defer func() {
 				close(done)
@@ -694,8 +694,8 @@ func TestCPMachineSetDelete(t *testing.T) {
 			done := make(chan struct{})
 			stopped := make(chan struct{})
 			go func() {
+				defer close(stopped)
 				gs.Expect(mgr.Start(done)).To(Succeed())
-				close(stopped)
 			}()
 			defer func() {
 				close(done)
@@ -876,8 +876,8 @@ func TestCPMachineSetUpdate(t *testing.T) {
 			done := make(chan struct{})
 			stopped := make(chan struct{})
 			go func() {
+				defer close(stopped)
 				gs.Expect(mgr.Start(done)).To(Succeed())
-				close(stopped)
 			}()
 			defer func() {
 				close(done)