Test: fix trie test case

This commit is contained in:
SukkaW 2024-12-14 20:14:46 +08:00
parent 226f736e2a
commit e53e6b9375
3 changed files with 45 additions and 38 deletions

View File

@ -117,12 +117,12 @@ describe('Trie', () => {
trie.add('cdn.example.com'); trie.add('cdn.example.com');
trie.add('example.org'); trie.add('example.org');
expect(trie.find('example.com')).toStrictEqual(['example.com', 'cdn.example.com', 'blog.example.com']); expect(trie.find('example.com')).toStrictEqual(['example.com', 'blog.example.com', 'cdn.example.com']);
expect(trie.find('com')).toStrictEqual(['example.com', 'cdn.example.com', 'blog.example.com']); expect(trie.find('com')).toStrictEqual(['example.com', 'blog.example.com', 'cdn.example.com']);
expect(trie.find('.example.com')).toStrictEqual(['cdn.example.com', 'blog.example.com']); expect(trie.find('.example.com')).toStrictEqual(['blog.example.com', 'cdn.example.com']);
expect(trie.find('org')).toStrictEqual(['example.org']); expect(trie.find('org')).toStrictEqual(['example.org']);
expect(trie.find('example.net')).toStrictEqual([]); expect(trie.find('example.net')).toStrictEqual([]);
expect(trie.find('')).toStrictEqual(['example.org', 'example.com', 'cdn.example.com', 'blog.example.com']); expect(trie.find('')).toStrictEqual(['example.com', 'example.org', 'blog.example.com', 'cdn.example.com']);
}); });
it('should be possible to retrieve items matching the given prefix even with a smol trie', () => { it('should be possible to retrieve items matching the given prefix even with a smol trie', () => {
@ -139,7 +139,7 @@ describe('Trie', () => {
expect(trie.find('.example.com')).toStrictEqual(['.example.com']); expect(trie.find('.example.com')).toStrictEqual(['.example.com']);
expect(trie.find('org')).toStrictEqual(['example.org']); expect(trie.find('org')).toStrictEqual(['example.org']);
expect(trie.find('example.net')).toStrictEqual([]); expect(trie.find('example.net')).toStrictEqual([]);
expect(trie.find('')).toStrictEqual(['example.org', '.example.com']); expect(trie.find('')).toStrictEqual(['.example.com', 'example.org']);
}); });
it('should be possible to create a trie from an arbitrary iterable.', () => { it('should be possible to create a trie from an arbitrary iterable.', () => {
@ -165,14 +165,14 @@ describe('surge domainset dedupe', () => {
it('should match subdomain - 1', () => { it('should match subdomain - 1', () => {
const trie = createTrie(['www.noc.one', 'www.sukkaw.com', 'blog.skk.moe', 'image.cdn.skk.moe', 'cdn.sukkaw.net'], false); const trie = createTrie(['www.noc.one', 'www.sukkaw.com', 'blog.skk.moe', 'image.cdn.skk.moe', 'cdn.sukkaw.net'], false);
expect(trie.find('.skk.moe')).toStrictEqual(['image.cdn.skk.moe', 'blog.skk.moe']); expect(trie.find('.skk.moe')).toStrictEqual(['blog.skk.moe', 'image.cdn.skk.moe']);
expect(trie.find('.sukkaw.com')).toStrictEqual(['www.sukkaw.com']); expect(trie.find('.sukkaw.com')).toStrictEqual(['www.sukkaw.com']);
}); });
it('should match subdomain - 2', () => { it('should match subdomain - 2', () => {
const trie = createTrie(['www.noc.one', 'www.sukkaw.com', '.skk.moe', 'blog.skk.moe', 'image.cdn.skk.moe', 'cdn.sukkaw.net'], false); const trie = createTrie(['www.noc.one', 'www.sukkaw.com', '.skk.moe', 'blog.skk.moe', 'image.cdn.skk.moe', 'cdn.sukkaw.net'], false);
expect(trie.find('.skk.moe')).toStrictEqual(['.skk.moe', 'image.cdn.skk.moe', 'blog.skk.moe']); expect(trie.find('.skk.moe')).toStrictEqual(['.skk.moe', 'blog.skk.moe', 'image.cdn.skk.moe']);
expect(trie.find('.sukkaw.com')).toStrictEqual(['www.sukkaw.com']); expect(trie.find('.sukkaw.com')).toStrictEqual(['www.sukkaw.com']);
}); });
@ -191,9 +191,10 @@ describe('smol tree', () => {
], true); ], true);
expect(trie.dump()).toStrictEqual([ expect(trie.dump()).toStrictEqual([
'.sub.example.com', '.skk.moe',
'cdn.noc.one', 'www.noc.one', 'www.noc.one',
'.skk.moe' 'cdn.noc.one',
'.sub.example.com'
]); ]);
}); });
@ -231,10 +232,10 @@ describe('smol tree', () => {
], true); ], true);
expect(trie.dump()).toStrictEqual([ expect(trie.dump()).toStrictEqual([
'cdn.creative.medialytics.com',
'px.cdn.creative.medialytics.com',
'commercial.shouji.360.cn', 'commercial.shouji.360.cn',
'act.commercial.shouji.360.cn' 'cdn.creative.medialytics.com',
'act.commercial.shouji.360.cn',
'px.cdn.creative.medialytics.com'
]); ]);
}); });
@ -247,10 +248,10 @@ describe('smol tree', () => {
], true); ], true);
expect(trie.dump()).toStrictEqual([ expect(trie.dump()).toStrictEqual([
'anotherskk.moe',
'blog.anotherskk.moe',
'skk.moe', 'skk.moe',
'blog.skk.moe' 'anotherskk.moe',
'blog.skk.moe',
'blog.anotherskk.moe'
]); ]);
}); });
@ -266,46 +267,46 @@ describe('smol tree', () => {
], true); ], true);
expect(trie.dump()).toStrictEqual([ expect(trie.dump()).toStrictEqual([
'img.skk.local',
'blog.img.skk.local',
'.cdn.local',
'anotherskk.moe',
'blog.anotherskk.moe',
'skk.moe', 'skk.moe',
'blog.skk.moe' 'anotherskk.moe',
'.cdn.local',
'blog.skk.moe',
'blog.anotherskk.moe',
'img.skk.local',
'blog.img.skk.local'
]); ]);
trie.whitelist('.skk.moe'); trie.whitelist('.skk.moe');
expect(trie.dump()).toStrictEqual([ expect(trie.dump()).toStrictEqual([
'img.skk.local',
'blog.img.skk.local',
'.cdn.local',
'anotherskk.moe', 'anotherskk.moe',
'blog.anotherskk.moe' '.cdn.local',
'blog.anotherskk.moe',
'img.skk.local',
'blog.img.skk.local'
]); ]);
trie.whitelist('anotherskk.moe'); trie.whitelist('anotherskk.moe');
expect(trie.dump()).toStrictEqual([ expect(trie.dump()).toStrictEqual([
'img.skk.local',
'blog.img.skk.local',
'.cdn.local', '.cdn.local',
'blog.anotherskk.moe' 'blog.anotherskk.moe',
'img.skk.local',
'blog.img.skk.local'
]); ]);
trie.add('anotherskk.moe'); trie.add('anotherskk.moe');
trie.whitelist('.anotherskk.moe'); trie.whitelist('.anotherskk.moe');
expect(trie.dump()).toStrictEqual([ expect(trie.dump()).toStrictEqual([
'.cdn.local',
'img.skk.local', 'img.skk.local',
'blog.img.skk.local', 'blog.img.skk.local'
'.cdn.local'
]); ]);
trie.whitelist('img.skk.local'); trie.whitelist('img.skk.local');
expect(trie.dump()).toStrictEqual([ expect(trie.dump()).toStrictEqual([
'blog.img.skk.local', '.cdn.local',
'.cdn.local' 'blog.img.skk.local'
]); ]);
trie.whitelist('cdn.local'); trie.whitelist('cdn.local');
@ -328,15 +329,15 @@ describe('smol tree', () => {
], true); ], true);
expect(trie.dump()).toStrictEqual([ expect(trie.dump()).toStrictEqual([
'cdn.example.com', 'blog.cdn.example.com', '.t.co',
'.skk.moe', '.skk.moe',
'.t.co' 'cdn.example.com', 'blog.cdn.example.com'
]); ]);
trie.whitelist('.t.co'); trie.whitelist('.t.co');
expect(trie.dump()).toStrictEqual([ expect(trie.dump()).toStrictEqual([
'cdn.example.com', 'blog.cdn.example.com', '.skk.moe',
'.skk.moe' 'cdn.example.com', 'blog.cdn.example.com'
]); ]);
trie.whitelist('skk.moe'); trie.whitelist('skk.moe');

View File

@ -15,7 +15,7 @@
"build-profile": "pnpm run dexnode -r @swc-node/register ./Build/index.ts", "build-profile": "pnpm run dexnode -r @swc-node/register ./Build/index.ts",
"build-webstream": "ENABLE_EXPERIMENTAL_WEBSTREAMS=true pnpm run node ./Build/index.ts", "build-webstream": "ENABLE_EXPERIMENTAL_WEBSTREAMS=true pnpm run node ./Build/index.ts",
"lint": "eslint --format=sukka .", "lint": "eslint --format=sukka .",
"test": "SWCRC=true mocha --require @swc-node/register --watch-extensions ts,tsx" "test": "SWCRC=true SWC_NODE_PROJECT=tsconfig.test.json mocha --require @swc-node/register --watch-extensions ts,tsx"
}, },
"author": "", "author": "",
"license": "ISC", "license": "ISC",

6
tsconfig.test.json Normal file
View File

@ -0,0 +1,6 @@
{
"extends": "./tsconfig.json",
"compilerOptions": {
"inlineSourceMap": true
}
}