1
0
mirror of https://github.com/laurent22/joplin.git synced 2024-12-12 08:54:00 +02:00
joplin/packages/lib/testing/syncTargetUtils.js
2021-05-21 15:17:21 +02:00

138 lines
3.8 KiB
JavaScript

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,
};