From 9795880385f633815c9d7ce4e027365a25cab2f4 Mon Sep 17 00:00:00 2001
From: Alex Kristiansen <alex.kristiansen@elastic.co>
Date: Mon, 27 Feb 2023 10:56:18 -0800
Subject: [PATCH] fix position of done call in Log test

---
 controller/controller_client_test.go | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/controller/controller_client_test.go b/controller/controller_client_test.go
index d66e266..584eb9d 100644
--- a/controller/controller_client_test.go
+++ b/controller/controller_client_test.go
@@ -506,7 +506,6 @@ func TestUnitLogChange(t *testing.T) {
 				t.FailNow()
 			}
 			if gotStopped {
-				doneWaiter.Done()
 				return nil
 			}
 			// initial checkin
@@ -540,7 +539,7 @@ func TestUnitLogChange(t *testing.T) {
 				}
 			} else if unitsAreState(t, proto.State_HEALTHY, observed.Units) && gotHealthy {
 				// shut down
-				t.Logf("Got unit state healthy, stoppinng")
+				t.Logf("Got unit state healthy, stopping")
 				//check to see if log level has changed
 				observedLogLevel = logp.GetLevel()
 				unitOut.ConfigStateIdx++
@@ -552,7 +551,7 @@ func TestUnitLogChange(t *testing.T) {
 				}
 			} else if unitsAreState(t, proto.State_STOPPED, observed.Units) {
 				gotStopped = true
-				//doneWaiter.Done()
+				doneWaiter.Done()
 				t.Logf("Got unit state STOPPED, removing")
 				return &proto.CheckinExpected{
 					Units: []*proto.UnitExpected{},