storage - add test for storage main service

src/vs/platform/storage/test/electron-main/storageMainService.test.ts

@@ -0,0 +1,99 @@
+/*---------------------------------------------------------------------------------------------
+ *  Copyright (c) Microsoft Corporation. All rights reserved.
+ *  Licensed under the MIT License. See License.txt in the project root for license information.
+ *--------------------------------------------------------------------------------------------*/
+
+import { promises } from 'fs';
+import { tmpdir } from 'os';
+import { strictEqual } from 'assert';
+import { URI } from 'vs/base/common/uri';
+import { rimraf } from 'vs/base/node/pfs';
+import { flakySuite, getRandomTestPath } from 'vs/base/test/node/testUtils';
+import { OPTIONS, parseArgs } from 'vs/platform/environment/node/argv';
+import { NativeEnvironmentService } from 'vs/platform/environment/node/environmentService';
+import { NullLogService } from 'vs/platform/log/common/log';
+import { IStorageMainService, StorageMainService } from 'vs/platform/storage/electron-main/storageMainService';
+import { instanceStorageKey } from 'vs/platform/telemetry/common/telemetry';
+import { IStorageChangeEvent, IStorageMain } from 'vs/platform/storage/electron-main/storageMain';
+import { generateUuid } from 'vs/base/common/uuid';
+import { isWindows } from 'vs/base/common/platform';
+
+flakySuite('StorageMainService (native)', function () {
+
+	class StorageTestEnvironmentService extends NativeEnvironmentService {
+
+		constructor(private globalStorageFolderPath: URI, private _extensionsPath: string) {
+			super(parseArgs(process.argv, OPTIONS));
+		}
+
+		get globalStorageHome(): URI {
+			return this.globalStorageFolderPath;
+		}
+
+		get extensionsPath(): string {
+			return this._extensionsPath;
+		}
+	}
+
+	let testDir: string;
+	let storageMainService: IStorageMainService;
+
+	setup(async () => {
+		testDir = getRandomTestPath(tmpdir(), 'vsctests', 'storageMainService');
+
+		await promises.mkdir(testDir, { recursive: true });
+
+		storageMainService = new StorageMainService(new NullLogService(), new StorageTestEnvironmentService(URI.file(testDir), testDir));
+	});
+
+	teardown(async () => {
+		await storageMainService.globalStorage.close();
+
+		return rimraf(testDir);
+	});
+
+	async function testStorage(storage: IStorageMain, isGlobal: boolean): Promise<void> {
+
+		// Telemetry: added after init
+		if (isGlobal) {
+			strictEqual(storageMainService.globalStorage.items.size, 0);
+			strictEqual(storageMainService.globalStorage.get(instanceStorageKey), undefined);
+			await storageMainService.globalStorage.initialize();
+			strictEqual(typeof storageMainService.globalStorage.get(instanceStorageKey), 'string');
+		}
+
+		let storageChangeEvent: IStorageChangeEvent | undefined = undefined;
+		const listener = storageMainService.globalStorage.onDidChangeStorage(e => {
+			storageChangeEvent = e;
+		});
+
+		// Basic store/get/remove
+		const size = storageMainService.globalStorage.items.size;
+
+		storageMainService.globalStorage.store('bar', 'foo');
+		strictEqual(storageChangeEvent!.key, 'bar');
+		storageMainService.globalStorage.store('barNumber', 55);
+		storageMainService.globalStorage.store('barBoolean', true);
+
+		strictEqual(storageMainService.globalStorage.get('bar'), 'foo');
+		strictEqual(storageMainService.globalStorage.getNumber('barNumber'), 55);
+		strictEqual(storageMainService.globalStorage.getBoolean('barBoolean'), true);
+
+		strictEqual(storageMainService.globalStorage.items.size, size + 3);
+
+		storageMainService.globalStorage.remove('bar');
+		strictEqual(storageMainService.globalStorage.get('bar'), undefined);
+
+		strictEqual(storageMainService.globalStorage.items.size, size + 2);
+
+		listener.dispose();
+	}
+
+	test('basics (global)', async function () {
+		testStorage(storageMainService.globalStorage, true);
+	});
+
+	test('basics (workspace)', async function () {
+		testStorage(storageMainService.workspaceStorage({ id: generateUuid(), uri: URI.file(isWindows ? 'C:\\testWorkspace' : '/testWorkspace') }), false);
+	});
+});