1 /* 2 * Copyright (C) 2017 The Android Open Source Project 3 * 4 * Licensed under the Apache License, Version 2.0 (the "License"); 5 * you may not use this file except in compliance with the License. 6 * You may obtain a copy of the License at 7 * 8 * http://www.apache.org/licenses/LICENSE-2.0 9 * 10 * Unless required by applicable law or agreed to in writing, software 11 * distributed under the License is distributed on an "AS IS" BASIS, 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 * See the License for the specific language governing permissions and 14 * limitations under the License. 15 */ 16 17 package com.android.apksig.util; 18 19 import java.io.IOException; 20 import java.nio.ByteBuffer; 21 import org.junit.runner.RunWith; 22 import org.junit.runners.JUnit4; 23 24 /** 25 * Tests for the {@link DataSink} returned by {@link DataSinks#newInMemoryDataSink()}. 26 */ 27 @RunWith(JUnit4.class) 28 public class InMemoryDataSinkTest extends DataSinkTestBase<ReadableDataSink> { 29 30 @Override 31 protected CloseableWithDataSink<ReadableDataSink> createDataSink() { 32 return CloseableWithDataSink.of(DataSinks.newInMemoryDataSink()); 33 } 34 35 @Override 36 protected ByteBuffer getContents(ReadableDataSink dataSink) throws IOException { 37 if (dataSink.size() > Integer.MAX_VALUE) { 38 throw new IOException("Too much data: " + dataSink.size()); 39 } 40 return dataSink.getByteBuffer(0, (int) dataSink.size()); 41 } 42 43 } 44