google-drive-mock 1.0.8 → 1.0.10

package/dist/mappers.js

@@ -30,6 +30,8 @@ function toV2File(file) {
             isRoot: false // Mock simplification
         })),
         version: file.version,
+        fileSize: file.size,
+        md5Checksum: file.md5Checksum,
         downloadUrl: `http://localhost/drive/v2/files/${file.id}?alt=media`
     };
 }

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/dist/store.d.ts

@@ -9,6 +9,8 @@ export interface DriveFile {
     trashed: boolean;
     createdTime: string;
     modifiedTime: string;
+    size: string;
+    md5Checksum: string;
     [key: string]: unknown;
 }
 export interface DriveChange {
@@ -38,6 +40,7 @@ export declare class DriveStore {
     private files;
     private changes;
     constructor();
+    private calculateStats;
     createFile(file: Partial<DriveFile> & {
         name: string;
     }): DriveFile;

package/dist/store.js

@@ -1,18 +1,71 @@
 "use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.driveStore = exports.DriveStore = void 0;
+const crypto = __importStar(require("crypto"));
 class DriveStore {
     constructor() {
         this.files = new Map();
         this.changes = [];
     }
+    calculateStats(content) {
+        let buffer;
+        if (typeof content === 'string') {
+            buffer = Buffer.from(content);
+        }
+        else if (content === undefined || content === null) {
+            buffer = Buffer.from('');
+        }
+        else {
+            buffer = Buffer.from(JSON.stringify(content));
+        }
+        return {
+            size: String(buffer.length),
+            md5Checksum: crypto.createHash('md5').update(buffer).digest('hex')
+        };
+    }
     createFile(file) {
         if (!file.name) {
             throw new Error("File name is required");
         }
         const id = file.id || Math.random().toString(36).substring(7);
         const now = new Date().toISOString();
-        const newFile = Object.assign(Object.assign({ kind: "drive#file", mimeType: "application/octet-stream", trashed: false, createdTime: now, modifiedTime: now }, file), { id, version: 1, etag: "1" });
+        const stats = this.calculateStats(file.content);
+        const newFile = Object.assign(Object.assign({ kind: "drive#file", mimeType: "application/octet-stream", trashed: false, createdTime: now, modifiedTime: now }, file), { id, version: 1, etag: "1", 
+            // Ensure calculated stats override provided ones
+            size: stats.size, md5Checksum: stats.md5Checksum });
         this.files.set(id, newFile);
         this.addChange(newFile);
         return newFile;
@@ -21,9 +74,14 @@ class DriveStore {
         const file = this.files.get(id);
         if (!file)
             return null;
+        // If content is being updated, recalculate stats
+        let statsUpdates = {};
+        if (updates.content !== undefined) {
+            statsUpdates = this.calculateStats(updates.content);
+        }
         // Merge updates and increment version
         const newVersion = file.version + 1;
-        const updatedFile = Object.assign(Object.assign(Object.assign({}, file), updates), { version: newVersion, etag: String(newVersion), modifiedTime: new Date().toISOString() });
+        const updatedFile = Object.assign(Object.assign(Object.assign(Object.assign({}, file), updates), statsUpdates), { version: newVersion, etag: String(newVersion), modifiedTime: new Date().toISOString() });
         this.files.set(id, updatedFile);
         this.addChange(updatedFile);
         return updatedFile;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "google-drive-mock",
-  "version": "1.0.8",
+  "version": "1.0.10",
   "description": "Mock-Server that simulates being google-drive. Used for testing.",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
@@ -12,7 +12,7 @@
     "test:slow": "LATENCY=20 vitest run",
     "test:browser": "start-server-and-test dev http://localhost:3000 'TEST_TARGET=mock BROWSER_ENABLED=true vitest run --browser --no-file-parallelism'",
     "test:browser:real": "TEST_TARGET=real BROWSER_ENABLED=true vitest run --browser",
-    "test:real": "TEST_TARGET=real vitest run",
+    "test:real": "ts-node scripts/check-token.ts && TEST_TARGET=real vitest run",
     "example:login": "ts-node examples/serve-login.ts",
     "lint": "eslint .",
     "lint:fix": "eslint . --fix"

package/scripts/check-token.ts

@@ -0,0 +1,75 @@
+import * as fs from 'fs';
+import * as path from 'path';
+import * as https from 'https';
+
+// Load .ENV manually to avoid devDependency issues if dotenv isn't available in this context, 
+// though project seems to use it.
+const envPath = path.resolve(__dirname, '../.ENV');
+if (fs.existsSync(envPath)) {
+    const envContent = fs.readFileSync(envPath, 'utf8');
+    envContent.split('\n').forEach(line => {
+        const match = line.match(/^([^=]+)=(.*)$/);
+        if (match) {
+            const key = match[1].trim();
+            const value = match[2].trim().replace(/^['"]|['"]$/g, ''); // strip quotes
+            if (!process.env[key]) {
+                process.env[key] = value;
+            }
+        }
+    });
+}
+
+const token = process.env.GDRIVE_TOKEN;
+
+if (!token) {
+    console.error('❌ Error: GDRIVE_TOKEN not found in environment or .ENV file.');
+    process.exit(1);
+}
+
+// Simple check only if running real tests (though script is likely invoked specifically for that)
+// The user asked to run this BEFORE test:real.
+
+console.log('🔄 Verifying GDRIVE_TOKEN...');
+
+const options = {
+    hostname: 'www.googleapis.com',
+    path: '/drive/v3/about?fields=user',
+    method: 'GET',
+    headers: {
+        'Authorization': `Bearer ${token}`,
+        'User-Agent': 'node-script'
+    }
+};
+
+const req = https.request(options, (res) => {
+    let data = '';
+
+    res.on('data', (chunk) => {
+        data += chunk;
+    });
+
+    res.on('end', () => {
+        if (res.statusCode === 200) {
+            try {
+                const body = JSON.parse(data);
+                console.log(`✅ Token is valid. User: ${body.user?.emailAddress || 'Unknown'}`);
+                process.exit(0);
+            } catch (e: unknown) {
+                const msg = e instanceof Error ? e.message : String(e);
+                console.error('❌ Error parsing response:', msg);
+                process.exit(1);
+            }
+        } else {
+            console.error(`❌ Token verification failed. Tell the human to update the .ENV file with a valid token. Status: ${res.statusCode}`);
+            console.error('Response:', data);
+            process.exit(1);
+        }
+    });
+});
+
+req.on('error', (e) => {
+    console.error(`❌ Request error: ${e.message}`);
+    process.exit(1);
+});
+
+req.end();

package/src/mappers.ts

@@ -28,6 +28,8 @@ export function toV2File(file: DriveFile): Record<string, unknown> {
             isRoot: false // Mock simplification
         })),
         version: file.version,
+        fileSize: file.size,
+        md5Checksum: file.md5Checksum,
         downloadUrl: `http://localhost/drive/v2/files/${file.id}?alt=media`
     };
 }

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/src/store.ts

@@ -1,3 +1,5 @@
+import * as crypto from 'crypto';
+
 export interface DriveFile {
     id: string;
     name: string;
@@ -9,6 +11,8 @@ export interface DriveFile {
     trashed: boolean;
     createdTime: string;
     modifiedTime: string;
+    size: string;
+    md5Checksum: string;
     [key: string]: unknown;
 }
 
@@ -46,22 +50,45 @@ export class DriveStore {
         this.changes = [];
     }
 
+    private calculateStats(content: unknown): { size: string, md5Checksum: string } {
+        let buffer: Buffer;
+        if (typeof content === 'string') {
+            buffer = Buffer.from(content);
+        } else if (content === undefined || content === null) {
+            buffer = Buffer.from('');
+        } else {
+            buffer = Buffer.from(JSON.stringify(content));
+        }
+
+        return {
+            size: String(buffer.length),
+            md5Checksum: crypto.createHash('md5').update(buffer).digest('hex')
+        };
+    }
+
     createFile(file: Partial<DriveFile> & { name: string }): DriveFile {
         if (!file.name) {
             throw new Error("File name is required");
         }
         const id = file.id || Math.random().toString(36).substring(7);
         const now = new Date().toISOString();
+
+        const stats = this.calculateStats(file.content);
+
         const newFile: DriveFile = {
             kind: "drive#file",
             mimeType: "application/octet-stream",
             trashed: false,
             createdTime: now,
             modifiedTime: now,
+
             ...file,
             id,
             version: 1, // Initialize version
             etag: "1", // Initialize etag
+            // Ensure calculated stats override provided ones
+            size: stats.size,
+            md5Checksum: stats.md5Checksum
         };
 
         this.files.set(id, newFile);
@@ -73,11 +100,18 @@ export class DriveStore {
         const file = this.files.get(id);
         if (!file) return null;
 
+        // If content is being updated, recalculate stats
+        let statsUpdates = {};
+        if (updates.content !== undefined) {
+            statsUpdates = this.calculateStats(updates.content);
+        }
+
         // Merge updates and increment version
         const newVersion = file.version + 1;
         const updatedFile = {
             ...file,
             ...updates,
+            ...statsUpdates,
             version: newVersion,
             etag: String(newVersion),
             modifiedTime: new Date().toISOString()

package/test/mime_types.test.ts

@@ -54,7 +54,7 @@ describe('MIME Type Handling', () => {
         expect(await getRes.json()).toEqual(content);
     });
 
-    it('should update MIME type and return new Content-Type header', async () => {
+    it('should update MIME type and return new Content-Type header', { timeout: 10000 }, async () => {
         // Create as text/plain
         const createRes = await fetch(`${config.baseUrl}/drive/v3/files`, {
             method: 'POST',

package/test/missing_fields.test.ts

@@ -0,0 +1,123 @@
+import { describe, it, expect, beforeAll, afterAll } from 'vitest';
+import { getTestConfig, TestConfig } from './config';
+
+describe('Missing Fields Support (Size & MD5)', () => {
+    let config: TestConfig;
+
+    beforeAll(async () => {
+        config = await getTestConfig();
+    });
+
+    afterAll(() => {
+        if (config) config.stop();
+    });
+
+    it('should return size and md5Checksum in V3 GET', async () => {
+        const content = JSON.stringify({ foo: 'bar' });
+        const metadata = { name: 'V3 Size Test', mimeType: 'application/json' };
+
+        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(metadata) +
+            delimiter +
+            'Content-Type: application/json\r\n\r\n' +
+            content +
+            closeDelim;
+
+        const createRes = await fetch(`${config.baseUrl}/upload/drive/v3/files?uploadType=multipart`, {
+            method: 'POST',
+            headers: {
+                'Authorization': `Bearer ${config.token}`,
+                'Content-Type': `multipart/related; boundary="${boundary}"`
+            },
+            body: body
+        });
+
+        expect(createRes.status).toBe(200);
+        const file = await createRes.json();
+
+        const fields = 'id,name,size,md5Checksum';
+        const getRes = await fetch(`${config.baseUrl}/drive/v3/files/${file.id}?fields=${fields}`, {
+            headers: { 'Authorization': `Bearer ${config.token}` }
+        });
+        expect(getRes.status).toBe(200);
+        const getFile = await getRes.json();
+
+        expect(getFile.size).toBeDefined();
+        // size should be string in V3
+        expect(String(getFile.size)).toBe(String(content.length));
+        expect(getFile.md5Checksum).toBeDefined();
+    });
+
+    it('should return fileSize and md5Checksum in V2 GET', async () => {
+        const content = 'V2 Content';
+
+        // Use simple upload for V2 to ensure content and metadata
+        const uploadRes = await fetch(`${config.baseUrl}/upload/drive/v2/files?uploadType=media`, {
+            method: 'POST',
+            headers: {
+                'Authorization': `Bearer ${config.token}`,
+                'Content-Type': 'text/plain'
+            },
+            body: content
+        });
+        expect(uploadRes.status).toBe(200);
+        const file = await uploadRes.json();
+
+        expect(file.fileSize).toBeDefined(); // V2 uses fileSize
+        expect(Number(file.fileSize)).toBe(content.length);
+        expect(file.md5Checksum).toBeDefined();
+
+        // Verify GET
+        const getRes = await fetch(`${config.baseUrl}/drive/v2/files/${file.id}`, {
+            headers: { 'Authorization': `Bearer ${config.token}` }
+        });
+        const getFile = await getRes.json();
+        expect(getFile.fileSize).toBeDefined();
+        expect(Number(getFile.fileSize)).toBe(content.length);
+        expect(getFile.md5Checksum).toBeDefined();
+    });
+
+    it('should return correct size and md5Checksum for EMPTY file', async () => {
+        const content = '';
+        const md5Empty = 'd41d8cd98f00b204e9800998ecf8427e'; // MD5 of empty string
+
+        // Upload empty file via V2 media upload
+        const uploadRes = await fetch(`${config.baseUrl}/upload/drive/v2/files?uploadType=media`, {
+            method: 'POST',
+            headers: {
+                'Authorization': `Bearer ${config.token}`,
+                'Content-Type': 'text/plain'
+            },
+            body: content
+        });
+        expect(uploadRes.status).toBe(200);
+        const file = await uploadRes.json();
+
+        // Check response
+        expect(file.fileSize).toBeDefined();
+        expect(Number(file.fileSize)).toBe(0);
+        expect(file.md5Checksum).toBe(md5Empty);
+
+        // Verify GET V2
+        const getV2 = await fetch(`${config.baseUrl}/drive/v2/files/${file.id}`, {
+            headers: { 'Authorization': `Bearer ${config.token}` }
+        });
+        const v2File = await getV2.json();
+        expect(Number(v2File.fileSize)).toBe(0);
+        expect(v2File.md5Checksum).toBe(md5Empty);
+
+        // Verify GET V3
+        const fields = 'id,name,size,md5Checksum';
+        const getV3 = await fetch(`${config.baseUrl}/drive/v3/files/${file.id}?fields=${fields}`, {
+            headers: { 'Authorization': `Bearer ${config.token}` }
+        });
+        const v3File = await getV3.json();
+        expect(Number(v3File.size)).toBe(0);
+        expect(v3File.md5Checksum).toBe(md5Empty);
+    });
+});

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');
+    });
 });