From c3ec73c79137f36c178e0585f40cf30c07cdf065 Mon Sep 17 00:00:00 2001
From: Steve Gordon <sgordon@hotmail.co.uk>
Date: Tue, 13 Jul 2021 15:44:56 +0100
Subject: [PATCH] Add conditional assertions to status check (#5873)

---
 .../XPack/Sql/SqlSearchApiCoordinatedTests.cs    | 16 +++++++++++++---
 1 file changed, 13 insertions(+), 3 deletions(-)

diff --git a/tests/Tests/XPack/Sql/SqlSearchApiCoordinatedTests.cs b/tests/Tests/XPack/Sql/SqlSearchApiCoordinatedTests.cs
index 5979cb0d39f..1154be168fd 100644
--- a/tests/Tests/XPack/Sql/SqlSearchApiCoordinatedTests.cs
+++ b/tests/Tests/XPack/Sql/SqlSearchApiCoordinatedTests.cs
@@ -114,10 +114,20 @@ [I] public async Task SqlSearchStatusResponse() => await Assert<SqlSearchStatusR
 		{
 			r.ShouldBeValid();
 			r.Id.Should().NotBeNullOrEmpty();
-			r.IsPartial.Should().BeTrue();
-			r.IsRunning.Should().BeTrue();
+
+			if (r.CompletionStatus.HasValue)
+			{
+				r.IsPartial.Should().BeFalse();
+				r.IsRunning.Should().BeFalse();
+			}
+			else
+			{
+				r.IsPartial.Should().BeTrue();
+				r.IsRunning.Should().BeTrue();
+				r.StartTimeInMillis.Should().BeGreaterThan(0);
+			}
+			
 			r.ExpirationTimeInMillis.Should().BeGreaterThan(0);
-			r.StartTimeInMillis.Should().BeGreaterThan(0);
 		});
 
 		[I] public async Task SqlGetResponse() => await Assert<SqlGetResponse>(GetStep, r =>