Adding changes to the complete chassis reboot test

openconfig · Dec 4, 2024 · 4053a09 · 4053a09
1 parent e394152
commit 4053a09
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/feature/gnoi/system/tests/complete_chassis_reboot/complete_chassis_reboot_test.go b/feature/gnoi/system/tests/complete_chassis_reboot/complete_chassis_reboot_test.go
@@ -148,8 +148,18 @@ func TestChassisReboot(t *testing.T) {
 						t.Logf("Time elapsed %.2f seconds > %d reboot delay", time.Since(start).Seconds(), rebootDelay)
 						break
 					}
-					latestTime, err := time.Parse(time.RFC3339, gnmi.Get(t, dut, gnmi.OC().System().CurrentDatetime().State()))
-					if err != nil {
+					var latestTime time.Time
+					var err error
+					if errMsg := testt.CaptureFatal(t, func(t testing.TB) {
+						latestTime, err = time.Parse(time.RFC3339, gnmi.Get(t, dut, gnmi.OC().System().CurrentDatetime().State()))
+						if err != nil {
+							t.Fatalf("Failed parsing current-datetime: %s", err)
+						}
+					}); errMsg != nil && time.Since(start).Seconds() < rebootDelay {
+						t.Fatalf("Get request failed before reboot delay, %s", *errMsg)
+					}
+
+					if err != nil && time.Since(start).Seconds() < rebootDelay {
 						t.Fatalf("Failed parsing current-datetime: %s", err)
 					}
 					if latestTime.Before(prevTime) || latestTime.Equal(prevTime) {