insomnia/packages/insomnia-app/app/common/__tests__/database.test.js

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(18);

    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'}
        ]
      }
    ]);
  });
});