test(vfs): comprehensive writeFile + delete/rename/stat/watch coverage (#365)

Co-authored-by: google-labs-jules[bot] <161369871+google-labs-jules[bot]@users.noreply.github.com>

Author: google-labs-jules[bot]

tests/unit/virtualFileSystem.test.ts

@@ -176,4 +176,141 @@ describe('SQLiteFileSystemProvider', () => {
             }, /FileNotFound/);
         });
     });
+    describe('writeFile', () => {
+        const provider = new SQLiteFileSystemProvider();
+        const docKey = 'test-doc';
+
+        it('should throw NoPermissions if writing to __create__.sql', async () => {
+            setupMockDocument(docKey, {});
+            const uri = vscode.Uri.parse(`vscode-sqlite://${docKey}/users/group/__create__.sql/create.sql`);
+            await assert.rejects(async () => {
+                await provider.writeFile(uri, new Uint8Array(), { create: false, overwrite: true });
+            }, (err: any) => err.message.includes('Cannot edit CREATE statement directly'));
+        });
+
+        it('should throw Unavailable for invalid row ID', async () => {
+            setupMockDocument(docKey, {});
+            const uri = vscode.Uri.parse(`vscode-sqlite://${docKey}/users/group/invalid-id/col.txt`);
+            await assert.rejects(async () => {
+                await provider.writeFile(uri, new Uint8Array(), { create: false, overwrite: true });
+            }, (err: any) => err.message.includes('Invalid Row ID'));
+        });
+
+        it('should map string error to Unavailable', async () => {
+            const dbOps = {
+                updateCell: mock.fn(async () => {
+                    throw 'Database write string error';
+                })
+            };
+            setupMockDocument(docKey, dbOps);
+
+            const uri = vscode.Uri.parse(`vscode-sqlite://${docKey}/users/group/1/col.txt`);
+
+            await assert.rejects(async () => {
+                await provider.writeFile(uri, new Uint8Array(), { create: false, overwrite: true });
+            }, (err: any) => err.message.includes('Database write string error'));
+        });
+
+        it('should map Error object to Unavailable', async () => {
+            const dbOps = {
+                updateCell: mock.fn(async () => {
+                    throw new Error('Database write error');
+                })
+            };
+            setupMockDocument(docKey, dbOps);
+
+            const uri = vscode.Uri.parse(`vscode-sqlite://${docKey}/users/group/1/col.txt`);
+
+            await assert.rejects(async () => {
+                await provider.writeFile(uri, new Uint8Array(), { create: false, overwrite: true });
+            }, (err: any) => err.message.includes('Database write error'));
+        });
+
+        it('should write text content correctly', async () => {
+            const dbOps = {
+                updateCell: mock.fn(async () => {})
+            };
+            const doc = setupMockDocument(docKey, dbOps);
+
+            const uri = vscode.Uri.parse(`vscode-sqlite://${docKey}/users/group/1/col.txt`);
+            const content = new TextEncoder().encode('Hello World');
+            await provider.writeFile(uri, content, { create: false, overwrite: true });
+
+            assert.strictEqual(dbOps.updateCell.mock.callCount(), 1);
+            assert.deepStrictEqual(dbOps.updateCell.mock.calls[0].arguments, ['users', 1, 'col', 'Hello World']);
+            assert.strictEqual(doc.recordExternalModification.mock.callCount(), 1);
+        });
+
+        it('should write binary content if not valid UTF-8', async () => {
+            const dbOps = {
+                updateCell: mock.fn(async () => {})
+            };
+            const doc = setupMockDocument(docKey, dbOps);
+
+            const uri = vscode.Uri.parse(`vscode-sqlite://${docKey}/users/group/1/col.txt`);
+            const content = new Uint8Array([0xff, 0xff, 0xff]); // Invalid UTF-8
+            await provider.writeFile(uri, content, { create: false, overwrite: true });
+
+            assert.strictEqual(dbOps.updateCell.mock.callCount(), 1);
+            assert.deepStrictEqual(dbOps.updateCell.mock.calls[0].arguments, ['users', 1, 'col', content]);
+            assert.strictEqual(doc.recordExternalModification.mock.callCount(), 1);
+        });
+    });
+
+    describe('delete / rename / stat / watch', () => {
+        const provider = new SQLiteFileSystemProvider();
+        const docKey = 'test-doc';
+
+        it('should throw NoPermissions for createDirectory', async () => {
+            const uri = vscode.Uri.parse(`vscode-sqlite://${docKey}/users/group`);
+            await assert.rejects(async () => {
+                await provider.createDirectory(uri);
+            }, (err: any) => err.message.includes('NoPermissions'));
+        });
+
+        it('should return empty array for readDirectory', async () => {
+            const uri = vscode.Uri.parse(`vscode-sqlite://${docKey}/users/group`);
+            const res = await provider.readDirectory(uri);
+            assert.deepStrictEqual(res, []);
+        });
+
+        it('stat should return generic file stat for normal cell', async () => {
+            const uri = vscode.Uri.parse(`vscode-sqlite://${docKey}/users/group/1/col.txt`);
+            setupMockDocument(docKey, {});
+            const s = await provider.stat(uri);
+            assert.strictEqual(s.type, vscode.FileType.File);
+            assert.strictEqual(s.size, 0);
+            assert.strictEqual(s.permissions, undefined);
+        });
+
+        it('stat should return generic file stat with Readonly for __create__.sql', async () => {
+            const uri = vscode.Uri.parse(`vscode-sqlite://${docKey}/users/group/__create__.sql/create.sql`);
+            setupMockDocument(docKey, {});
+            const s = await provider.stat(uri);
+            assert.strictEqual(s.type, vscode.FileType.File);
+            assert.strictEqual(s.size, 0);
+            assert.strictEqual(s.permissions, vscode.FilePermission.Readonly);
+        });
+
+        it('watch should return a generic Disposable', async () => {
+            const uri = vscode.Uri.parse(`vscode-sqlite://${docKey}/users/group/1/col.txt`);
+            const disposable = provider.watch(uri, { recursive: false, excludes: [] });
+            assert.ok(disposable.dispose);
+        });
+
+        it('should throw NoPermissions for delete', async () => {
+            const uri = vscode.Uri.parse(`vscode-sqlite://${docKey}/users/group/1/col.txt`);
+            await assert.rejects(async () => {
+                await provider.delete(uri, { recursive: false });
+            }, (err: any) => err.message.includes('NoPermissions'));
+        });
+
+        it('should throw NoPermissions for rename', async () => {
+            const uri1 = vscode.Uri.parse(`vscode-sqlite://${docKey}/users/group/1/col.txt`);
+            const uri2 = vscode.Uri.parse(`vscode-sqlite://${docKey}/users/group/1/col2.txt`);
+            await assert.rejects(async () => {
+                await provider.rename(uri1, uri2, { overwrite: true });
+            }, (err: any) => err.message.includes('NoPermissions'));
+        });
+    });
 });