Used utility function

Author: maneesht

packages/database-compat/test/helpers/util.ts

@@ -22,6 +22,7 @@ import '../../src/index';
 import firebase from '@firebase/app-compat';
 import { _FirebaseNamespace } from '@firebase/app-types/private';
 import { Component, ComponentType } from '@firebase/component';
+import { Deferred } from '@firebase/util';
 
 import { Path } from '../../../database/src/core/util/Path';
 import { Query, Reference } from '../../src/api/Reference';
@@ -175,3 +176,11 @@ export function canCreateExtraConnections() {
     typeof MozWebSocket !== 'undefined' || typeof WebSocket !== 'undefined'
   );
 }
+
+// TODO: Move this to @firebase/util
+export function timeoutResolve<T>(promise: Promise<T>, timeInMS = 2000) {
+  const deferredPromise = new Deferred<T>();
+  setTimeout(() => deferredPromise.reject('timeout!'), timeInMS);
+  promise.then(deferredPromise.resolve, deferredPromise.reject);
+  return deferredPromise.promise;
+}

packages/database-compat/test/query.test.ts

@@ -29,7 +29,7 @@ import {
 } from '../../database/test/helpers/EventAccumulator';
 import { DataSnapshot, Query, Reference } from '../src/api/Reference';
 
-import { getFreshRepo, getPath, getRandomNode, pause } from './helpers/util';
+import { getFreshRepo, getPath, getRandomNode, pause, timeoutResolve } from './helpers/util';
 
 use(chaiAsPromised);
 
@@ -3226,10 +3226,7 @@ describe('Query Tests', () => {
     const node = getRandomNode() as Reference;
     node.database.goOffline();
     try {
-      const getPromise = new Promise((resolve, reject) => {
-        setTimeout(reject, 2000);
-        node.get().then(resolve);
-      });
+      const getPromise = timeoutResolve(node.get());
       await expect(getPromise).to.eventually.be.rejected;
     } finally {
       node.database.goOnline();