Fixed test cases

Sunbird-RC · Jan 29, 2024 · 5454b73 · 5454b73
1 parent 95bd6f5
commit 5454b73
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/java/registry/src/main/java/dev/sunbirdrc/registry/helper/RegistryHelper.java b/java/registry/src/main/java/dev/sunbirdrc/registry/helper/RegistryHelper.java
@@ -1057,7 +1057,7 @@ public Vertex deleteEntity(String entityName, String entityId, String userId) th
         Shard shard = shardManager.activateShard(shardId);
         ReadConfigurator configurator = ReadConfiguratorFactory.getOne(false);
         Vertex vertex = registryService.deleteEntityById(shard, entityName, userId, recordId.getUuid());
-        VertexReader vertexReader = new VertexReader(shard.getDatabaseProvider(), vertex.graph(), configurator, uuidPropertyName, definitionsManager);
+        VertexReader vertexReader = new VertexReader(shard.getDatabaseProvider(), vertex.graph(), configurator, uuidPropertyName, definitionsManager, true);
         JsonNode deletedNode = JsonNodeFactory.instance.objectNode().set(entityName, vertexReader.constructObject(vertex));
         if(notificationEnabled) notificationHelper.sendNotification(deletedNode, DELETE);
         return vertex;

diff --git a/java/registry/src/test/java/dev/sunbirdrc/registry/service/impl/RegistryServiceImplTest.java b/java/registry/src/test/java/dev/sunbirdrc/registry/service/impl/RegistryServiceImplTest.java
@@ -334,7 +334,7 @@ public void shouldUpdateArrayFieldsInEntity() throws Exception {
 		when(shard.getDatabaseProvider()).thenReturn(mockDatabaseProvider);
 		String instituteOsid = addInstituteToGraph();
 		ReadConfigurator readConfigurator = ReadConfiguratorFactory.getForUpdateValidation();
-		VertexReader vertexReader = new VertexReader(mockDatabaseProvider, graph, readConfigurator, "osid", definitionsManager);
+		VertexReader vertexReader = new VertexReader(mockDatabaseProvider, graph, readConfigurator, "osid", definitionsManager, true );
 		JsonNode instituteNode = vertexReader.read("Institute", instituteOsid);
 		ObjectNode affiliationNode = (ObjectNode) instituteNode.get("Institute").get("affiliation").get(0);
 		ArrayNode arrayNode = JsonNodeFactory.instance.arrayNode();
@@ -365,7 +365,7 @@ public void shouldUpdateTextFieldsInEntity() throws Exception {
 		when(shard.getDatabaseProvider()).thenReturn(mockDatabaseProvider);
 		String instituteOsid = addInstituteToGraph();
 		ReadConfigurator readConfigurator = ReadConfiguratorFactory.getForUpdateValidation();
-		VertexReader vertexReader = new VertexReader(mockDatabaseProvider, graph, readConfigurator, "osid", definitionsManager);
+		VertexReader vertexReader = new VertexReader(mockDatabaseProvider, graph, readConfigurator, "osid", definitionsManager, true);
 		JsonNode instituteNode = vertexReader.read("Institute", instituteOsid);
 		((ObjectNode)instituteNode.get("Institute")).set("instituteName", JsonNodeFactory.instance.textNode("Holy Cross"));
 		when(shard.getShardLabel()).thenReturn("");