set time

Signed-off-by: Tim Vaillancourt <tim@timvaillancourt.com>
vitessio · Oct 5, 2024 · 2038df5 · 2038df5
1 parent 42ce0f5
commit 2038df5
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/go/vt/discovery/healthcheck_test.go b/go/vt/discovery/healthcheck_test.go
@@ -149,6 +149,7 @@ func TestHealthCheck(t *testing.T) {
 	input := make(chan *querypb.StreamHealthResponse)
 	conn := createFakeConn(tablet, input)
 	now := time.Now()
+	hc.nowTimeFunc = func() time.Time { return now }
 
 	// create a channel and subscribe to healthcheck
 	resultChan := hc.Subscribe()
@@ -497,6 +498,7 @@ func TestHealthCheckErrorOnPrimaryAfterExternalReparent(t *testing.T) {
 		Serving:              true,
 		Stats:                &querypb.RealtimeStats{ReplicationLagSeconds: 0, CpuUsage: 0.2},
 		PrimaryTermStartTime: 20,
+		Timestamp:            protoutil.TimeToProto(now),
 	}}
 	a = hc.GetHealthyTabletStats(&querypb.Target{Keyspace: "k", Shard: "s", TabletType: topodatapb.TabletType_PRIMARY})
 	mustMatch(t, health, a, "unexpected result")