From 6c33fbbef0041e7a3795c506afbd04074d7808a5 Mon Sep 17 00:00:00 2001
From: saipraneeth <2506664+msaipraneeth@users.noreply.github.com>
Date: Thu, 14 Dec 2023 07:18:36 +0000
Subject: [PATCH] add test case with array object in first level

---
 tests/test_utils_build_entities_from_data.py | 45 ++++++++++++++++++++
 1 file changed, 45 insertions(+)

diff --git a/tests/test_utils_build_entities_from_data.py b/tests/test_utils_build_entities_from_data.py
index e23d13d..dfec3d6 100644
--- a/tests/test_utils_build_entities_from_data.py
+++ b/tests/test_utils_build_entities_from_data.py
@@ -67,3 +67,48 @@ def test_single_object_one_level():
                 EntityPath("country", False, is_attribute=True)
             ]
         )
+
+def test_single_object_one_level_array():
+    """test write to single object"""
+    test_data = """
+{
+  "name": "sai",
+  "email": "saipraneeth@example.com",
+  "city": [{
+    "name": "San Francisco",
+    "country": "United States"
+  },
+  {
+    "name": "New York",
+    "country": "United States"
+  }]
+}"""
+    data = json.loads(test_data)
+    entities = build_entities_from_data(data)
+    assert len(entities.entities) == 1
+    for _ in entities.entities:
+        assert len(_.values) == 3
+        assert _.values[0:2] == [["sai"], ["saipraneeth@example.com"]]
+        assert _.values[2][0].startswith("urn:x-ulid:")
+    assert len(entities.schema.paths) == 3
+    assert entities.schema == EntitySchema(
+        type_uri="",
+        paths=[
+            EntityPath("name", False, is_attribute=True),
+            EntityPath("email", False, is_attribute=True),
+            EntityPath("city", True, is_attribute=False),
+        ]
+    )
+    # Validate sub entities
+    for _ in entities.sub_entities:
+        assert len(list(_.entities)) == 2
+        for _entity in _.entities:
+            assert len(_entity.values) == 2
+        assert _.schema == EntitySchema(
+            type_uri="",
+            paths=[
+                EntityPath("name", False, is_attribute=True),
+                EntityPath("country", False, is_attribute=True)
+            ]
+        )
+