fix(map): fix semantic validation

Signed-off-by: Jonathan Casey <jonathan.casey@docusign.com>
accordproject · Apr 18, 2024 · 6eee6b7 · 6eee6b7
1 parent 3dddd1e
commit 6eee6b7
Show file tree

Hide file tree

Showing 4 changed files with 37 additions and 50 deletions.
diff --git a/packages/concerto-core/lib/introspect/mapkeytype.js b/packages/concerto-core/lib/introspect/mapkeytype.js
@@ -69,22 +69,14 @@ class MapKeyType extends Decorated {
      * @protected
      */
     validate() {
-
         if (!ModelUtil.isPrimitiveType(this.type)) {
-
             const decl = this.modelFile.getType(this.ast.type.name);
-
+            // All but StringScalar & DateTimeScalar are unsupported.
             if  (!ModelUtil.isValidMapKeyScalar(decl)) {
                 throw new IllegalModelException(
                     `Scalar must be one of StringScalar, DateTimeScalar in context of MapKeyType. Invalid Scalar: ${this.type}, for MapDeclaration ${this.parent.name}`
                 );
             }
-
-            if (decl?.isConcept?.() || decl?.isClassDeclaration?.()) {
-                throw new IllegalModelException(
-                    `Invalid Map key type in MapDeclaration ${this.parent.name}. Only String and DateTime types are supported for Map key types`
-                );
-            }
         }
     }
 
@@ -95,8 +87,7 @@ class MapKeyType extends Decorated {
      * @private
      */
     processType(ast) {
-        let decl;
-        switch(this.ast.$class) {
+        switch(ast.$class) {
         case `${MetaModelNamespace}.DateTimeMapKeyType`:
             this.type = 'DateTime';
             break;

diff --git a/packages/concerto-core/lib/modelutil.js b/packages/concerto-core/lib/modelutil.js
@@ -316,9 +316,8 @@ class ModelUtil {
      * @return {boolean} true if the Key is a valid Map Key Scalar type
     */
     static isValidMapKeyScalar(decl) {
-        return (decl?.isScalarDeclaration?.() &&
-        (decl?.ast.$class !== `${MetaModelNamespace}.StringScalar`  ||
-        decl?.ast.$class !== `${MetaModelNamespace}.DateTimeScalar`));
+        return (decl?.isScalarDeclaration?.() && decl?.ast.$class === `${MetaModelNamespace}.StringScalar`)  ||
+        (decl?.isScalarDeclaration?.() && decl?.ast.$class === `${MetaModelNamespace}.DateTimeScalar`);
     }
 
     /**

diff --git a/...concerto-core/test/data/parser/mapdeclaration/mapdeclaration.goodkey.declaration.enum.cto b/...concerto-core/test/data/parser/mapdeclaration/mapdeclaration.goodkey.declaration.enum.cto
diff --git a/packages/concerto-core/test/introspect/mapdeclaration.js b/packages/concerto-core/test/introspect/mapdeclaration.js
@@ -298,36 +298,55 @@ describe('MapDeclaration', () => {
         it('should throw if ast contains illegal Map Key Type - Concept', () => {
             (() =>  {
                 let decl = introspectUtils.loadLastDeclaration('test/data/parser/mapdeclaration/mapdeclaration.badkey.declaration.concept.cto', MapDeclaration);
-                decl.validate().should.throw(IllegalModelException);
-            });
+                (() => {
+                    decl.validate();
+                }).should.throw(IllegalModelException);
+            })();
+        });
+
+        it('should throw if ast contains illegal Map Key Type - Enum', () => {
+            (() =>  {
+                let decl = introspectUtils.loadLastDeclaration('test/data/parser/mapdeclaration/mapdeclaration.badkey.declaration.enum.cto', MapDeclaration);
+                (() => {
+                    decl.validate();
+                }).should.throw(IllegalModelException);
+            })();
         });
 
         it('should throw if ast contains illegal Map Key Type - Scalar Long', () => {
             (() =>  {
                 let decl = introspectUtils.loadLastDeclaration('test/data/parser/mapdeclaration/mapdeclaration.badkey.scalar.long.cto', MapDeclaration);
-                decl.validate();
-            });
+                (() => {
+                    decl.validate();
+                }).should.throw(IllegalModelException);
+            })();
         });
 
         it('should throw if ast contains illegal Map Key Type - Scalar Integer', () => {
             (() =>  {
                 let decl = introspectUtils.loadLastDeclaration('test/data/parser/mapdeclaration/mapdeclaration.badkey.scalar.integer.cto', MapDeclaration);
-                decl.validate().should.throw(IllegalModelException);
-            });
+                (() => {
+                    decl.validate();
+                }).should.throw(IllegalModelException);
+            })();
         });
 
         it('should throw if ast contains illegal Map Key Type - Scalar Double', () => {
             (() =>  {
                 let decl = introspectUtils.loadLastDeclaration('test/data/parser/mapdeclaration/mapdeclaration.badkey.scalar.double.cto', MapDeclaration);
-                decl.validate().should.throw(IllegalModelException);
-            });
+                (() => {
+                    decl.validate();
+                }).should.throw(IllegalModelException);
+            })();
         });
 
         it('should throw if ast contains illegal Map Key Type - Scalar Boolean', () => {
             (() =>  {
                 let decl = introspectUtils.loadLastDeclaration('test/data/parser/mapdeclaration/mapdeclaration.badkey.scalar.boolean.cto', MapDeclaration);
-                decl.validate().should.throw(IllegalModelException);
-            });
+                (() => {
+                    decl.validate();
+                }).should.throw(IllegalModelException);
+            })();
         });
 
         it('should throw if ast contains illegal Map Key Type - Boolean', () => {
@@ -414,8 +433,10 @@ describe('MapDeclaration', () => {
                 const base_cto = fs.readFileSync('test/data/parser/mapdeclaration/base.cto', 'utf-8');
                 introspectUtils.modelManager.addCTOModel(base_cto, 'base.cto');
                 let decl = introspectUtils.loadLastDeclaration('test/data/parser/mapdeclaration/mapdeclaration.badkey.imported.thing.cto', MapDeclaration);
-                decl.validate().should.throw(IllegalModelException);
-            });
+                (() => {
+                    decl.validate();
+                }).should.throw(IllegalModelException);
+            })();
         });
     });