Tools: Moved lib-specific tests under lib package

packages/lib/testing/syncTargetUtils.js

@@ -0,0 +1,137 @@
+const { syncDir, synchronizer, supportDir, loadEncryptionMasterKey, setupDatabaseAndSynchronizer, switchClient } = require('../testing/test-utils.js');
+const Setting = require('../models/Setting').default;
+const Folder = require('../models/Folder').default;
+const Note = require('../models/Note').default;
+const Tag = require('../models/Tag').default;
+const Resource = require('../models/Resource').default;
+const markdownUtils = require('../markdownUtils').default;
+const shim = require('../shim').default;
+const fs = require('fs-extra');
+
+const snapshotBaseDir = `${supportDir}/syncTargetSnapshots`;
+
+const testData = {
+	folder1: {
+		subFolder1: {},
+		subFolder2: {
+			note1: {
+				resource: true,
+				tags: ['tag1'],
+			},
+			note2: {},
+		},
+		note3: {
+			tags: ['tag1', 'tag2'],
+		},
+		note4: {
+			tags: ['tag2'],
+		},
+	},
+	folder2: {},
+	folder3: {
+		note5: {
+			resource: true,
+			tags: ['tag2'],
+		},
+	},
+};
+
+async function createTestData(data) {
+	async function recurseStruct(s, parentId = '') {
+		for (const n in s) {
+			if (n.toLowerCase().includes('folder')) {
+				const folder = await Folder.save({ title: n, parent_id: parentId });
+				await recurseStruct(s[n], folder.id);
+			} else {
+				const note = await Note.save({ title: n, parent_id: parentId });
+				if (s[n].resource) {
+					await shim.attachFileToNote(note, `${supportDir}/photo.jpg`);
+				}
+
+				if (s[n].tags) {
+					for (const tagTitle of s[n].tags) {
+						await Tag.addNoteTagByTitle(note.id, tagTitle);
+					}
+				}
+			}
+		}
+	}
+
+	await recurseStruct(data);
+}
+
+async function checkTestData(data) {
+	async function recurseCheck(s) {
+		for (const n in s) {
+			const obj = s[n];
+
+			if (n.toLowerCase().includes('folder')) {
+				const folder = await Folder.loadByTitle(n);
+				if (!folder) throw new Error(`Cannot load folder: ${n}`);
+				await recurseCheck(obj);
+			} else {
+				const note = await Note.loadByTitle(n);
+				if (!note) throw new Error(`Cannot load note: ${n}`);
+
+				const parent = await Folder.load(note.parent_id);
+				if (!parent) throw new Error(`Cannot load note parent: ${n}`);
+
+				if (obj.resource) {
+					const urls = markdownUtils.extractImageUrls(note.body);
+					const resourceId = urls[0].substr(2);
+					const resource = await Resource.load(resourceId);
+					if (!resource) throw new Error(`Cannot load note resource: ${n}`);
+				}
+
+				if (obj.tags) {
+					for (const tagTitle of obj.tags) {
+						const tag = await Tag.loadByTitle(tagTitle);
+						if (!tag) throw new Error(`Cannot load note tag: ${tagTitle}`);
+						const hasNote = await Tag.hasNote(tag.id, note.id);
+						if (!hasNote) throw new Error(`Tag not associated with note: ${tagTitle}`);
+					}
+				}
+			}
+		}
+	}
+
+	await recurseCheck(data);
+}
+
+async function deploySyncTargetSnapshot(syncTargetType, syncVersion) {
+	const sourceDir = `${snapshotBaseDir}/${syncVersion}/${syncTargetType}`;
+	await fs.remove(syncDir);
+	await fs.copy(sourceDir, syncDir);
+}
+
+async function main(syncTargetType) {
+	const validSyncTargetTypes = ['normal', 'e2ee'];
+	if (!validSyncTargetTypes.includes(syncTargetType)) throw new Error(`Sync target type must be: ${validSyncTargetTypes.join(', ')}`);
+
+	await setupDatabaseAndSynchronizer(1);
+	await switchClient(1);
+	await createTestData(testData);
+
+	if (syncTargetType === 'e2ee') {
+		Setting.setValue('encryption.enabled', true);
+		await loadEncryptionMasterKey();
+	}
+
+	await synchronizer().start();
+
+	if (!Setting.value('syncVersion')) throw new Error('syncVersion is not set');
+	const destDir = `${snapshotBaseDir}/${Setting.value('syncVersion')}/${syncTargetType}`;
+	await fs.mkdirp(destDir); // Create intermediate directories
+	await fs.remove(destDir);
+	await fs.mkdirp(destDir);
+	await fs.copy(syncDir, destDir);
+
+	console.info(`Sync target snapshot created in: ${destDir}`);
+}
+
+module.exports = {
+	checkTestData,
+	main,
+	testData,
+	deploySyncTargetSnapshot,
+};