diff --git a/libraries/datasource/build.gradle b/libraries/datasource/build.gradle
index 2e4c036f79..7706ffc0b3 100644
--- a/libraries/datasource/build.gradle
+++ b/libraries/datasource/build.gradle
@@ -42,6 +42,7 @@ dependencies {
     androidTestImplementation 'androidx.test:runner:' + androidxTestRunnerVersion
     androidTestImplementation 'com.linkedin.dexmaker:dexmaker:' + dexmakerVersion
     androidTestImplementation 'com.linkedin.dexmaker:dexmaker-mockito:' + dexmakerVersion
+    androidTestImplementation 'com.squareup.okhttp3:mockwebserver:' + okhttpVersion
     androidTestImplementation(project(modulePrefix + 'test-utils')) {
         exclude module: modulePrefix.substring(1) + 'library-core'
     }
diff --git a/libraries/datasource/src/test/java/androidx/media3/datasource/DefaultHttpDataSourceTest.java b/libraries/datasource/src/androidTest/java/androidx/media3/datasource/DefaultHttpDataSourceTest.java
similarity index 100%
rename from libraries/datasource/src/test/java/androidx/media3/datasource/DefaultHttpDataSourceTest.java
rename to libraries/datasource/src/androidTest/java/androidx/media3/datasource/DefaultHttpDataSourceTest.java