fix unit test

Author: tejasd

firebase-perf/src/test/java/com/google/firebase/perf/FirebasePerformanceTestBase.java

@@ -25,13 +25,15 @@
 import com.google.firebase.perf.config.ConfigResolver;
 import com.google.firebase.perf.session.PerfSession;
 import com.google.firebase.perf.session.SessionManager;
+import com.google.firebase.perf.session.gauges.GaugeCounter;
 import com.google.firebase.perf.util.ImmutableBundle;
 import org.junit.After;
 import org.junit.Before;
-import org.robolectric.shadows.ShadowLog;
+import org.junit.BeforeClass;
 import org.robolectric.shadows.ShadowPackageManager;
 
 public class FirebasePerformanceTestBase {
+
   /**
    * The following values are needed by Firebase to identify the project and application that all
    * data stored in Firebase databases gets associated with. This is important to determine data
@@ -54,9 +56,14 @@ public class FirebasePerformanceTestBase {
 
   protected Context appContext;
 
+  @BeforeClass
+  public static void setUpBeforeClass() {
+    // TODO(b/394127311): Explore removing this.
+    GaugeCounter.INSTANCE.resetCounter();
+  }
+
   @Before
   public void setUpFirebaseApp() {
-    ShadowLog.stream = System.out;
     appContext = ApplicationProvider.getApplicationContext();
 
     ShadowPackageManager shadowPackageManager = shadowOf(appContext.getPackageManager());

firebase-perf/src/test/java/com/google/firebase/perf/session/gauges/GaugeManagerTest.java

@@ -130,7 +130,6 @@ public void setUp() {
   @After
   public void tearDown() {
     shadowOf(Looper.getMainLooper()).idle();
-    GaugeCounter.INSTANCE.resetCounter();
   }
 
   @Test
@@ -332,11 +331,11 @@ public void stopCollectingGauges_invalidGaugeCollectionFrequency_appInForeground
   }
 
   @Test
-  public void testGaugeCounterStartsAJobToConsumeTheGeneratedMetrics() {
+  public void testGaugeCounterStartsAJobToConsumeTheGeneratedMetrics() throws InterruptedException {
     PerfSession fakeSession = createTestSession(1);
     testGaugeManager.setApplicationProcessState(ApplicationProcessState.FOREGROUND);
     testGaugeManager.startCollectingGauges(fakeSession);
-    GaugeCounter.setGaugeManager(testGaugeManager);
+    GaugeCounter.INSTANCE.setGaugeManager(testGaugeManager);
 
     // There's no job to log the gauges.
     assertThat(fakeScheduledExecutorService.isEmpty()).isTrue();
@@ -378,7 +377,7 @@ public void testUpdateAppStateHandlesMultipleAppStates() {
     fakeSession.setGaugeAndEventCollectionEnabled(true);
     testGaugeManager.setApplicationProcessState(ApplicationProcessState.FOREGROUND);
     testGaugeManager.startCollectingGauges(fakeSession);
-    GaugeCounter.setGaugeManager(testGaugeManager);
+    GaugeCounter.INSTANCE.setGaugeManager(testGaugeManager);
 
     // Generate metrics that don't exceed the GaugeCounter.MAX_COUNT.
     generateMetricsAndIncrementCounter(10);
@@ -430,7 +429,7 @@ public void testGaugeManagerHandlesMultipleSessionIds() {
     fakeSession.setGaugeAndEventCollectionEnabled(true);
     testGaugeManager.setApplicationProcessState(ApplicationProcessState.BACKGROUND);
     testGaugeManager.startCollectingGauges(fakeSession);
-    GaugeCounter.setGaugeManager(testGaugeManager);
+    GaugeCounter.INSTANCE.setGaugeManager(testGaugeManager);
 
     // Generate metrics that don't exceed the GaugeCounter.MAX_COUNT.
     generateMetricsAndIncrementCounter(10);
@@ -520,7 +519,7 @@ public void testStopCollectingGaugesCreatesOneLastJobToConsumeAnyPendingMetrics(
             + recordedGaugeMetric.getCpuMetricReadingsCount();
     assertThat(recordedGaugeMetricsCount).isEqualTo(2);
 
-    // TODO(b/394127311): Investigate why this isn't 0.
+    // TODO(b/394127311): Investigate why this isn't 0 on local runs.
     //    assertThat(GaugeCounter.INSTANCE.count()).isEqualTo(0);
   }