local testscript

algorand · Aug 22, 2022 · 8a91825 · 8a91825
1 parent 0427e98
commit 8a91825
Show file tree

Hide file tree

Showing 2 changed files with 31 additions and 3 deletions.
diff --git a/src/main/java/com/algorand/algosdk/logic/Logic.java b/src/main/java/com/algorand/algosdk/logic/Logic.java
@@ -170,9 +170,6 @@ public static VarintResult getUVarint(byte [] buffer, int bufferOffset) {
      * @param program
      */
     public static void sanityCheckProgram(final byte[] program) {
-        if (Base64.isBase64(program))
-            throw new IllegalArgumentException("program should not be b64 encoded");
-
         boolean isAddress = false;
         try {
             new Address(new String(program));
@@ -184,6 +181,9 @@ public static void sanityCheckProgram(final byte[] program) {
         }
         if (isAddress)
             throw new IllegalArgumentException("requesting program bytes, but get Algorand address");
+
+        if (Base64.isBase64(program))
+            throw new IllegalArgumentException("program should not be b64 encoded");
     }
 
     @Deprecated

diff --git a/src/test/java/com/algorand/algosdk/logic/TestLogic.java b/src/test/java/com/algorand/algosdk/logic/TestLogic.java
@@ -3,10 +3,12 @@
 import com.google.common.collect.ImmutableList;
 import org.junit.jupiter.api.Test;
 
+import java.nio.charset.StandardCharsets;
 import java.util.*;
 
 import static com.algorand.algosdk.logic.Logic.*;
 import static org.assertj.core.api.Assertions.*;
+import static org.junit.jupiter.api.Assertions.*;
 
 public class TestLogic {
     @Test
@@ -447,4 +449,30 @@ public void testCheckProgramV6() throws Exception {
             assertThat(valid).isTrue();
         }
     }
+
+    @Test
+    public void testSanityCheckProgram() {
+        {
+            // pass in address
+            String addressNotProg = "DN7MBMCL5JQ3PFUQS7TMX5AH4EEKOBJVDUF4TCV6WERATKFLQF4MQUPZTA";
+            assertThatThrownBy(() -> sanityCheckProgram(addressNotProg.getBytes(StandardCharsets.UTF_8)))
+                    .isInstanceOf(IllegalArgumentException.class)
+                    .hasMessage("requesting program bytes, but get Algorand address");
+        }
+
+        {
+            // pass in b64 encoded msg
+            String exampleB64 = "gqRtc2lng6ZzdWJzaWeTgqJwa8QgG37AsEvqYbeWkJfmy/QH4QinBTUdC8mKvrEiCairgXihc8RAdvZ3y9GsInBPutdwKc7Jy+an13CcjSV1lcvRAYQKYOxXwfgT5B/mK14R57ueYJTYyoDO8zBY6kQmBalWkm95AIGicGvEIAljMglTc4nwdWcRdzmRx9A+G3PIxPUr9q/wGqJc+cJxgaJwa8Qg5/D4TQaBHfnzHI2HixFV9GcdUaGFwgCQhmf0SVhwaKGjdGhyAqF2AaN0eG6Jo2FtdM0TiKNmZWXOAANPqKJmds4ADtbco2dlbq10ZXN0bmV0LXYzMS4womx2zgAO2sSkbm90ZcQItFF5Ofz60nGjcmN2xCAbfsCwS+pht5aQl+bL9AfhCKcFNR0LyYq+sSIJqKuBeKNzbmTEII2StImQAXOgTfpDWaNmamr86ixCoF3Zwfc+66VHgDfppHR5cGWjcGF5";
+            assertThatThrownBy(() -> sanityCheckProgram(exampleB64.getBytes(StandardCharsets.UTF_8)))
+                    .isInstanceOf(IllegalArgumentException.class)
+                    .hasMessage("program should not be b64 encoded");
+        }
+
+        {
+            byte[] program = {
+                    0x06, 0x31, 0x3f, 0x15, (byte) 0x81, 0x40, 0x12, 0x33, 0x00, 0x3e, 0x15, (byte) 0x81, 0x0a, 0x12, 0x10
+            };
+            assertDoesNotThrow(() -> sanityCheckProgram(program));
+        }
+    }
 }