1 files changed, 11 insertions, 4 deletions
diff --git a/spec/frontend/content_editor/services/code_block_language_loader_spec.js b/spec/frontend/content_editor/services/code_block_language_loader_spec.js
index 943de327762..795f5219a3f 100644
--- a/spec/frontend/content_editor/services/code_block_language_loader_spec.js
+++ b/spec/frontend/content_editor/services/code_block_language_loader_spec.js
@@ -18,25 +18,32 @@ describe('content_editor/services/code_block_language_loader', () => {
     languageLoader.lowlight = lowlight;
   });
 
-  describe('findLanguageBySyntax', () => {
+  describe('findOrCreateLanguageBySyntax', () => {
     it.each`
       syntax          | language
       ${'javascript'} | ${{ syntax: 'javascript', label: 'Javascript' }}
       ${'js'}         | ${{ syntax: 'javascript', label: 'Javascript' }}
       ${'jsx'}        | ${{ syntax: 'javascript', label: 'Javascript' }}
     `('returns a language by syntax and its variants', ({ syntax, language }) => {
-      expect(languageLoader.findLanguageBySyntax(syntax)).toMatchObject(language);
+      expect(languageLoader.findOrCreateLanguageBySyntax(syntax)).toMatchObject(language);
     });
 
     it('returns Custom (syntax) if the language does not exist', () => {
-      expect(languageLoader.findLanguageBySyntax('foobar')).toMatchObject({
+      expect(languageLoader.findOrCreateLanguageBySyntax('foobar')).toMatchObject({
         syntax: 'foobar',
         label: 'Custom (foobar)',
       });
     });
 
+    it('returns Diagram (syntax) if the language does not exist, and isDiagram = true', () => {
+      expect(languageLoader.findOrCreateLanguageBySyntax('foobar', true)).toMatchObject({
+        syntax: 'foobar',
+        label: 'Diagram (foobar)',
+      });
+    });
+
     it('returns plaintext if no syntax is passed', () => {
-      expect(languageLoader.findLanguageBySyntax('')).toMatchObject({
+      expect(languageLoader.findOrCreateLanguageBySyntax('')).toMatchObject({
         syntax: 'plaintext',
         label: 'Plain text',
       });