Implement extendLifetime() method for shared diff repository

backend/src/v2/SharedDiffRepository.ts

@@ -1,7 +1,8 @@
 import {  SharedDiff } from './SharedDiff';
 
 export interface SharedDiffRepository {
-    insert: (diff: SharedDiff) => Promise<SharedDiff>;
-    fetchById: (id: string) => Promise<SharedDiff>;
-    deleteById: (id: string) => Promise<number>;
+    insert: (diff: SharedDiff)                     => Promise<SharedDiff>;
+    fetchById: (id: string)                        => Promise<SharedDiff>;
+    deleteById: (id: string)                       => Promise<number>;
+    extendLifetime: (id: string, noOfDays: number) => Promise<SharedDiff>;
 }

backend/src/v2/SharedDiffRepository/MongoSharedDiffRepository.ts

@@ -40,6 +40,25 @@ export class MongoSharedDiffRepository implements SharedDiffRepository {
             .then(doc => ({id: doc._id, ...doc}));
     }
 
+    extendLifetime(id: string, hours: number): Promise<SharedDiff> {
+        return this.fetchById(id)
+            .then(sharedDiff => new Date(sharedDiff.expiresAt.getTime() + (hours * 60 * 60 * 1000)))
+            .then(newExpiredDate => {
+                this.client
+                    .then(client => client.db(this.db_name))
+                    .then(db => db.collection(COLLECTION_NAME))
+                    .then(collection => collection.updateOne(
+                        { "_id": id },
+                        {
+                            $set: {
+                                expiresAt: newExpiredDate
+                            }
+                        }
+                    ))
+            })
+            .then(updateResult => this.fetchById(id));
+    }
+
     // returns a promise of how many items where deleted
     deleteById(id: string): Promise<number> {
         return this.client

backend/tests/v2/CreateSharedDiffAPIAction.test.ts

@@ -18,6 +18,7 @@ index 1456e89..e1da2da 100644
         insert: jest.fn(diff => Promise.resolve(diff)),
         fetchById: (id: string) => null,
         deleteById: (id: string) => Promise.resolve(0),
+        extendLifetime: jest.fn(diff => Promise.resolve(diff)),
     };
     const action = new CreateSharedDiffAPIAction(repo, metrics);
     expect(action).toBeDefined();
@@ -45,6 +46,7 @@ index 1456e89..e1da2da 100644
         insert: jest.fn((diff) => Promise.reject(new Error('fake error'))),
         fetchById: (id: string) => null,
         deleteById: (id: string) => Promise.resolve(0),
+        extendLifetime: jest.fn(diff => Promise.reject("random err")),
     };
     const action = new CreateSharedDiffAPIAction(repo, metrics);
     expect(action).toBeDefined();

backend/tests/v2/CreateSharedDiffAction.test.ts

@@ -18,6 +18,7 @@ index 1456e89..e1da2da 100644
         insert: jest.fn(diff => Promise.resolve(diff)),
         fetchById: (id: string) => null,
         deleteById: (id: string) => Promise.resolve(0),
+        extendLifetime: (id: string, noOfDays: number) => Promise.reject("random err"),
     };
     const action = new CreateSharedDiffAction(repo, metrics);
     expect(action).toBeDefined();
@@ -45,6 +46,7 @@ index 1456e89..e1da2da 100644
         insert: jest.fn((diff) => Promise.reject(new Error('fake error'))),
         fetchById: (id: string) => null,
         deleteById: (id: string) => Promise.resolve(0),
+        extendLifetime: (id: string, noOfDays: number) => Promise.reject("random err"),
     };
     const action = new CreateSharedDiffAction(repo, metrics);
     expect(action).toBeDefined();
@@ -72,6 +74,7 @@ index 1456e89..e1da2da 100644
         insert: jest.fn().mockReturnValueOnce(new Promise(() => {})),
         fetchById: (id: string) => null,
         deleteById: (id: string) => Promise.resolve(0),
+        extendLifetime: (id: string, noOfDays: number) => Promise.reject("random err"),
     };
     const action = new CreateSharedDiffAction(repo, metrics);
     const is_valid = action.isValidRawDiff(raw_diff);

backend/tests/v2/DeleteSharedDiffAction.test.ts

@@ -7,6 +7,7 @@ test('should create a DeleteSharedDiffAction and delete a SharedDiff by id', ()
         insert: jest.fn(),
         fetchById: jest.fn(),
         deleteById: (id: string) => Promise.resolve(1),
+        extendLifetime: (id: string, noOfDays: number) => Promise.reject("random err"),
     }
     const action = new DeleteSharedDiffAction(repo, metrics);
     expect(action).toBeDefined();

backend/tests/v2/GetSharedDiffAction.test.ts

@@ -17,6 +17,7 @@ index 1456e89..e1da2da 100644
         insert: jest.fn(),
         fetchById: (id: string) => Promise.resolve({id, ...makeSharedDiff(raw_diff)}),
         deleteById: (id: string) => Promise.resolve(0),
+        extendLifetime: (id: string, noOfDays: number) => Promise.reject("random err"),
     }
     const action = new GetSharedDiffAction(repo, metrics);
     expect(action).toBeDefined();

backend/tests/v2/SharedDiffRepository/MongoSharedDiffRepository.test.ts

@@ -64,4 +64,25 @@ describe('MongoSharedDiff tests', () => {
             .then(stored_diff => repo.deleteById(stored_diff.id))
             .then(deletedCount => expect(deletedCount).toEqual(1));
     });
+
+    test('Mongo test: extend lifetime of a SharedDiff', () => {
+        const raw_diff = `
+            diff --git a/file.json b/file.json
+            index 1456e89..e1da2da 100644
+            --- a/file.json
+            +++ b/file.json
+            @@ -1,1 +1,1 @@
+            -a
+            +b
+            `
+        const shared_diff = makeSharedDiff(raw_diff);
+        const msPerDay = 24 * 60 * 60 * 1000;
+        return repo.insert(shared_diff)
+            .then((diff) => {
+                expect(diff.expiresAt.getTime() - diff.created.getTime()).toBeLessThan(2 * msPerDay);
+                return diff;
+            })
+            .then(stored_diff => repo.extendLifetime(stored_diff.id, 24))
+            .then((diff) => expect(diff.expiresAt.getTime() - diff.created.getTime()).toBeGreaterThanOrEqual(2 * msPerDay));
+    });
 });