fix: lifecycle owner being garbage collected

awslabs · Mar 13, 2024 · ed554ec · ed554ec
1 parent 8fef15e
commit ed554ec
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/...eam/src/test/java/software/aws/solution/clickstream/ActivityLifecycleManagerUnitTest.java b/...eam/src/test/java/software/aws/solution/clickstream/ActivityLifecycleManagerUnitTest.java
@@ -47,6 +47,7 @@ public final class ActivityLifecycleManagerUnitTest {
     private Application.ActivityLifecycleCallbacks callbacks;
     private Log log;
     private LifecycleRegistry lifecycle;
+    private LifecycleOwner owner;
     private ActivityLifecycleManager lifecycleManager;
 
     /**
@@ -65,8 +66,8 @@ public void setup() throws Exception {
         this.callbacks = lifecycleManager;
         log = mock(Log.class);
         ReflectUtil.modifyFiled(this.callbacks, "LOG", log);
-
-        lifecycle = new LifecycleRegistry(mock(LifecycleOwner.class));
+        owner = mock(LifecycleOwner.class);
+        lifecycle = new LifecycleRegistry(owner);
         lifecycleManager.startLifecycleTracking(ApplicationProvider.getApplicationContext(), lifecycle);
     }
 

diff --git a/clickstream/src/test/java/software/aws/solution/clickstream/AutoRecordEventClientTest.java b/clickstream/src/test/java/software/aws/solution/clickstream/AutoRecordEventClientTest.java
@@ -73,6 +73,7 @@ public class AutoRecordEventClientTest {
     private ClickstreamContext clickstreamContext;
     private AutoRecordEventClient client;
     private LifecycleRegistry lifecycle;
+    private LifecycleOwner owner;
 
     /**
      * prepare AutoRecordEventClient and context.
@@ -95,7 +96,8 @@ public void setup() {
         callbacks = new ActivityLifecycleManager(clickstreamManager);
 
         ActivityLifecycleManager lifecycleManager = new ActivityLifecycleManager(clickstreamManager);
-        lifecycle = new LifecycleRegistry(mock(LifecycleOwner.class));
+        owner = mock(LifecycleOwner.class);
+        lifecycle = new LifecycleRegistry(owner);
         lifecycleManager.startLifecycleTracking(ApplicationProvider.getApplicationContext(), lifecycle);
     }