When codelist is used on an array, the enum is set for items in that …

…array

Needed for Open-Telecoms-Data/lib-cove-ofds#81

CHANGELOG.md

@@ -9,6 +9,7 @@ and this project adheres to [Semantic Versioning](http://semver.org/spec/v2.0.0.
 ### Added
 
 - Process anyOf and allOf and well as oneOf https://github.com/OpenDataServices/compile-to-json-schema/issues/28
+- When codelist is used on an array, the enum is set for items in that array
 
 ### Removed
 

compiletojsonschema/compiletojsonschema.py

@@ -93,7 +93,12 @@ def __process(self, source):
                         if len(row) > 0 and row[0]:
                             values.append(row[0])
                 if values:
-                    out["enum"] = values
+                    if out.get("type") == "array" and isinstance(
+                        out.get("items"), dict
+                    ):
+                        out["items"]["enum"] = values
+                    else:
+                        out["enum"] = values
             else:
                 raise CodeListNotFoundException(
                     "Can not find codelist: " + source["codelist"]

tests/fixtures/codelists/schema-closed-codelist-array.json

@@ -0,0 +1,15 @@
+{
+  "properties": {
+    "pet": {
+      "title": "Pet",
+      "codelist": "pets.csv",
+      "openCodelist": false,
+      "type": "array",
+      "items": {
+        "type": [
+          "string"
+        ]
+      }
+    }
+  }
+}

tests/test_codelist.py

@@ -96,3 +96,27 @@ def test_closed_codelist():
     out = ctjs.get()
 
     assert out["properties"]["pet"]["enum"] == ["Dog", "Cat", "Parrot"]
+
+
+def test_closed_codelist_array():
+
+    input_filename = os.path.join(
+        os.path.dirname(os.path.realpath(__file__)),
+        "fixtures",
+        "codelists",
+        "schema-closed-codelist-array.json",
+    )
+
+    codelist_dir = os.path.join(
+        os.path.dirname(os.path.realpath(__file__)),
+        "fixtures",
+        "codelists",
+    )
+
+    ctjs = CompileToJsonSchema(
+        input_filename=input_filename, codelist_base_directory=codelist_dir
+    )
+    out = ctjs.get()
+
+    assert not out["properties"]["pet"].get("enum")
+    assert out["properties"]["pet"]["items"]["enum"] == ["Dog", "Cat", "Parrot"]