68 lines
1.4 KiB
TypeScript
68 lines
1.4 KiB
TypeScript
|
// Copyright 2023 Signal Messenger, LLC
|
||
|
|
// SPDX-License-Identifier: AGPL-3.0-only
|
||
|
|
|
||
|
|
import { assert } from 'chai';
|
||
|
|
import * as sinon from 'sinon';
|
||
|
|
|
||
|
|
import { DNSCache } from '../../util/dns';
|
||
|
|
import { SECOND } from '../../util/durations';
|
||
|
|
|
||
|
|
const NOW = 1680726906000;
|
||
|
|
|
||
|
|
describe('dns/DNSCache', () => {
|
||
|
|
let sandbox: sinon.SinonSandbox;
|
||
|
|
let cache: DNSCache;
|
||
|
|
beforeEach(() => {
|
||
|
|
sandbox = sinon.createSandbox();
|
||
|
|
cache = new DNSCache();
|
||
|
|
});
|
||
|
|
|
||
|
|
afterEach(() => {
|
||
|
|
sandbox.restore();
|
||
|
|
});
|
||
|
|
|
||
|
|
it('should cache records and pick a random one', () => {
|
||
|
|
sandbox.useFakeTimers({
|
||
|
|
now: NOW,
|
||
|
|
});
|
||
|
|
|
||
|
|
const result = cache.setAndPick('signal.org', 4, [
|
||
|
|
{
|
||
|
|
data: '10.0.0.1',
|
||
|
|
expiresAt: NOW + SECOND,
|
||
|
|
},
|
||
|
|
{
|
||
|
|
data: '10.0.0.2',
|
||
|
|
expiresAt: NOW + SECOND,
|
||
|
|
},
|
||
|
|
]);
|
||
|
|
|
||
|
|
assert.oneOf(result, ['10.0.0.1', '10.0.0.2']);
|
||
|
|
});
|
||
|
|
|
||
|
|
it('should invalidate cache after expiration', () => {
|
||
|
|
const clock = sandbox.useFakeTimers({
|
||
|
|
now: NOW,
|
||
|
|
});
|
||
|
|
|
||
|
|
cache.setAndPick('signal.org', 4, [
|
||
|
|
{
|
||
|
|
data: '10.0.0.1',
|
||
|
|
expiresAt: NOW + SECOND,
|
||
|
|
},
|
||
|
|
{
|
||
|
|
data: '10.0.0.2',
|
||
|
|
expiresAt: NOW + 2 * SECOND,
|
||
|
|
},
|
||
|
|
]);
|
||
|
|
|
||
|
|
assert.oneOf(cache.get('signal.org', 4), ['10.0.0.1', '10.0.0.2']);
|
||
|
|
|
||
|
|
clock.tick(SECOND);
|
||
|
|
assert.strictEqual(cache.get('signal.org', 4), '10.0.0.2');
|
||
|
|
|
||
|
|
clock.tick(SECOND);
|
||
|
|
assert.strictEqual(cache.get('signal.org', 4), undefined);
|
||
|
|
});
|
||
|
|
});
|