You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
102 lines
3.7 KiB
TypeScript
102 lines
3.7 KiB
TypeScript
// tslint:disable: no-implicit-dependencies max-func-body-length no-unused-expression
|
|
|
|
import chai from 'chai';
|
|
import * as sinon from 'sinon';
|
|
import _ from 'lodash';
|
|
import { describe } from 'mocha';
|
|
|
|
import { TestUtils } from '../../../test-utils';
|
|
import { Onions, SnodePool } from '../../../../session/apis/snode_api';
|
|
import * as Data from '../../../../data/data';
|
|
|
|
import chaiAsPromised from 'chai-as-promised';
|
|
import * as OnionPaths from '../../../../session/onions/onionPath';
|
|
import { generateFakeSnodes, generateFakeSnodeWithEdKey } from '../../../test-utils/utils';
|
|
import { SeedNodeAPI } from '../../../../session/apis/seed_node_api';
|
|
chai.use(chaiAsPromised as any);
|
|
chai.should();
|
|
|
|
const { expect } = chai;
|
|
|
|
const guard1ed = 'e3ec6fcc79e64c2af6a48a9865d4bf4b739ec7708d75f35acc3d478f9161534e';
|
|
const guard2ed = 'e3ec6fcc79e64c2af6a48a9865d4bf4b739ec7708d75f35acc3d478f91615349';
|
|
const guard3ed = 'e3ec6fcc79e64c2af6a48a9865d4bf4b739ec7708d75f35acc3d478f9161534a';
|
|
|
|
const fakeSnodePool: Array<Data.Snode> = [
|
|
...generateFakeSnodes(12),
|
|
generateFakeSnodeWithEdKey(guard1ed),
|
|
generateFakeSnodeWithEdKey(guard2ed),
|
|
generateFakeSnodeWithEdKey(guard3ed),
|
|
...generateFakeSnodes(3),
|
|
];
|
|
// tslint:disable: variable-name
|
|
|
|
// tslint:disable-next-line: max-func-body-length
|
|
describe('OnionPaths', () => {
|
|
// Initialize new stubbed cache
|
|
const sandbox = sinon.createSandbox();
|
|
|
|
describe('getSnodePoolFromDBOrFetchFromSeed', () => {
|
|
let getSnodePoolFromDb: sinon.SinonStub;
|
|
let fetchFromSeedWithRetriesAndWriteToDb: sinon.SinonStub;
|
|
let fetchSnodePoolFromSeedNodeWithRetries: sinon.SinonStub;
|
|
|
|
beforeEach(() => {
|
|
// Utils Stubs
|
|
OnionPaths.clearTestOnionPath();
|
|
|
|
TestUtils.stubWindow('getSeedNodeList', () => ['seednode1']);
|
|
TestUtils.stubWindowLog();
|
|
|
|
Onions.resetSnodeFailureCount();
|
|
OnionPaths.resetPathFailureCount();
|
|
SnodePool.TEST_resetState();
|
|
});
|
|
|
|
afterEach(() => {
|
|
TestUtils.restoreStubs();
|
|
sandbox.restore();
|
|
});
|
|
it('if the cached snode pool has at least 12 snodes, just return it without fetching from seed', async () => {
|
|
getSnodePoolFromDb = sandbox.stub(Data, 'getSnodePoolFromDb').resolves(fakeSnodePool);
|
|
fetchFromSeedWithRetriesAndWriteToDb = sandbox.stub(
|
|
SnodePool,
|
|
'TEST_fetchFromSeedWithRetriesAndWriteToDb'
|
|
);
|
|
|
|
const fetched = await SnodePool.getSnodePoolFromDBOrFetchFromSeed();
|
|
expect(getSnodePoolFromDb.callCount).to.be.eq(1);
|
|
expect(fetchFromSeedWithRetriesAndWriteToDb.callCount).to.be.eq(0);
|
|
|
|
expect(fetched).to.deep.equal(fakeSnodePool);
|
|
});
|
|
|
|
it('if the cached snode pool 12 or less snodes, trigger a fetch from the seed nodes', async () => {
|
|
const length12 = fakeSnodePool.slice(0, 12);
|
|
expect(length12.length).to.eq(12);
|
|
getSnodePoolFromDb = sandbox.stub(Data, 'getSnodePoolFromDb').resolves(length12);
|
|
|
|
sandbox.stub(Data, 'updateSnodePoolOnDb').resolves();
|
|
fetchFromSeedWithRetriesAndWriteToDb = sandbox
|
|
.stub(SnodePool, 'TEST_fetchFromSeedWithRetriesAndWriteToDb')
|
|
.callThrough();
|
|
fetchSnodePoolFromSeedNodeWithRetries = sandbox
|
|
.stub(SeedNodeAPI, 'fetchSnodePoolFromSeedNodeWithRetries')
|
|
.resolves(fakeSnodePool);
|
|
|
|
// run the command
|
|
const fetched = await SnodePool.getSnodePoolFromDBOrFetchFromSeed();
|
|
expect(getSnodePoolFromDb.callCount).to.be.eq(1);
|
|
expect(
|
|
fetchFromSeedWithRetriesAndWriteToDb.callCount,
|
|
'fetchFromSeedWithRetriesAndWriteToDb eq 1'
|
|
).to.be.eq(1);
|
|
expect(
|
|
fetchSnodePoolFromSeedNodeWithRetries.callCount,
|
|
'fetchSnodePoolFromSeedNodeWithRetries eq 1'
|
|
).to.be.eq(1);
|
|
expect(fetched).to.deep.equal(fakeSnodePool);
|
|
});
|
|
});
|
|
});
|