Merge pull request #1844 from tronprotocol/stest_Case_merge_to_master

Stest case merge to master

Author: wangzichichi

src/test/java/stest/tron/wallet/account/WalletTestAccount006.java

@@ -70,6 +70,9 @@ public void beforeClass() {
 
   @Test(enabled = true)
   public void testGetAccountNet() {
+    ecKey = new ECKey(Utils.getRandom());
+    account006Address = ecKey.getAddress();
+    account006Key = ByteArray.toHexString(ecKey.getPrivKeyBytes());
     //Sendcoin to this account
     ByteString addressBS1 = ByteString.copyFrom(account006Address);
     Account request1 = Account.newBuilder().setAddress(addressBS1).build();

src/test/java/stest/tron/wallet/account/WalletTestAccount011.java

@@ -1,5 +1,6 @@
 package stest.tron.wallet.account;
 
+import com.google.protobuf.ByteString;
 import io.grpc.ManagedChannel;
 import io.grpc.ManagedChannelBuilder;
 import java.util.concurrent.TimeUnit;
@@ -9,15 +10,20 @@
 import org.testng.annotations.BeforeClass;
 import org.testng.annotations.BeforeSuite;
 import org.testng.annotations.Test;
-import org.tron.api.GrpcAPI.AccountResourceMessage;
+import org.tron.api.GrpcAPI.BytesMessage;
+import org.tron.api.GrpcAPI.EasyTransferByPrivateMessage;
+import org.tron.api.GrpcAPI.EasyTransferMessage;
 import org.tron.api.GrpcAPI.EmptyMessage;
 import org.tron.api.WalletGrpc;
 import org.tron.api.WalletSolidityGrpc;
 import org.tron.common.crypto.ECKey;
 import org.tron.common.utils.ByteArray;
 import org.tron.common.utils.Utils;
 import org.tron.core.Wallet;
+import org.tron.protos.Contract;
+import org.tron.protos.Protocol;
 import org.tron.protos.Protocol.Account;
+import org.tron.protos.Protocol.TransactionSign;
 import stest.tron.wallet.common.client.Configuration;
 import stest.tron.wallet.common.client.Parameter.CommonConstant;
 import stest.tron.wallet.common.client.utils.PublicMethed;
@@ -71,6 +77,64 @@ public void testgenerateAddress() {
     blockingStubSolidity.generateAddress(builder.build());
   }
 
+  @Test(enabled = true)
+  public void testeasyTransfer() {
+    ecKey1 = new ECKey(Utils.getRandom());
+    account011Address = ecKey1.getAddress();
+    account011Key = ByteArray.toHexString(ecKey1.getPrivKeyBytes());
+    PublicMethed.printAddress(testKey002);
+    PublicMethed.printAddress(account011Key);
+    Assert.assertTrue(PublicMethed.sendcoin(account011Address,10000000L,fromAddress,
+        testKey002,blockingStubFull));
+
+
+    String password = Long.toString(System.currentTimeMillis());
+    BytesMessage.Builder builder = BytesMessage.newBuilder();
+    builder.setValue(ByteString.copyFrom(password.getBytes()));
+    BytesMessage result = blockingStubFull.createAddress(builder.build());
+    byte[] address = result.getValue().toByteArray();
+
+    Account fromAccount = PublicMethed.queryAccount(account011Key,blockingStubFull);
+    final long beforeFromBaslance = fromAccount.getBalance();
+
+    EasyTransferByPrivateMessage.Builder builder1 = EasyTransferByPrivateMessage.newBuilder();
+    builder1.setPrivateKey(ByteString.copyFrom(ByteArray.fromHexString(account011Key)));
+    builder1.setToAddress(ByteString.copyFrom(address));
+    builder1.setAmount(2000000L);
+    Assert.assertTrue(blockingStubFull.easyTransferByPrivate(builder1.build())
+        .getResult().getCodeValue() == 0);
+
+    fromAccount = PublicMethed.queryAccount(account011Key,blockingStubFull);
+    final long afterFromBaslance = fromAccount.getBalance();
+    logger.info("beforeFromBaslance is " + beforeFromBaslance);
+    logger.info("afterFromBaslance is  " + afterFromBaslance);
+    logger.info("min is " + (beforeFromBaslance - afterFromBaslance));
+    Assert.assertTrue(beforeFromBaslance - afterFromBaslance == 2000000L + 100000);
+
+
+    EasyTransferMessage.Builder builder2 = EasyTransferMessage.newBuilder();
+    builder2.setPassPhrase(ByteString.copyFrom(password.getBytes()));
+    builder2.setToAddress(ByteString.copyFrom(account011Address));
+    builder2.setAmount(100);
+    Assert.assertTrue(blockingStubFull.easyTransfer(builder2.build()).getResult()
+        .getCodeValue() == 0);
+
+    Contract.TransferContract.Builder builder5 = Contract.TransferContract.newBuilder();
+    ByteString bsTo = ByteString.copyFrom(address);
+    ByteString bsOwner = ByteString.copyFrom(account011Address);
+    builder5.setToAddress(bsTo);
+    builder5.setOwnerAddress(bsOwner);
+    builder5.setAmount(100L);
+
+    Contract.TransferContract contract = builder5.build();
+    Protocol.Transaction transaction = blockingStubFull.createTransaction(contract);
+
+    TransactionSign.Builder builder4 = TransactionSign.newBuilder();
+    builder4.setPrivateKey(ByteString.copyFrom(ByteArray.fromHexString(account011Key)));
+    builder4.setTransaction(transaction);
+    blockingStubFull.getTransactionSign(builder4.build());
+    blockingStubFull.getTransactionSign2(builder4.build());
+  }
 
   @AfterClass(enabled = true)
   public void shutdown() throws InterruptedException {