adding extra tests

rsksmart · Feb 5, 2024 · 87a3e6a · 87a3e6a
1 parent c9695c6
commit 87a3e6a
Show file tree

Hide file tree

Showing 2 changed files with 39 additions and 1 deletion.
diff --git a/rskj-core/src/test/java/co/rsk/config/RskSystemPropertiesTest.java b/rskj-core/src/test/java/co/rsk/config/RskSystemPropertiesTest.java
@@ -281,4 +281,41 @@ void testPeerLastSession() throws IOException {
 
         assertEquals(expectedPeers, actualPeers);
     }
+
+    @Test
+    void checkPeerLastSessionProperty(){
+        TestSystemProperties testSystemProperties = new TestSystemProperties(rawConfig ->
+                ConfigFactory.parseString("{" +
+                        "peer {\n" +
+                                "  usePeersFromLastSession = true\n" +
+                                "}" +
+                        " }").withFallback(rawConfig));
+
+        assertTrue(testSystemProperties.usePeersFromLastSession());
+
+        testSystemProperties = new TestSystemProperties(rawConfig ->
+                ConfigFactory.parseString("{" +
+                        "peer {\n" +
+                        "  usePeersFromLastSession = false\n" +
+                        "}" +
+                        " }").withFallback(rawConfig));
+        assertFalse(testSystemProperties.usePeersFromLastSession());
+
+        testSystemProperties = new TestSystemProperties();
+
+        assertFalse(testSystemProperties.usePeersFromLastSession());
+    }
+
+    @Test
+    void checkLastSessionPeersFilePathProperty(){
+        TestSystemProperties testSystemProperties = new TestSystemProperties(rawConfig ->
+                ConfigFactory.parseString("{" +
+                        "database {\n" +
+                        "  dir = \"/dbdir\"\n" +
+                        "}" +
+                        " }").withFallback(rawConfig));
+
+        assertEquals("/dbdir/lastKnewPeers.txt", testSystemProperties.getLastKnewPeersFilePath().toString());
+    }
+
 }
diff --git a/rskj-core/src/test/java/co/rsk/net/discovery/DiscoveredPeersPersistenceServiceTest.java b/rskj-core/src/test/java/co/rsk/net/discovery/DiscoveredPeersPersistenceServiceTest.java
@@ -52,7 +52,8 @@ void testStop() throws IOException {
 
         service.stop();
 
-        verify(fileDataSaver, times(1)).saveDataIntoFile(anyString(), any(Path.class));
+        verify(fileDataSaver, times(1)).saveDataIntoFile(eq("peer1\npeer2\npeer3\n"), eq(filePath));
+
     }
 
 }