diff --git a/testutils/src/main/java/com/google/android/exoplayer2/testutil/DataSourceContractTest.java b/testutils/src/main/java/com/google/android/exoplayer2/testutil/DataSourceContractTest.java
index 5f5a237124..f07baf97ac 100644
--- a/testutils/src/main/java/com/google/android/exoplayer2/testutil/DataSourceContractTest.java
+++ b/testutils/src/main/java/com/google/android/exoplayer2/testutil/DataSourceContractTest.java
@@ -88,7 +88,10 @@ public abstract class DataSourceContractTest {
       DataSource dataSource = createDataSource();
       try {
         long length = dataSource.open(new DataSpec(resource.getUri()));
-        byte[] data = readToEnd(dataSource, resource);
+        byte[] data =
+            resource.isEndOfInputExpected()
+                ? Util.readToEnd(dataSource)
+                : Util.readExactly(dataSource, resource.getExpectedBytes().length);
 
         assertThat(length).isEqualTo(resource.getExpectedResolvedLength());
         assertThat(data).isEqualTo(resource.getExpectedBytes());
@@ -112,7 +115,10 @@ public abstract class DataSourceContractTest {
         long length =
             dataSource.open(
                 new DataSpec.Builder().setUri(resource.getUri()).setPosition(3).build());
-        byte[] data = readToEnd(dataSource, resource);
+        byte[] data =
+            resource.isEndOfInputExpected()
+                ? Util.readToEnd(dataSource)
+                : Util.readExactly(dataSource, resource.getExpectedBytes().length - 3);
 
         if (resource.getExpectedResolvedLength() != C.LENGTH_UNSET) {
           assertThat(length).isEqualTo(resource.getExpectedResolvedLength() - 3);
@@ -139,7 +145,10 @@ public abstract class DataSourceContractTest {
       try {
         long length =
             dataSource.open(new DataSpec.Builder().setUri(resource.getUri()).setLength(4).build());
-        byte[] data = readToEnd(dataSource, resource);
+        byte[] data =
+            resource.isEndOfInputExpected()
+                ? Util.readToEnd(dataSource)
+                : Util.readExactly(dataSource, /* length= */ 4);
 
         assertThat(length).isEqualTo(4);
         byte[] expectedData = Arrays.copyOf(resource.getExpectedBytes(), 4);
@@ -168,7 +177,10 @@ public abstract class DataSourceContractTest {
                     .setPosition(2)
                     .setLength(2)
                     .build());
-        byte[] data = readToEnd(dataSource, resource);
+        byte[] data =
+            resource.isEndOfInputExpected()
+                ? Util.readToEnd(dataSource)
+                : Util.readExactly(dataSource, /* length= */ 2);
 
         assertThat(length).isEqualTo(2);
         byte[] expectedData = Arrays.copyOfRange(resource.getExpectedBytes(), 2, 4);
@@ -201,13 +213,6 @@ public abstract class DataSourceContractTest {
     }
   }
 
-  private static byte[] readToEnd(DataSource dataSource, TestResource expectedResource)
-      throws IOException {
-    return expectedResource.isEndOfInputExpected()
-        ? Util.readToEnd(dataSource)
-        : Util.readExactly(dataSource, expectedResource.getExpectedBytes().length);
-  }
-
   /** Information about a resource that can be used to test the {@link DataSource} instance. */
   public static final class TestResource {