From 6b396e7c4a073af639c55cd0b7dbfd8dd99a39a5 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Mart=C3=ADn=20Pablo=20Caballero?= <mpccolorado@gmail.com>
Date: Fri, 23 Jul 2021 17:04:39 -0300
Subject: [PATCH] Fixed issue #473. Enum works properly now

---
 src/js/Node.js        |   7 +-
 test/Node.test.js     | 187 ++++++++++++++++++++++++++++++++++++++++++
 test/test_enum.html   |  62 ++++++++++++++
 test/test_enum_2.html | 123 +++++++++++++++++++++++++++
 4 files changed, 378 insertions(+), 1 deletion(-)
 create mode 100644 test/test_enum.html
 create mode 100644 test/test_enum_2.html

diff --git a/src/js/Node.js b/src/js/Node.js
index aabf1ee01..73957e59e 100644
--- a/src/js/Node.js
+++ b/src/js/Node.js
@@ -4482,7 +4482,12 @@ Node._findSchema = (topLevelSchema, schemaRefs, path, currentSchema = topLevelSc
         if (schemaUrl in schemaRefs) {
           const referencedSchema = schemaRefs[schemaUrl]
           const reference = { $ref: '#/'.concat(relativePath) }
-          return Node._findSchema(referencedSchema, schemaRefs, nextPath, reference)
+          let auxNextPath = []
+          auxNextPath.push(nextKey)
+          if(nextPath.length > 0) {
+            auxNextPath.push(...nextPath)
+          }
+          return Node._findSchema(referencedSchema, schemaRefs, auxNextPath, reference)
         } else {
           throw Error(`Unable to resolve reference ${ref}`)
         }
diff --git a/test/Node.test.js b/test/Node.test.js
index 70b43298f..26957a233 100644
--- a/test/Node.test.js
+++ b/test/Node.test.js
@@ -77,6 +77,69 @@ describe('Node', () => {
       )
     })
 
+    it('should find referenced schema within multi-level object properties', () => {
+      const schema = {
+        type: 'object',
+        properties: {
+          aProperty: {
+            "$ref": "second_schema#/definitions/some_def"
+          }
+        }
+      }
+      const schemaRefs = {
+        second_schema: {
+          definitions: {
+            some_def: {
+              type: 'object',
+              properties: {
+                enumProp: {
+                  enum: [1, 2, 3]
+                }
+              }
+            }
+          }
+        }
+      }
+      let path = ['aProperty', 'enumProp']
+      let expected_schema = {
+        enum: [1, 2, 3]
+      }
+      assert.deepStrictEqual(Node._findSchema(schema, schemaRefs, path), expected_schema)
+    })
+
+    it('should find array referenced schema within multi-level object properties', () => {
+      const schema = {
+        type: 'object',
+        properties: {
+          aProperty: {
+            type: "array",
+            items: {
+              "$ref": "second_schema#/definitions/some_def"
+            }
+          }
+        }
+      }
+      const schemaRefs = {
+        "second_schema": {
+          "definitions": {
+            "some_def": {
+              type: 'object',
+              properties: {
+                enumProp: {
+                  enum: [1, 2, 3]
+                }
+              }
+            }
+          }
+        }
+      }
+      let path = ['aProperty', 0, 'enumProp']
+      let expected_schema = {
+          enum: [1, 2, 3]
+      }
+      assert.deepStrictEqual(Node._findSchema(schema, schemaRefs, path), expected_schema)
+    })
+
     it('should return null for path that has no schema', () => {
       const schema = {
         type: 'object',
@@ -152,6 +215,121 @@ describe('Node', () => {
         const path = ['foo']
         assert.strictEqual(Node._findSchema(schema, { foo: fooSchema }, path), fooSchema)
       })
+
+      it('should find a referenced schema property', () => {
+        const schema = {
+          type: 'object',
+          properties: {
+            foo: {
+              $ref: 'foo'
+            }
+          }
+        }
+        const fooSchema = {
+          type: 'object',
+          properties: {
+            levelTwo: {
+              type: 'string'
+            }
+          }
+        }
+        const path = ['foo', 'levelTwo']
+        assert.strictEqual(Node._findSchema(schema, { foo: fooSchema }, path), fooSchema.properties.levelTwo)
+      })
+
+      it('should find a referenced schema definition', () => {
+        const schema = {
+          type: 'object',
+          properties: {
+            foo: {
+              type: "array",
+              items: {
+                "$ref": "foo#/definitions/some_def"
+              }
+            }
+          }
+        }
+        const fooSchema = {
+          "definitions": {
+            "some_def": {
+              type: 'object',
+              properties: {
+                propA: {
+                  type: 'string'
+                },
+                propB: {
+                  type: 'string'
+                }
+              }
+            }
+          }
+        }
+        const path = ['foo', 0]
+        assert.strictEqual(Node._findSchema(schema, { foo: fooSchema }, path), fooSchema.definitions.some_def)
+      })
+
+      it('should find a referenced schema definition 2', () => {
+        const schema = {
+          type: 'object',
+          properties: {
+            foo: {
+              type: "array",
+              items: {
+                "$ref": "foo#/definitions/some_def"
+              }
+            }
+          }
+        }
+        const fooSchema = {
+          definitions: {
+            some_def: {
+              type: 'object',
+              properties: {
+                propA: {
+                  type: 'string'
+                },
+                propB: {
+                  type: 'string'
+                }
+              }
+            }
+          }
+        }
+        const path = ['foo', 0, 'propA']
+        assert.strictEqual(Node._findSchema(schema, { foo: fooSchema }, path), fooSchema.definitions.some_def.properties.propA)
+      })
+
+      it('should find a referenced schema definition 3', () => {
+        const schema = {
+          type: 'object',
+          properties: {
+            foo: {
+              type: "array",
+              items: {
+                "$ref": "foo#/definitions/some_def"
+              }
+            }
+          }
+        }
+        const fooSchema = {
+          "definitions": {
+            "some_def": {
+              type: 'object',
+              properties: {
+                propA: {
+                  type: 'object',
+                  properties: {
+                    propA1: { type: 'boolean' }
+                  }
+                },
+                propB: { type: 'string' }
+              }
+            }
+          }
+        }
+        const path = ['foo', 0, 'propA', 'propA1']
+        assert.strictEqual(Node._findSchema(schema, { foo: fooSchema }, path), fooSchema.definitions.some_def.properties.propA.properties.propA1)
+      })
     })
 
     describe('with $ref to internal definition', () => {
@@ -423,4 +601,13 @@ describe('Node', () => {
       })
     })
   })
+  describe('_findEnum', () => {
+    it('should find enum', () => {
+      const schema = {
+        type: 'object',
+        enum: [1, 2, 3]
+      }
+      assert.strictEqual(Node._findEnum(schema), schema.enum)
+    })
+  })
 })
diff --git a/test/test_enum.html b/test/test_enum.html
new file mode 100644
index 000000000..399f88d3b
--- /dev/null
+++ b/test/test_enum.html
@@ -0,0 +1,62 @@
+<!DOCTYPE HTML>
+<html>
+<head>
+    <title>JSONEditor | template + enums</title>
+
+    <link href="../dist/jsoneditor.css" rel="stylesheet" type="text/css">
+    <script src="../dist/jsoneditor.js"></script>
+
+    <style type="text/css">
+        #jsoneditor {
+            width: 500px;
+            height: 700px;
+            background: white;
+        }
+
+        p {
+            width: 500px;
+            font-family: sans;
+        }
+    </style>
+</head>
+<body>
+
+<div id="jsoneditor"></div>
+
+<script>
+    var schema = {
+        "title": "Example Schema",
+        "type": "object",
+        "properties": {
+            "firstName": {
+                "type": "string",
+                "enum": ["a", "b"]
+            },
+            "testObj": {
+                "type": "object",
+                "additionalProperties": {
+                    "type": "string",
+                    "enum": ["a", "b"]
+                },
+            }
+        },
+        "required": ["firstName"]
+    };
+
+
+    var json =
+        {
+            firstName: 'a',
+            testObj: {},
+        };
+
+    var options = {
+        schema: schema
+    };
+
+    // create the editor
+    var container = document.getElementById('jsoneditor');
+    var editor = new JSONEditor(container, options, json);
+</script>
+</body>
+</html>
diff --git a/test/test_enum_2.html b/test/test_enum_2.html
new file mode 100644
index 000000000..6f27f033f
--- /dev/null
+++ b/test/test_enum_2.html
@@ -0,0 +1,123 @@
+<!DOCTYPE HTML>
+<html>
+<head>
+    <title>JSONEditor | template + enums</title>
+
+    <link href="../dist/jsoneditor.css" rel="stylesheet" type="text/css">
+    <script src="../dist/jsoneditor.js"></script>
+
+    <style type="text/css">
+        #jsoneditor {
+            width: 500px;
+            height: 700px;
+            background: white;
+        }
+
+        p {
+            width: 500px;
+            font-family: sans;
+        }
+    </style>
+</head>
+<body>
+<p>Demonstrates a template with JSON schema validation. To use: click the context menu of the first or second employee, click insert or append, click "Employee".</p>
+<p><a href="https://github.com/josdejong/jsoneditor/issues/473">See github issue #473</a></p>
+
+<div id="jsoneditor"></div>
+
+<script>
+    var schema = {
+        "title": "Example Schema",
+        "type": "array",
+        "items": {
+            "type": "object",
+            "properties": {
+                "firstName": {
+                    "type": "string"
+                },
+                "lastName": {
+                    "type": "string"
+                },
+                "gender": {
+                    "enum": ["male", "female"]
+                },
+                "age": {
+                    "description": "Age in years",
+                    "type": "integer",
+                    "minimum": 0
+                },
+                "job": {
+                    "$ref": "job"
+                }
+            },
+            "required": ["firstName", "lastName"]
+        }
+    };
+
+    var job = {
+        "title": "Job description",
+        "type": "object",
+        "properties": {
+            "company": {
+                "type": "string"
+            },
+            "role": {
+                "type": "string"
+            }
+        }
+    };
+
+    var json = [
+        {
+            firstName: 'John',
+            lastName: 'Doe',
+            gender: 'male',
+            age: 28,
+            job: {
+                company: 'freelance',
+                role: 'developer'
+            }
+        },
+        {
+            firstName: 'Susan',
+            lastName: 'Smith',
+            gender: null,
+            age: 28,
+            job: {
+                company: 'freelance',
+                role: 'sales'
+            }
+        }
+    ];
+
+    var templates = [
+        {
+            text: 'Employee',
+            title: 'Insert a new employee',
+            className: 'jsoneditor-type-object',
+            field: 'employee',
+            value: {
+                firstName: '',
+                lastName: '',
+                gender: '',
+                age: '',
+                job: {
+                    company: '',
+                    role: ''
+                }
+            }
+        }
+    ]
+
+    var options = {
+        schema: schema,
+        schemaRefs: {"job": job},
+        templates: templates
+    };
+
+    // create the editor
+    var container = document.getElementById('jsoneditor');
+    var editor = new JSONEditor(container, options, json);
+</script>
+</body>
+</html>