status-im/ethereumj-personal
2
0
Fork 0
mirror of https://github.com/status-im/ethereumj-personal.git synced 2025-01-11 12:24:13 +00:00
Code Issues Projects Releases Wiki Activity

testing for trie rollback

Browse Source
This commit is contained in:
romanman 2014-09-14 23:57:38 +02:00
parent 777ba0f17a
commit cee31133da
2 changed files with 70 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
ethereumj-core/src/main/java/org/ethereum/trie/Cache.java
View File

@ -114,4 +114,19 @@ public class Cache {
public DB getDb() {
return db;
}
public String cacheDump(){
StringBuffer cacheDump = new StringBuffer();
for (ByteArrayWrapper key : nodes.keySet()){
Node node = nodes.get(key);
if (node.getValue() != null)
cacheDump.append(key.toString()).append(" : ").append(node.getValue().toString()).append("\n");
}
return cacheDump.toString();
}
}

61
ethereumj-core/src/test/java/org/ethereum/trie/TrieTest.java
View File

@ -12,11 +12,7 @@ import java.net.URISyntaxException;
import java.net.URL;
import java.nio.charset.StandardCharsets;
import java.nio.file.Files;
import java.util.Arrays;
import java.util.HashMap;
import java.util.Iterator;
import java.util.List;
import java.util.Random;
import java.util.*;
import org.ethereum.core.AccountState;
import org.ethereum.db.DatabaseImpl;
@ -646,7 +642,7 @@ public class TrieTest {
if(massiveUpdateFromDBEnabled) {
List<String> randomWords = Arrays.asList(randomDictionary.split(","));
HashMap<String, String> testerMap = new HashMap<>();
Trie trie = new Trie(mockDb);
Random generator = new Random();
@ -693,6 +689,59 @@ public class TrieTest {
}
@Test
public void testRollbackTrie() throws URISyntaxException, IOException {
Trie trieSingle = new Trie(mockDb);
URL massiveUpload_1 = ClassLoader
.getSystemResource("trie/massive-upload.dmp");
File file = new File(massiveUpload_1.toURI());
List<String> strData = Files.readAllLines(file.toPath(), StandardCharsets.UTF_8);
List<byte[]> roots = new ArrayList<>();
Map<String, String> trieDumps = new HashMap<>();
for (int i = 0; i < 100; ++i){
String[] keyVal= strData.get(i).split("=");
if (keyVal[0].equals("*"))
trieSingle.delete(keyVal[1].trim());
else
trieSingle.update(keyVal[0].trim(), keyVal[1].trim());
byte[] hash = trieSingle.getRootHash();
roots.add(hash);
String key = Hex.toHexString(hash);
String dump = trieSingle.getTrieDump();
System.out.print("(" + i + ") : ");
System.out.println(key + " ==> " + dump);
trieDumps.put(key, dump);
System.out.print("(cache) : \n");
System.out.println(trieSingle.getCache().cacheDump());
}
// compare all 100 rollback dumps and
// the originaly saved dumps
for (int i = 1; i < roots.size(); ++i){
byte[] root = roots.get(i);
trieSingle.setRoot(root);
String currDump = trieSingle.getTrieDump();
String originDump = trieDumps.get(Hex.toHexString(root));
System.out.println(currDump);
Assert.assertEquals(currDump, originDump);
}
}
@Test
public void testGetFromRootNode() {
Trie trie1 = new Trie(mockDb);