Spec tests for updateClientMetadataAndTryBecomePrimary and 'Lookup mutation documents' recovery

Author: schmidt-sebastian

packages/firestore/test/unit/specs/recovery_spec.test.ts

@@ -57,6 +57,28 @@ describeSpec('Persistence Recovery', ['no-ios', 'no-android'], () => {
     }
   );
 
+  specTest(
+    'Clients fail to lookup mutations (with recovery)',
+    ['multi-client'],
+    () => {
+      return client(0)
+        .expectPrimaryState(true)
+        .failDatabaseTransactions('Lookup mutation documents')
+        .client(1)
+        .failDatabaseTransactions('Lookup mutation documents')
+        .expectPrimaryState(false)
+        .userSets('collection/a', { v: 1 })
+        .client(0)
+        .recoverDatabase()
+        .runTimer(TimerId.AsyncQueueRetry)
+        .writeAcks('collection/a', 1, { expectUserCallback: false })
+        .client(1)
+        .recoverDatabase()
+        .runTimer(TimerId.AsyncQueueRetry)
+        .expectUserCallbacks({ acknowledged: ['collection/a'] });
+    }
+  );
+
   specTest(
     'Query raises events in secondary client  (with recovery)',
     ['multi-client'],
@@ -114,6 +136,31 @@ describeSpec('Persistence Recovery', ['no-ios', 'no-android'], () => {
     }
   );
 
+  specTest(
+    'Ignores intermittent lease refresh failures (with recovery)',
+    ['multi-client'],
+    () => {
+      return client(0)
+        .expectPrimaryState(true)
+        .client(1)
+        .expectPrimaryState(false)
+        .client(0)
+        .failDatabaseTransactions('updateClientMetadataAndTryBecomePrimary')
+        .runTimer(TimerId.ClientMetadataRefresh)
+        .client(1)
+        .failDatabaseTransactions('updateClientMetadataAndTryBecomePrimary')
+        .runTimer(TimerId.ClientMetadataRefresh)
+        .client(0)
+        .recoverDatabase()
+        .runTimer(TimerId.ClientMetadataRefresh)
+        .expectPrimaryState(true)
+        .client(1)
+        .recoverDatabase()
+        .runTimer(TimerId.ClientMetadataRefresh)
+        .expectPrimaryState(false);
+    }
+  );
+
   specTest('Recovers when write cannot be persisted', [], () => {
     return (
       spec()