use getters in *_equal testutils

singholt · Aug 13, 2024 · 462272c · 462272c
1 parent c13cd70
commit 462272c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/agent/api/testutils/container_equal.go b/agent/api/testutils/container_equal.go
@@ -76,7 +76,7 @@ func ContainersEqual(lhs, rhs *apicontainer.Container) bool {
 	if !ContainerOverridesEqual(lhs.Overrides, rhs.Overrides) {
 		return false
 	}
-	if lhs.DesiredStatusUnsafe != rhs.DesiredStatusUnsafe || lhs.KnownStatusUnsafe != rhs.KnownStatusUnsafe {
+	if lhs.GetDesiredStatus() != rhs.GetDesiredStatus() || lhs.GetKnownStatus() != rhs.GetKnownStatus() {
 		return false
 	}
 	if lhs.AppliedStatus != rhs.AppliedStatus {

diff --git a/agent/api/testutils/task_equal.go b/agent/api/testutils/task_equal.go
@@ -41,7 +41,7 @@ func TasksEqual(lhs, rhs *apitask.Task) bool {
 		}
 	}
 
-	if lhs.DesiredStatusUnsafe != rhs.DesiredStatusUnsafe {
+	if lhs.GetDesiredStatus() != rhs.GetDesiredStatus() {
 		return false
 	}
 	if lhs.GetKnownStatus() != rhs.GetKnownStatus() {