add isStringType, isNumberType

dsherret · Mar 16, 2018 · 66eecde · 66eecde
1 parent e2b4a99
commit 66eecde
Show file tree

Hide file tree

Showing 2 changed files with 38 additions and 0 deletions.
diff --git a/src/compiler/type/Type.ts b/src/compiler/type/Type.ts
@@ -248,6 +248,20 @@ export class Type<TType extends ts.Type = ts.Type> {
         return (this.compilerType.flags & TypeFlags.Boolean) !== 0;
     }
 
+    /**
+     * Gets if this is a string type.
+     */
+    isStringType() {
+        return (this.compilerType.flags & TypeFlags.String) !== 0;
+    }
+
+    /**
+     * Gets if this is a number type.
+     */
+    isNumberType() {
+        return (this.compilerType.flags & TypeFlags.Number) !== 0;
+    }
+
     /**
      * Gets if this is an enum type.
      */

diff --git a/src/tests/compiler/type/typeTests.ts b/src/tests/compiler/type/typeTests.ts
@@ -156,6 +156,30 @@ describe(nameof(Type), () => {
         });
     });
 
+    describe(nameof<Type>(t => t.isStringType), () => {
+        it("should get when it is a string type", () => {
+            const {firstType} = getTypeFromText("let myType: string;");
+            expect(firstType.isStringType()).to.equal(true);
+        });
+
+        it("should get when it's not a string type", () => {
+            const {firstType} = getTypeFromText("let myType: boolean;");
+            expect(firstType.isStringType()).to.equal(false);
+        });
+    });
+
+    describe(nameof<Type>(t => t.isNumberType), () => {
+        it("should get when it is a number type", () => {
+            const {firstType} = getTypeFromText("let myType: number;");
+            expect(firstType.isNumberType()).to.equal(true);
+        });
+
+        it("should get when it's not a number type", () => {
+            const {firstType} = getTypeFromText("let myType: boolean;");
+            expect(firstType.isNumberType()).to.equal(false);
+        });
+    });
+
     describe(nameof<Type>(t => t.isEnumType), () => {
         it("should get when it is an enum type", () => {
             const {firstType} = getTypeFromText("let myType: MyEnum; enum MyEnum {}");