add witness config test

Author: Federico2014

framework/src/test/java/org/tron/core/config/args/LocalWitnessTest.java

@@ -18,17 +18,24 @@
 import static org.junit.Assert.fail;
 
 import com.google.common.collect.Lists;
+import java.io.IOException;
 import org.junit.Assert;
 import org.junit.Before;
+import org.junit.Rule;
 import org.junit.Test;
+import org.junit.rules.TemporaryFolder;
 import org.tron.common.utils.LocalWitnesses;
 import org.tron.common.utils.PublicMethod;
+import org.tron.core.Constant;
 
 public class LocalWitnessTest {
 
   private final LocalWitnesses localWitness = new LocalWitnesses();
   private static final String PRIVATE_KEY = PublicMethod.getRandomPrivateKey();
 
+  @Rule
+  public final TemporaryFolder temporaryFolder = new TemporaryFolder();
+
   @Before
   public void setLocalWitness() {
     localWitness
@@ -89,4 +96,24 @@ public void testConstructor() {
     LocalWitnesses localWitnesses3 = new LocalWitnesses();
     Assert.assertNull(localWitnesses3.getWitnessAccountAddress(true));
   }
+
+  @Test
+  public void testLocalWitnessConfig() throws IOException {
+    Args.setParam(
+        new String[]{"--output-directory", temporaryFolder.newFolder().toString(), "--debug"},
+        "config-localtest.conf");
+    LocalWitnesses witness = Args.getLocalWitnesses();
+    Assert.assertNotNull(witness.getPrivateKey());
+    Assert.assertNotNull(witness.getWitnessAccountAddress(true));
+  }
+
+  @Test
+  public void testNullLocalWitnessConfig() throws IOException {
+    Args.setParam(
+        new String[]{"--output-directory", temporaryFolder.newFolder().toString(), "--debug"},
+        Constant.TEST_CONF);
+    LocalWitnesses witness = Args.getLocalWitnesses();
+    Assert.assertNull(witness.getPrivateKey());
+    Assert.assertNull(witness.getWitnessAccountAddress(true));
+  }
 }

framework/src/test/java/org/tron/core/net/services/RelayServiceTest.java

@@ -188,8 +188,8 @@ public void testNullWitnessAddress() {
 
       Object result = isActiveWitnessMethod.invoke(service);
       Assert.assertFalse(Boolean.TRUE.equals(result));
-    } catch (NoSuchMethodException | NoSuchFieldException |
-        IllegalAccessException | InvocationTargetException e) {
+    } catch (NoSuchMethodException | NoSuchFieldException
+        | IllegalAccessException | InvocationTargetException e) {
       Assert.fail("Reflection invocation failed: " + e.getMessage());
     }
   }