Support new readSnapshots context info in fetchSnapshotByTimestamp

share · Feb 7, 2019 · 7e12683 · 7e12683
1 parent 15a438f
commit 7e12683
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 2 deletions.
diff --git a/lib/backend.js b/lib/backend.js
@@ -90,6 +90,7 @@ Backend.prototype.READ_SNAPSHOTS_METHODS = {
   fetch: 'fetch',
   fetchBulk: 'fetchBulk',
   fetchSnapshot: 'fetchSnapshot',
+  fetchSnapshotByTimestamp: 'fetchSnapshotByTimestamp',
   queryFetch: 'queryFetch',
   querySubscribe: 'querySubscribe'
 };
@@ -718,8 +719,16 @@ Backend.prototype.fetchSnapshotByTimestamp = function (agent, index, id, timesta
     if (error) return callback(error);
     var snapshotProjection = backend._getSnapshotProjection(backend.db, projection);
     var snapshots = [snapshot];
-    var snapshotType = backend.SNAPSHOT_TYPES.byTimestamp;
-    backend._sanitizeSnapshots(agent, snapshotProjection, collection, snapshots, snapshotType, function (error) {
+    var requestContext = new ReadSnapshotsContext(
+      backend.READ_SNAPSHOTS_METHODS.fetchSnapshotByTimestamp,
+      {
+        index: index,
+        id: id,
+        timestamp: timestamp,
+      },
+      backend.SNAPSHOT_TYPES.byTimestamp
+    );
+    backend._sanitizeSnapshots(agent, snapshotProjection, collection, snapshots, requestContext, function (error) {
       if (error) return callback(error);
       backend.emit('timing', 'fetchSnapshot', Date.now() - start, request);
       callback(null, snapshot);

diff --git a/test/client/snapshot-timestamp-request.js b/test/client/snapshot-timestamp-request.js
@@ -326,6 +326,8 @@ describe('SnapshotTimestampRequest', function () {
       it('triggers the middleware', function (done) {
         backend.use(backend.MIDDLEWARE_ACTIONS.readSnapshots,
           function (request) {
+            expect(request.method).to.eql(backend.READ_SNAPSHOTS_METHODS.fetchSnapshotByTimestamp);
+            expect(request.parameters).to.eql({index: 'books', id: 'time-machine', timestamp: day3});
             expect(request.snapshots[0]).to.eql(v3);
             expect(request.snapshotType).to.be(backend.SNAPSHOT_TYPES.byTimestamp);
             done();