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 java.nio.charset.StandardCharsets; 22 import org.junit.Test; 23 import org.junit.runner.RunWith; 24 import org.junit.runners.JUnit4; 25 26 /** 27 * Tests for the {@link DataSource} returned by {@link DataSources#asDataSource(ByteBuffer)}. 28 */ 29 @RunWith(JUnit4.class) 30 public class DataSourceFromByteBufferTest extends DataSourceTestBase { 31 32 @Test testChangesToBufferPosAndLimitNotVisible()33 public void testChangesToBufferPosAndLimitNotVisible() throws Exception { 34 ByteBuffer buf = ByteBuffer.wrap("abcdefgh".getBytes(StandardCharsets.UTF_8)); 35 buf.position(1); 36 buf.limit(4); 37 DataSource ds = DataSources.asDataSource(buf); 38 buf.position(2); 39 buf.limit(buf.capacity()); 40 assertGetByteBufferEquals("bcd", ds, 0, (int) ds.size()); 41 assertFeedEquals("bcd", ds, 0, (int) ds.size()); 42 assertSliceEquals("bcd", ds, 0, (int) ds.size()); 43 assertCopyToEquals("bcd", ds, 0, (int) ds.size()); 44 } 45 46 @Override createDataSource(byte[] contents)47 protected CloseableWithDataSource createDataSource(byte[] contents) throws IOException { 48 return CloseableWithDataSource.of(DataSources.asDataSource(ByteBuffer.wrap(contents))); 49 } 50 } 51