package org.tron.program;

import java.io.File;

import lombok.extern.slf4j.Slf4j;

import org.junit.AfterClass;

import org.junit.Assert;

import org.junit.BeforeClass;

import org.junit.Test;

import org.tron.common.application.TronApplicationContext;

import org.tron.common.application.Application;

import org.tron.common.application.ApplicationFactory;

import org.tron.common.overlay.client.DatabaseGrpcClient;

import org.tron.core.Constant;

import org.tron.core.config.DefaultConfig;

import org.tron.core.config.args.Args;

import org.tron.core.services.RpcApiService;

import org.tron.protos.Protocol.Block;

import org.tron.protos.Protocol.DynamicProperties;

@Slf4j

public class SolidityNodeTest {

private static TronApplicationContext context;

private static RpcApiService rpcApiService;

private static Application appT;

private static String dbPath = "output_witness_test";

static {

Args.setParam(new String[]{"-d", dbPath}, Constant.TEST_CONF);

context = new TronApplicationContext(DefaultConfig.class);

Args.getInstance().setSolidityNode(true);

appT = ApplicationFactory.create(context);

rpcApiService = context.getBean(RpcApiService.class);

}

/**

* init db.

*/

@BeforeClass

public static void init() {

rpcApiService.start();

}

/**

* remo db when after test.

*/

@AfterClass

public static void removeDb() {

Args.clearParam();

rpcApiService.stop();

File dbFolder = new File(dbPath);

if (deleteFolder(dbFolder)) {

logger.info("Release resources successful.");

} else {

logger.info("Release resources failure.");

}

context.destroy();

}

private static Boolean deleteFolder(File index) {

if (!index.isDirectory() || index.listFiles().length <= 0) {

return index.delete();

}

for (File file : index.listFiles()) {

if (null != file && !deleteFolder(file)) {

return false;

}

}

return index.delete();

}

@Test

public void testSolidityArgs() {

Assert.assertNotNull(Args.getInstance().getTrustNodeAddr());

Assert.assertTrue(Args.getInstance().isSolidityNode());

}

@Test

public void testSolidityGrpcCall() {

DatabaseGrpcClient databaseGrpcClient = null;

String addr = Args.getInstance().getTrustNodeAddr();

try {

databaseGrpcClient = new DatabaseGrpcClient(addr);

} catch (Exception e) {

logger.error("Failed to create database grpc client {}", addr);

}

Assert.assertNotNull(databaseGrpcClient);

DynamicProperties dynamicProperties = databaseGrpcClient.getDynamicProperties();

Assert.assertNotNull(dynamicProperties);

Block genisisBlock = databaseGrpcClient.getBlock(0);

Assert.assertNotNull(genisisBlock);

Assert.assertFalse(genisisBlock.getTransactionsList().isEmpty());

}

}