fix tests

open-telemetry · Dec 8, 2023 · f137f5a · f137f5a
1 parent 40b40f2
commit f137f5a
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/instrumentation/src/test/java/io/opentelemetry/android/OpenTelemetryRumBuilderTest.java b/instrumentation/src/test/java/io/opentelemetry/android/OpenTelemetryRumBuilderTest.java
@@ -34,6 +34,7 @@
 import java.time.Duration;
 import java.util.List;
 import java.util.function.Function;
+import org.junit.jupiter.api.BeforeEach;
 import org.junit.jupiter.api.Test;
 import org.junit.jupiter.api.extension.ExtendWith;
 import org.mockito.ArgumentCaptor;
@@ -49,11 +50,17 @@ class OpenTelemetryRumBuilderTest {
     final InMemorySpanExporter spanExporter = InMemorySpanExporter.create();
 
     @Mock Application application;
+    @Mock android.content.Context applicationContext;
     @Mock Activity activity;
     @Mock ApplicationStateListener listener;
 
     @Captor ArgumentCaptor<Application.ActivityLifecycleCallbacks> activityCallbacksCaptor;
 
+    @BeforeEach
+    void setup() {
+        when(application.getApplicationContext()).thenReturn(applicationContext);
+    }
+
     @Test
     void shouldRegisterApplicationStateWatcher() {
         makeBuilder().build();