test: setupTokenHighlights

Author: andreruffert

src/utils.js

@@ -1,9 +1,12 @@
 /**
  * Create & register the token `Highlight`'s in the `CSS.highlights` registry.
  * This enables the use of `::highlight(tokenType)` in CSS to style them.
- * @param {Object} [languageTokens={}] - Language specific overwrites.
+ *
+ * @param {Array} [tokenTypes=[]] - Language token types
+ * @param {Object} [options={}]
+ * @param {Object} [options.languageTokens={}] - Language specific overwrites.
  */
-export function setupTokenHighlights(tokenTypes, { languageTokens = {} }) {
+export function setupTokenHighlights(tokenTypes = [], { languageTokens = {} } = {}) {
   const languageTokenTypes = Object.entries(languageTokens).flatMap((entry) => {
     const [lang, tokenTypes] = entry;
     return tokenTypes.map((tokenType) => `${lang}-${tokenType}`);

src/utils.test.js

@@ -0,0 +1,20 @@
+import { expect, test } from 'vitest';
+import { setupTokenHighlights } from './utils';
+
+const tokenTypes = ['token-type-1', 'token-type-2'];
+
+test('setupTokenHighlights', async () => {
+  setupTokenHighlights(tokenTypes);
+  for (const tokenType of tokenTypes) {
+    expect(CSS.highlights.get(tokenType)).toBeInstanceOf(Highlight);
+  }
+});
+
+test('setupTokenHighlights - Language specific overwrites', async () => {
+  const languageTokens = { css: ['important'], md: ['title', 'list'] };
+  const languageTokenTypes = ['css-important', 'md-title', 'md-list'];
+  setupTokenHighlights(tokenTypes, { languageTokens });
+  for (const tokenType of [...tokenTypes, ...languageTokenTypes]) {
+    expect(CSS.highlights.get(tokenType)).toBeInstanceOf(Highlight);
+  }
+});