Fix instrumentation registry to get target context for LevelDbManagerTest

Test: atest MediaProviderTests
Bug: 394152098
Flag: EXEMPT test fix
Change-Id: I046f86950fd2e368ef9cfb96a3f927a51df89e2d

1 files changed, 2 insertions, 2 deletions

diff --git a/tests/src/com/android/providers/media/leveldb/LevelDBManagerTest.java b/tests/src/com/android/providers/media/leveldb/LevelDBManagerTest.java
index f684e8ce5..191f11a59 100644
--- a/tests/src/com/android/providers/media/leveldb/LevelDBManagerTest.java
+++ b/tests/src/com/android/providers/media/leveldb/LevelDBManagerTest.java
@@ -21,8 +21,8 @@ import static com.google.common.truth.Truth.assertThat;
 
 import android.content.Context;
 
+import androidx.test.InstrumentationRegistry;
 import androidx.test.ext.junit.runners.AndroidJUnit4;
-import androidx.test.platform.app.InstrumentationRegistry;
 
 import org.junit.Test;
 import org.junit.runner.RunWith;
@@ -39,7 +39,7 @@ public class LevelDBManagerTest {
 
     @Test
     public void testLevelDbOperations() {
-        final Context context = InstrumentationRegistry.getInstrumentation().getContext();
+        final Context context = InstrumentationRegistry.getTargetContext();
         String levelDBFile = "test-leveldb";
         final String levelDBPath = context.getFilesDir().getPath() + "/" + levelDBFile;
         LevelDBInstance levelDBInstance = LevelDBManager.getInstance(levelDBPath);