Move fileType check

Author: mrmckeb

src/helpers/__tests__/cssSnapshots.test.ts

@@ -1,7 +1,12 @@
 import { readFileSync } from 'fs';
 import { IICSSExports } from 'icss-utils';
 import { join } from 'path';
-import { createExports, getClasses, FileTypes } from '../cssSnapshots';
+import {
+  createExports,
+  getClasses,
+  FileTypes,
+  getFileType,
+} from '../cssSnapshots';
 
 const testFileNames = [
   'test.module.css',
@@ -14,8 +19,7 @@ const testFileNames = [
 describe('utils / cssSnapshots', () => {
   testFileNames.forEach((fileName) => {
     let classes: IICSSExports;
-    const isLess = fileName.endsWith('less');
-    const fileType = isLess ? FileTypes.less : FileTypes.scss;
+    const fileType = getFileType(fileName);
     const testFile = readFileSync(
       join(__dirname, 'fixtures', fileName),
       'utf8',

src/helpers/cssSnapshots.ts

@@ -25,6 +25,9 @@ export const enum FileTypes {
   less = 'less',
 }
 
+export const getFileType = (fileName: string) =>
+  fileName.endsWith('less') ? FileTypes.less : FileTypes.scss;
+
 export const getClasses = (css: string, fileType: FileTypes) => {
   try {
     let transformedCss = '';
@@ -78,7 +81,7 @@ export const getDtsSnapshot = (
   options: Options,
 ) => {
   const css = scriptSnapshot.getText(0, scriptSnapshot.getLength());
-  const fileType = fileName.endsWith('less') ? FileTypes.less : FileTypes.scss;
+  const fileType = getFileType(fileName);
   const classes = getClasses(css, fileType);
   const dts = createExports(classes, options);
   return ts.ScriptSnapshot.fromString(dts);