support for Sequelize.REAL (mickhansen#638)

vugga · Nov 7, 2018 · d392090 · d392090
1 parent b036ef3
commit d392090
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/src/typeMapper.js b/src/typeMapper.js
@@ -41,6 +41,7 @@ export function toGraphQL(sequelizeType, sequelizeTypes) {
     BOOLEAN,
     ENUM,
     FLOAT,
+    REAL,
     CHAR,
     DECIMAL,
     DOUBLE,
@@ -54,7 +55,8 @@ export function toGraphQL(sequelizeType, sequelizeTypes) {
     TIME,
     ARRAY,
     VIRTUAL,
-    JSON
+    JSON,
+    JSONB
   } = sequelizeTypes;
 
   // Map of special characters
@@ -67,6 +69,7 @@ export function toGraphQL(sequelizeType, sequelizeTypes) {
   if (sequelizeType instanceof BOOLEAN) return GraphQLBoolean;
 
   if (sequelizeType instanceof FLOAT ||
+      sequelizeType instanceof REAL ||
       sequelizeType instanceof DOUBLE) return GraphQLFloat;
 
   if (sequelizeType instanceof DATE) {
@@ -110,7 +113,8 @@ export function toGraphQL(sequelizeType, sequelizeTypes) {
     return returnType;
   }
 
-  if (sequelizeType instanceof JSON) {
+  if (sequelizeType instanceof JSONB ||
+      sequelizeType instanceof JSON) {
     return JSONType;
   }
 

diff --git a/test/unit/typeMapper.test.js b/test/unit/typeMapper.test.js
@@ -9,6 +9,7 @@ const {
   BOOLEAN,
   ENUM,
   FLOAT,
+  REAL,
   CHAR,
   DECIMAL,
   DOUBLE,
@@ -138,6 +139,11 @@ describe('typeMapper', () => {
     });
   });
 
+  describe('REAL', function () {
+    it('should map to GraphQLFloat', function () {
+      expect(toGraphQL(new REAL(), Sequelize)).to.equal(GraphQLFloat);
+    });
+  });
 
   describe('INTEGER', function () {
     it('should map to GraphQLInt', function () {