google-drive-mock 1.0.8 → 1.0.9

package/dist/routes/v2.js

@@ -391,6 +391,14 @@ const createV2Router = (config) => {
             res.status(404).json({ error: { code: 404, message: "File not found" } });
             return;
         }
+        const ifMatchHeader = req.headers['if-match'];
+        const ifMatch = Array.isArray(ifMatchHeader) ? ifMatchHeader[0] : ifMatchHeader;
+        if (ifMatch && ifMatch !== '*' && ifMatch !== existingFile.etag) {
+            if (ifMatch !== existingFile.etag && ifMatch !== `"${existingFile.etag}"`) {
+                res.status(412).json({ error: { code: 412, message: "Precondition Failed" } });
+                return;
+            }
+        }
         const uploadType = req.query.uploadType;
         if (uploadType === 'media') {
             const rawBody = req.body;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "google-drive-mock",
-  "version": "1.0.8",
+  "version": "1.0.9",
   "description": "Mock-Server that simulates being google-drive. Used for testing.",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",

package/src/routes/v2.ts

@@ -454,6 +454,15 @@ export const createV2Router = (config: AppConfig) => {
             return;
         }
 
+        const ifMatchHeader = req.headers['if-match'];
+        const ifMatch = Array.isArray(ifMatchHeader) ? ifMatchHeader[0] : ifMatchHeader;
+        if (ifMatch && ifMatch !== '*' && ifMatch !== existingFile.etag) {
+            if (ifMatch !== existingFile.etag && ifMatch !== `"${existingFile.etag}"`) {
+                res.status(412).json({ error: { code: 412, message: "Precondition Failed" } });
+                return;
+            }
+        }
+
         const uploadType = req.query.uploadType as string;
 
         if (uploadType === 'media') {

package/test/v2_upload.test.ts

@@ -139,4 +139,103 @@ describe('V2 Upload Features', () => {
         });
         expect(await contentRes.text()).toBe(newContent);
     });
+    it('should respect If-Match header in V2 media upload (PUT)', async () => {
+        // 1. Create file
+        const createRes = await fetch(`${config.baseUrl}/drive/v2/files`, {
+            method: 'POST',
+            headers: {
+                'Authorization': `Bearer ${config.token}`,
+                'Content-Type': 'application/json'
+            },
+            body: JSON.stringify({ title: 'ETag Media Test', mimeType: 'text/plain' })
+        });
+        const file = await createRes.json();
+        const fileId = file.id;
+        const etag = file.etag;
+
+        const invalidEtag = etag ? etag.replace(/.$/, '0') : '"wrong-etag"';
+
+        // 2. Update with WRONG ETag
+        const wrongEtagRes = await fetch(`${config.baseUrl}/upload/drive/v2/files/${fileId}?uploadType=media`, {
+            method: 'PUT',
+            headers: {
+                'Authorization': `Bearer ${config.token}`,
+                'Content-Type': 'text/plain',
+                'If-Match': invalidEtag
+            },
+            body: 'Should Fail'
+        });
+        expect(wrongEtagRes.status).toBe(412);
+
+        // 3. Update with CORRECT ETag
+        const correctEtagRes = await fetch(`${config.baseUrl}/upload/drive/v2/files/${fileId}?uploadType=media`, {
+            method: 'PUT',
+            headers: {
+                'Authorization': `Bearer ${config.token}`,
+                'Content-Type': 'text/plain',
+                'If-Match': etag
+            },
+            body: 'Updated with ETag'
+        });
+        expect(correctEtagRes.status).toBe(200);
+
+        // Verify Content
+        const contentRes = await fetch(`${config.baseUrl}/drive/v2/files/${fileId}?alt=media`, {
+            headers: { 'Authorization': `Bearer ${config.token}` }
+        });
+        expect(await contentRes.text()).toBe('Updated with ETag');
+    });
+
+    it('should respect If-Match header in V2 multipart upload (PUT)', async () => {
+        // 1. Create file
+        const createRes = await fetch(`${config.baseUrl}/drive/v2/files`, {
+            method: 'POST',
+            headers: {
+                'Authorization': `Bearer ${config.token}`,
+                'Content-Type': 'application/json'
+            },
+            body: JSON.stringify({ title: 'ETag Multipart Test', mimeType: 'text/plain' })
+        });
+        const file = await createRes.json();
+        const fileId = file.id;
+        const etag = file.etag;
+
+        const boundary = 'foo_bar_baz';
+        const delimiter = `\r\n--${boundary}\r\n`;
+        const closeDelim = `\r\n--${boundary}--`;
+        const body = delimiter +
+            'Content-Type: application/json\r\n\r\n' +
+            JSON.stringify({ title: 'Updated Title' }) +
+            delimiter +
+            'Content-Type: text/plain\r\n\r\n' +
+            'Multipart Update' +
+            closeDelim;
+
+        const invalidEtag = etag ? etag.replace(/.$/, '0') : '"wrong-etag"';
+
+        // 2. Update with WRONG ETag
+        const wrongEtagRes = await fetch(`${config.baseUrl}/upload/drive/v2/files/${fileId}?uploadType=multipart`, {
+            method: 'PUT',
+            headers: {
+                'Authorization': `Bearer ${config.token}`,
+                'Content-Type': `multipart/related; boundary="${boundary}"`,
+                'If-Match': invalidEtag
+            },
+            body: body
+        });
+        expect(wrongEtagRes.status).toBe(412);
+
+        // 3. Update with CORRECT ETag
+        const correctEtagRes = await fetch(`${config.baseUrl}/upload/drive/v2/files/${fileId}?uploadType=multipart`, {
+            method: 'PUT',
+            headers: {
+                'Authorization': `Bearer ${config.token}`,
+                'Content-Type': `multipart/related; boundary="${boundary}"`,
+                'If-Match': etag
+            },
+            body: body
+        });
+        expect(correctEtagRes.status).toBe(200);
+        expect((await correctEtagRes.json()).title).toBe('Updated Title');
+    });
 });