perform the healthCheck assignment after checking the stream state.

Author: ehsannas

packages/firestore/src/remote/persistent_stream.ts

@@ -475,7 +475,15 @@ export abstract class PersistentStream<
     this.stream = this.startRpc(token);
     this.stream.onOpen(() => {
       dispatchIfNotClosed(() => {
-        debugAssert(this.healthCheck === null, 'Expected healthCheck to be null');
+        debugAssert(
+          this.state === PersistentStreamState.Starting,
+          'Expected stream to be in state Starting, but was ' + this.state
+        );
+        this.state = PersistentStreamState.Open;
+        debugAssert(
+          this.healthCheck === null,
+          'Expected healthCheck to be null'
+        );
         this.healthCheck = this.queue.enqueueAfterDelay(
           this.healthTimerId,
           HEALTHY_TIMEOUT_MS,
@@ -486,11 +494,6 @@ export abstract class PersistentStream<
             return Promise.resolve();
           }
         );
-        debugAssert(
-          this.state === PersistentStreamState.Starting,
-          'Expected stream to be in state Starting, but was ' + this.state
-        );
-        this.state = PersistentStreamState.Open;
         return this.listener!.onOpen();
       });
     });