Fix test fixture creating (pre executing storage)

ethereum · Oct 15, 2018 · 9ba7f37 · 9ba7f37
1 parent 9cb18e7
commit 9ba7f37
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/ethereumj-core/src/main/java/org/ethereum/vm/program/invoke/ProgramInvokeMockImpl.java b/ethereumj-core/src/main/java/org/ethereum/vm/program/invoke/ProgramInvokeMockImpl.java
@@ -240,7 +240,7 @@ public Repository getRepository() {
 
     @Override
     public Repository getOrigRepository() {
-        return this.repository;
+        return this.origRepository;
     }
 
     @Override
@@ -252,6 +252,10 @@ public void setRepository(Repository repository) {
         this.repository = repository;
     }
 
+    public void setOrigRepository(Repository repository) {
+        this.origRepository = repository.clone();
+    }
+
     @Override
     public int getCallDeep() {
         return 0;

diff --git a/ethereumj-core/src/test/java/org/ethereum/vm/VMMemoryOpTest.java b/ethereumj-core/src/test/java/org/ethereum/vm/VMMemoryOpTest.java
@@ -1023,6 +1023,7 @@ private void setStorageToOne(VM vm) {
         while (!program.isStopped())
             vm.step(program);
         invoke.getRepository().commit();
+        invoke.setOrigRepository(invoke.getRepository());
     }
 
     @Test // SSTORE EIP1283