feat: support setter for logMode and logLevel (#94)

amplitude · Aug 3, 2023 · d07f830 · d07f830
1 parent 38492ae
commit d07f830
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/src/main/java/com/amplitude/AmplitudeLog.java b/src/main/java/com/amplitude/AmplitudeLog.java
@@ -7,6 +7,10 @@ public void setLogMode(LogMode logMode) {
     this.logMode = logMode;
   }
 
+  public LogMode getLogMode() {
+    return this.logMode;
+  }
+
   public void debug(String tag, String message) {
     log(tag, message, LogMode.DEBUG);
   }
@@ -40,5 +44,9 @@ public enum LogMode {
     LogMode(int level) {
       this.level = level;
     }
+
+    public int getLogLevel() {
+      return this.level;
+    }
   }
 }
diff --git a/src/test/java/com/amplitude/AmplitudeLogTest.java b/src/test/java/com/amplitude/AmplitudeLogTest.java
@@ -40,6 +40,14 @@ public void restoreStreams() throws IOException {
     System.setErr(originalErr);
   }
 
+  @Test
+  public void testLogMode() {
+    AmplitudeLog.LogMode logMode = AmplitudeLog.LogMode.DEBUG;
+    amplitudeLog.setLogMode(logMode);
+    assertEquals(amplitudeLog.getLogMode(),logMode);
+    assertEquals(amplitudeLog.getLogMode().getLogLevel(),1);
+  }
+
   @ParameterizedTest
   @MethodSource("logArguments")
   public void testLog(