joplin/packages/app-cli/tests/services_KvStore.js

/* eslint-disable no-unused-vars */


const { asyncTest, fileContentEqual, setupDatabase, revisionService, setupDatabaseAndSynchronizer, db, synchronizer, fileApi, sleep, clearDatabase, switchClient, syncTargetId, objectsEqual, checkThrowAsync } = require('./test-utils.js');
const KvStore = require('@joplin/lib/services/KvStore').default;

process.on('unhandledRejection', (reason, p) => {
	console.log('Unhandled Rejection at: Promise', p, 'reason:', reason);
});

function setupStore() {
	const store = KvStore.instance();
	store.setDb(db());
	return store;
}

describe('services_KvStore', function() {

	beforeEach(async (done) => {
		await setupDatabaseAndSynchronizer(1);
		await switchClient(1);
		done();
	});

	it('should set and get values', asyncTest(async () => {
		const store = setupStore();
		await store.setValue('a', 123);
		expect(await store.value('a')).toBe(123);

		await store.setValue('a', 123);
		expect(await store.countKeys()).toBe(1);
		expect(await store.value('a')).toBe(123);

		await store.setValue('a', 456);
		expect(await store.countKeys()).toBe(1);
		expect(await store.value('a')).toBe(456);

		await store.setValue('b', 789);
		expect(await store.countKeys()).toBe(2);
		expect(await store.value('a')).toBe(456);
		expect(await store.value('b')).toBe(789);
	}));

	it('should set and get values with the right type', asyncTest(async () => {
		const store = setupStore();
		await store.setValue('string', 'something');
		await store.setValue('int', 123);
		expect(await store.value('string')).toBe('something');
		expect(await store.value('int')).toBe(123);
	}));

	it('should increment values', asyncTest(async () => {
		const store = setupStore();
		await store.setValue('int', 1);
		const newValue = await store.incValue('int');

		expect(newValue).toBe(2);
		expect(await store.value('int')).toBe(2);

		expect(await store.incValue('int2')).toBe(1);
		expect(await store.countKeys()).toBe(2);
	}));

	it('should handle non-existent values', asyncTest(async () => {
		const store = setupStore();
		expect(await store.value('nope')).toBe(null);
	}));

	it('should delete values', asyncTest(async () => {
		const store = setupStore();
		await store.setValue('int', 1);
		expect(await store.countKeys()).toBe(1);
		await store.deleteValue('int');
		expect(await store.countKeys()).toBe(0);

		await store.deleteValue('int'); // That should not throw
	}));

	it('should increment in an atomic way', asyncTest(async () => {
		const store = setupStore();
		await store.setValue('int', 0);

		const promises = [];
		for (let i = 0; i < 20; i++) {
			promises.push(store.incValue('int'));
		}

		await Promise.all(promises);

		expect(await store.value('int')).toBe(20);
	}));

	it('should search by prefix', asyncTest(async () => {
		const store = setupStore();
		await store.setValue('testing:1', 1);
		await store.setValue('testing:2', 2);

		const results = await store.searchByPrefix('testing:');
		expect(results.length).toBe(2);

		const numbers = results.map(r => r.value).sort();
		expect(numbers[0]).toBe(1);
		expect(numbers[1]).toBe(2);
	}));

});