test(case): add transaction test cases

Author: forfreeday

framework/src/test/java/org/tron/core/db/BlockIndexStoreTest.java

@@ -0,0 +1,64 @@
+package org.tron.core.db;
+
+import com.google.protobuf.ByteString;
+import javax.annotation.Resource;
+import org.junit.Assert;
+import org.junit.Test;
+import org.tron.common.BaseTest;
+import org.tron.common.utils.ByteArray;
+import org.tron.common.utils.Sha256Hash;
+import org.tron.core.Constant;
+import org.tron.core.capsule.BlockCapsule;
+import org.tron.core.capsule.BytesCapsule;
+import org.tron.core.config.args.Args;
+import org.tron.core.exception.ItemNotFoundException;
+
+public class BlockIndexStoreTest extends BaseTest {
+
+  @Resource
+  private BlockIndexStore blockIndexStore;
+
+  static {
+    Args.setParam(
+            new String[]{
+                "--output-directory", dbPath()
+            },
+            Constant.TEST_CONF
+    );
+  }
+
+  private BlockCapsule getBlockCapsule(long number) {
+    return new BlockCapsule(number, Sha256Hash.ZERO_HASH,
+            System.currentTimeMillis(), ByteString.EMPTY);
+  }
+
+  @Test
+  public void testPut() {
+    BlockCapsule blockCapsule = getBlockCapsule(1);
+    blockIndexStore.put(blockCapsule.getBlockId());
+    byte[] key = ByteArray.fromLong(blockCapsule.getBlockId().getNum());
+    Assert.assertTrue(blockIndexStore.has(key));
+  }
+
+  @Test
+  public void testGet() throws ItemNotFoundException {
+    BlockCapsule blockCapsule = getBlockCapsule(1);
+    blockIndexStore.put(blockCapsule.getBlockId());
+    byte[] key = ByteArray.fromLong(blockCapsule.getBlockId().getNum());
+    BytesCapsule bytesCapsule = blockIndexStore.get(key);
+    Assert.assertNotNull(bytesCapsule);
+  }
+
+  @Test
+  public void testDelete() throws ItemNotFoundException {
+    BlockCapsule blockCapsule = getBlockCapsule(1);
+    blockIndexStore.put(blockCapsule.getBlockId());
+    byte[] key = ByteArray.fromLong(blockCapsule.getBlockId().getNum());
+    BytesCapsule bytesCapsule = blockIndexStore.get(key);
+    Assert.assertNotNull(bytesCapsule);
+
+    blockIndexStore.delete(key);
+    BytesCapsule capsule = blockIndexStore.getUnchecked(key);
+    Assert.assertNull(capsule.getData());
+  }
+}

framework/src/test/java/org/tron/core/services/http/GetTransactionInfoByBlockNumServletTest.java

@@ -0,0 +1,40 @@
+package org.tron.core.services.http;
+
+import static java.nio.charset.StandardCharsets.UTF_8;
+import static org.tron.common.utils.client.utils.HttpMethed.createRequest;
+
+import javax.annotation.Resource;
+import org.apache.http.client.methods.HttpPost;
+import org.junit.Assert;
+import org.junit.Test;
+import org.springframework.mock.web.MockHttpServletRequest;
+import org.springframework.mock.web.MockHttpServletResponse;
+import org.tron.common.BaseTest;
+import org.tron.core.Constant;
+import org.tron.core.config.args.Args;
+
+public class GetTransactionInfoByBlockNumServletTest extends BaseTest {
+
+  @Resource
+  private GetTransactionInfoByBlockNumServlet getTransactionInfoByBlockNumServlet;
+
+  static {
+    Args.setParam(
+            new String[]{
+                "--output-directory", dbPath(),
+            }, Constant.TEST_CONF
+    );
+  }
+
+  @Test
+  public void testGetTransactionInfoByBlockNum() {
+    String jsonParam = "{\"num\" : 100}";
+    MockHttpServletRequest request = createRequest(HttpPost.METHOD_NAME);
+    request.setContentType("application/json");
+    request.setContent(jsonParam.getBytes(UTF_8));
+    MockHttpServletResponse response = new MockHttpServletResponse();
+
+    getTransactionInfoByBlockNumServlet.doPost(request, response);
+    Assert.assertEquals(200, response.getStatus());
+  }
+}

framework/src/test/java/org/tron/core/services/http/GetTransactionInfoByIdServletTest.java

@@ -0,0 +1,42 @@
+package org.tron.core.services.http;
+
+import static java.nio.charset.StandardCharsets.UTF_8;
+import static org.tron.common.utils.client.utils.HttpMethed.createRequest;
+
+import javax.annotation.Resource;
+import org.apache.http.client.methods.HttpPost;
+import org.junit.Assert;
+import org.junit.Test;
+import org.springframework.mock.web.MockHttpServletRequest;
+import org.springframework.mock.web.MockHttpServletResponse;
+import org.tron.common.BaseTest;
+import org.tron.core.Constant;
+import org.tron.core.config.args.Args;
+
+public class GetTransactionInfoByIdServletTest extends BaseTest {
+
+  @Resource
+  private GetTransactionInfoByIdServlet getTransactionInfoByIdServlet;
+
+  static {
+    Args.setParam(
+            new String[]{
+                "--output-directory", dbPath(),
+            }, Constant.TEST_CONF
+    );
+  }
+
+  @Test
+  public void testGetInfoById() {
+    String jsonParam = "{\"value\" : "
+            + "\"309b6fa3d01353e46f57dd8a8f276"
+            + "11f98e392b50d035cef213f2c55225a8bd2\"}";
+    MockHttpServletRequest request = createRequest(HttpPost.METHOD_NAME);
+    request.setContentType("application/json");
+    request.setContent(jsonParam.getBytes(UTF_8));
+    MockHttpServletResponse response = new MockHttpServletResponse();
+
+    getTransactionInfoByIdServlet.doPost(request, response);
+    Assert.assertEquals(200, response.getStatus());
+  }
+}

framework/src/test/java/org/tron/core/services/http/GetTransactionListFromPendingServletTest.java

@@ -0,0 +1,38 @@
+package org.tron.core.services.http;
+
+import static org.junit.Assert.assertEquals;
+import static org.tron.common.utils.client.utils.HttpMethed.createRequest;
+
+import javax.annotation.Resource;
+
+import org.apache.http.client.methods.HttpGet;
+import org.junit.Test;
+import org.springframework.mock.web.MockHttpServletRequest;
+import org.springframework.mock.web.MockHttpServletResponse;
+import org.tron.common.BaseTest;
+import org.tron.core.Constant;
+import org.tron.core.config.args.Args;
+
+
+public class GetTransactionListFromPendingServletTest extends BaseTest {
+
+  @Resource
+  private GetTransactionListFromPendingServlet getTransactionListFromPendingServlet;
+
+  static {
+    Args.setParam(
+            new String[]{
+                "--output-directory", dbPath(),
+            }, Constant.TEST_CONF
+    );
+  }
+
+  @Test
+  public void testGet() {
+    MockHttpServletRequest request = createRequest(HttpGet.METHOD_NAME);
+    MockHttpServletResponse response = new MockHttpServletResponse();
+    getTransactionListFromPendingServlet.doPost(request, response);
+    assertEquals(200, response.getStatus());
+  }
+
+}