Add basic tests on PreloadingNetworkStoreClient

Signed-off-by: BOUHOURS Antoine <[email protected]>
powsybl · Oct 18, 2024 · e50a021 · e50a021
1 parent 10ed186
commit e50a021
Show file tree

Hide file tree

Showing 2 changed files with 463 additions and 0 deletions.
diff --git a/...e-client/src/test/java/com/powsybl/network/store/client/PreloadingAllCollectionsTest.java b/...e-client/src/test/java/com/powsybl/network/store/client/PreloadingAllCollectionsTest.java
@@ -6,11 +6,16 @@
  */
 package com.powsybl.network.store.client;
 
+import com.powsybl.iidm.network.VariantManagerConstants;
 import com.powsybl.network.store.iidm.impl.CachedNetworkStoreClient;
 import com.powsybl.network.store.iidm.impl.OfflineNetworkStoreClient;
+import com.powsybl.network.store.model.NetworkAttributes;
+import com.powsybl.network.store.model.Resource;
 import com.powsybl.network.store.model.ResourceType;
 import org.junit.Test;
 
+import java.time.ZonedDateTime;
+import java.util.List;
 import java.util.UUID;
 import java.util.concurrent.ForkJoinPool;
 
@@ -30,4 +35,32 @@ public void testWithAllCollections() {
         client.getSubstations(networkUuid, 0);
         assertFalse(client.shouldLoadAllCollectionsNeededForBusView(ResourceType.SUBSTATION));
     }
+
+    @Test
+    public void testWithAllCollectionsDeleteNetwork() {
+        var client = new PreloadingNetworkStoreClient(new CachedNetworkStoreClient(new OfflineNetworkStoreClient()), true, ForkJoinPool.commonPool());
+        UUID networkUuid = UUID.fromString("7928181c-7977-4592-ba19-88027e4254e4");
+        Resource<NetworkAttributes> n1 = Resource.networkBuilder()
+                .id("n1")
+                .attributes(NetworkAttributes.builder()
+                        .uuid(networkUuid)
+                        .variantId(VariantManagerConstants.INITIAL_VARIANT_ID)
+                        .caseDate(ZonedDateTime.parse("2015-01-01T00:00:00.000Z"))
+                        .build())
+                .build();
+        client.createNetworks(List.of(n1));
+        assertTrue(client.shouldLoadAllCollectionsNeededForBusView(ResourceType.SUBSTATION));
+        client.getSubstations(networkUuid, 0);
+        assertFalse(client.shouldLoadAllCollectionsNeededForBusView(ResourceType.SUBSTATION));
+        // When we delete the network, we should load again all collection needed for bus view on next get
+        client.deleteNetwork(networkUuid);
+        assertTrue(client.shouldLoadAllCollectionsNeededForBusView(ResourceType.SUBSTATION));
+        client.createNetworks(List.of(n1));
+        assertTrue(client.shouldLoadAllCollectionsNeededForBusView(ResourceType.SUBSTATION));
+        client.getSubstations(networkUuid, 0);
+        assertFalse(client.shouldLoadAllCollectionsNeededForBusView(ResourceType.SUBSTATION));
+        // When we delete the network variant, we should load again all collection needed for bus view on next get
+        client.deleteNetwork(networkUuid, 0);
+        assertTrue(client.shouldLoadAllCollectionsNeededForBusView(ResourceType.SUBSTATION));
+    }
 }