import * as models from '../../models'; import * as db from '../database'; import {globalBeforeEach} from '../../__jest__/before-each'; function loadFixture (name) { const fixtures = require(`../__fixtures__/${name}`).data; const promises = []; for (const type of Object.keys(fixtures)) { for (const doc of fixtures[type]) { promises.push(db.insert(Object.assign({}, doc, {type}))); } } return Promise.all(promises); } describe('init()', () => { beforeEach(globalBeforeEach); it('handles being initialized twice', async () => { await db.init(models.types(), {inMemoryOnly: true}); await db.init(models.types(), {inMemoryOnly: true}); expect((await db.all(models.request.type)).length).toBe(0); }); }); describe('onChange()', () => { beforeEach(globalBeforeEach); it('handles change listeners', async () => { const doc = { type: models.request.type, parentId: 'nothing', name: 'foo' }; const changesSeen = []; const callback = change => { changesSeen.push(change); }; db.onChange(callback); const newDoc = await models.request.create(doc); const updatedDoc = await models.request.update(newDoc, {name: 'bar'}); expect(changesSeen.length).toBe(2); expect(changesSeen).toEqual([ [[db.CHANGE_INSERT, newDoc, false]], [[db.CHANGE_UPDATE, updatedDoc, false]] ]); db.offChange(callback); await models.request.create(doc); expect(changesSeen.length).toBe(2); }); }); describe('bufferChanges()', () => { beforeEach(globalBeforeEach); it('properly buffers changes', async () => { const doc = { type: models.request.type, parentId: 'n/a', name: 'foo' }; const changesSeen = []; const callback = change => { changesSeen.push(change); }; db.onChange(callback); await db.bufferChanges(); const newDoc = await models.request.create(doc); const updatedDoc = await models.request.update(newDoc, true); // Assert no change seen before flush expect(changesSeen.length).toBe(0); // Assert changes seen after flush await db.flushChanges(); expect(changesSeen).toEqual([[ [db.CHANGE_INSERT, newDoc, false], [db.CHANGE_UPDATE, updatedDoc, false] ]]); // Assert no more changes seen after flush again await db.flushChanges(); expect(changesSeen).toEqual([[ [db.CHANGE_INSERT, newDoc, false], [db.CHANGE_UPDATE, updatedDoc, false] ]]); }); }); describe('requestCreate()', () => { beforeEach(globalBeforeEach); it('creates a valid request', async () => { const now = Date.now(); const patch = { name: 'My Request', parentId: 'wrk_123' }; const r = await models.request.create(patch); expect(Object.keys(r).length).toBe(20); expect(r._id).toMatch(/^req_[a-zA-Z0-9]{32}$/); expect(r.created).toBeGreaterThanOrEqual(now); expect(r.modified).toBeGreaterThanOrEqual(now); expect(r.type).toBe('Request'); expect(r.name).toBe('My Request'); expect(r.url).toBe(''); expect(r.method).toBe('GET'); expect(r.body).toEqual({}); expect(r.parameters).toEqual([]); expect(r.headers).toEqual([]); expect(r.authentication).toEqual({}); expect(r.metaSortKey).toBeLessThanOrEqual(-1 * now); expect(r.parentId).toBe('wrk_123'); }); it('throws when missing parentID', () => { const fn = () => models.request.create({name: 'My Request'}); expect(fn).toThrowError('New Requests missing `parentId`'); }); }); describe('requestGroupDuplicate()', () => { beforeEach(async () => { await globalBeforeEach(); await loadFixture('nestedfolders'); }); it('duplicates a RequestGroup', async () => { const requestGroup = await models.requestGroup.getById('fld_1'); expect(requestGroup.name).toBe('Fld 1'); const newRequestGroup = await models.requestGroup.duplicate(requestGroup); expect(newRequestGroup._id).not.toBe(requestGroup._id); expect(newRequestGroup.name).toBe('Fld 1 (Copy)'); const allRequests = await models.request.all(); const allRequestGroups = await models.requestGroup.all(); const childRequests = await models.request.findByParentId(requestGroup._id); const childRequestGroups = await models.requestGroup.findByParentId(requestGroup._id); const newChildRequests = await models.request.findByParentId(newRequestGroup._id); const newChildRequestGroups = await models.requestGroup.findByParentId(newRequestGroup._id); // This asserting is pretty garbage but it at least checks // to see that the recursion worked (for the most part) expect(allRequests.length).toBe(8); expect(allRequestGroups.length).toBe(5); expect(childRequests.length).toBe(2); expect(childRequestGroups.length).toBe(1); expect(newChildRequests.length).toBe(2); expect(newChildRequestGroups.length).toBe(1); }); }); describe('_fixThings()', () => { beforeEach(globalBeforeEach); it('fixes duplicate environments', async () => { // Create Workspace with no children const workspace = await models.workspace.create({_id: 'w1'}); expect((await db.withDescendants(workspace)).length).toBe(1); // Create one set of sub environments await models.environment.create({_id: 'b1', parentId: 'w1', data: {foo: 'b1', b1: true}}); await models.environment.create({_id: 'b1_sub1', parentId: 'b1', data: {foo: '1'}}); await models.environment.create({_id: 'b1_sub2', parentId: 'b1', data: {foo: '2'}}); // Create second set of sub environments await models.environment.create({_id: 'b2', parentId: 'w1', data: {foo: 'b2', b2: true}}); await models.environment.create({_id: 'b2_sub1', parentId: 'b2', data: {foo: '3'}}); await models.environment.create({_id: 'b2_sub2', parentId: 'b2', data: {foo: '4'}}); // Make sure we have everything expect((await db.withDescendants(workspace)).length).toBe(7); const descendants = (await db.withDescendants(workspace)).map(d => ({ _id: d._id, parentId: d.parentId, data: d.data || null })); expect(descendants).toEqual([ {_id: 'w1', data: null, parentId: null}, {_id: 'b1', data: {foo: 'b1', b1: true}, parentId: 'w1'}, {_id: 'b2', data: {foo: 'b2', b2: true}, parentId: 'w1'}, {_id: 'b1_sub1', data: {foo: '1'}, parentId: 'b1'}, {_id: 'b1_sub2', data: {foo: '2'}, parentId: 'b1'}, {_id: 'b2_sub1', data: {foo: '3'}, parentId: 'b2'}, {_id: 'b2_sub2', data: {foo: '4'}, parentId: 'b2'} ]); // Run the fix algorithm await db._repairDatabase(); // Make sure things get adjusted const descendants2 = (await db.withDescendants(workspace)).map( d => ({_id: d._id, parentId: d.parentId, data: d.data || null}) ); expect(descendants2).toEqual([ {_id: 'w1', data: null, parentId: null}, {_id: 'b1', data: {foo: 'b1', b1: true, b2: true}, parentId: 'w1'}, // Extra base environments should have been deleted // {_id: 'b2', data: {foo: 'bar'}, parentId: 'w1'}, // Sub environments should have been moved to new "master" base environment {_id: 'b1_sub1', data: {foo: '1'}, parentId: 'b1'}, {_id: 'b1_sub2', data: {foo: '2'}, parentId: 'b1'}, {_id: 'b2_sub1', data: {foo: '3'}, parentId: 'b1'}, {_id: 'b2_sub2', data: {foo: '4'}, parentId: 'b1'} ]); }); it('fixes duplicate cookie jars', async () => { // Create Workspace with no children const workspace = await models.workspace.create({_id: 'w1'}); expect((await db.withDescendants(workspace)).length).toBe(1); // Create one set of sub environments await models.cookieJar.create({ _id: 'j1', parentId: 'w1', cookies: [ {id: '1', key: 'foo', value: '1'}, {id: 'j1_1', key: 'j1', value: '1'} ] }); await models.cookieJar.create({ _id: 'j2', parentId: 'w1', cookies: [ {id: '1', key: 'foo', value: '2'}, {id: 'j2_1', key: 'j2', value: '2'} ] }); // Make sure we have everything expect((await db.withDescendants(workspace)).length).toBe(3); const descendants = (await db.withDescendants(workspace)).map( d => ({_id: d._id, cookies: d.cookies || null, parentId: d.parentId}) ); expect(descendants).toEqual([ {_id: 'w1', cookies: null, parentId: null}, { _id: 'j1', parentId: 'w1', cookies: [ {id: '1', key: 'foo', value: '1'}, {id: 'j1_1', key: 'j1', value: '1'} ] }, { _id: 'j2', parentId: 'w1', cookies: [ {id: '1', key: 'foo', value: '2'}, {id: 'j2_1', key: 'j2', value: '2'} ] } ]); // Run the fix algorithm await db._repairDatabase(); // Make sure things get adjusted const descendants2 = (await db.withDescendants(workspace)).map( d => ({_id: d._id, cookies: d.cookies || null, parentId: d.parentId}) ); expect(descendants2).toEqual([ {_id: 'w1', cookies: null, parentId: null}, { _id: 'j1', parentId: 'w1', cookies: [ {id: '1', key: 'foo', value: '1'}, {id: 'j1_1', key: 'j1', value: '1'}, {id: 'j2_1', key: 'j2', value: '2'} ] } ]); }); });