test: add test for TranslatorCollection

SettingDust · SettingDust · commit c7fd8ff8c596 · 2022-12-06T19:58:24.000+08:00
diff --git a/test/e2e.test.ts b/test/e2e.test.ts
@@ -2,13 +2,15 @@ import { NodeHtmlMarkdown } from '../src';
 import { DOMParser } from 'linkedom';
 
 
-const html = `<html><body><p><b>F</b><span>oo</span></p><ul><li>Bar</li></ul></body></html>`
+const html = `<html><body><p><b>F</b><span>oo</span></p><ul><li>Bar</li></ul></body></html>`;
 
-test('Browser', () => {
-  __IS_BROWSER__ = true
-  globalThis.DOMParser = <typeof globalThis.DOMParser>DOMParser
-  const res = new NodeHtmlMarkdown({
-    preferNativeParser: true
-  }).translate(html);
-  expect(res).toBe(`**F**oo\n\n* Bar`);
-})
+describe('E2E', () => {
+  test('Browser', () => {
+    __IS_BROWSER__ = true;
+    globalThis.DOMParser = <typeof globalThis.DOMParser>DOMParser;
+    const res = new NodeHtmlMarkdown({
+      preferNativeParser: true
+    }).translate(html);
+    expect(res).toBe(`**F**oo\n\n* Bar`);
+  });
+});
diff --git a/test/translator-collection.test.ts b/test/translator-collection.test.ts
@@ -0,0 +1,42 @@
+import { TranslatorCollection, TranslatorConfigFactory } from '../src';
+
+
+describe('TranslatorCollection', () => {
+  let collection: TranslatorCollection;
+  beforeEach(() => {
+    collection = new TranslatorCollection();
+    collection.set('p', { ignore: true });
+  });
+  test('size', () =>
+    expect(collection.size).toBe(1)
+  );
+  test('get', () => {
+    expect(collection.get('p')).toEqual({ ignore: true });
+    expect(collection.get('P')).toEqual({ ignore: true });
+    expect(collection['P']).toEqual({ ignore: true });
+  });
+  test('set', () => {
+    collection.set('p', { ignore: false });
+    expect(collection.get('p')).toEqual({ ignore: false });
+  });
+  test('set with preserveBase', () => {
+    collection.set('p', { recurse: true }, true);
+    expect(collection.get('p')).toEqual({ ignore: true, recurse: true });
+  });
+  test('set with preserveBase and configFactory', () => {
+    collection.set('p', () => ({ recurse: true }), true);
+    expect(collection.get('p')).toBeInstanceOf(Function)
+    expect((<Function>collection.get('p'))()).toEqual({ recurse: true });
+    expect((<TranslatorConfigFactory>collection.get('p')).base).toEqual({ ignore: true });
+  });
+  test('set with multiple keys', () => {
+    collection.set('p,div', { ignore: false });
+    expect(collection.get('p')).toEqual({ ignore: false });
+    expect(collection.get('div')).toEqual({ ignore: false });
+  });
+  test('entries', () => expect(collection.entries()).toEqual([ [ 'P', { ignore: true } ] ]));
+  test('remove', () => {
+    collection.remove('p');
+    expect(collection.get('p')).toBeUndefined();
+  });
+});
