diff --git a/src/utils/getModelNames.spec.ts b/src/utils/getModelNames.spec.ts index 5acc47fb..24d16b4c 100644 --- a/src/utils/getModelNames.spec.ts +++ b/src/utils/getModelNames.spec.ts @@ -59,7 +59,7 @@ describe('getModelNames', () => { properties: [], }); - expect(getModelNames(new Map())).toEqual([]); - expect(getModelNames(models)).toEqual(['Doe', 'Jane', 'John']); + expect(getModelNames(new Map())).toEqual(['Dictionary']); + expect(getModelNames(models)).toEqual(['Dictionary', 'Doe', 'Jane', 'John']); }); }); diff --git a/src/utils/readHandlebarsTemplates.spec.ts b/src/utils/readHandlebarsTemplates.spec.ts index 9477ffb3..d9ae13ee 100644 --- a/src/utils/readHandlebarsTemplates.spec.ts +++ b/src/utils/readHandlebarsTemplates.spec.ts @@ -1,16 +1,20 @@ import { readHandlebarsTemplates } from './readHandlebarsTemplates'; import * as fs from 'fs'; import { Language } from '../index'; +import * as glob from 'glob'; jest.mock('fs'); +jest.mock('glob'); const fsExistsSync = fs.existsSync as jest.MockedFunction; const fsReadFileSync = fs.readFileSync as jest.MockedFunction; +const globSync = glob.sync as jest.MockedFunction; describe('readHandlebarsTemplates', () => { it('should read the templates', () => { fsExistsSync.mockReturnValue(true); fsReadFileSync.mockReturnValue('{{{message}}}'); + globSync.mockReturnValue([]); const template = readHandlebarsTemplates(Language.TYPESCRIPT); @@ -18,8 +22,10 @@ describe('readHandlebarsTemplates', () => { expect(template.index).toBeDefined(); expect(template.model).toBeDefined(); expect(template.service).toBeDefined(); + expect(template.settings).toBeDefined(); expect(template.index({ message: 'Hello World!' })).toEqual('Hello World!'); expect(template.model({ message: 'Hello World!' })).toEqual('Hello World!'); expect(template.service({ message: 'Hello World!' })).toEqual('Hello World!'); + expect(template.settings({ message: 'Hello World!' })).toEqual('Hello World!'); }); });