1 /*
2 * ====================================================================
3 * Licensed to the Apache Software Foundation (ASF) under one
4 * or more contributor license agreements. See the NOTICE file
5 * distributed with this work for additional information
6 * regarding copyright ownership. The ASF licenses this file
7 * to you under the Apache License, Version 2.0 (the
8 * "License"); you may not use this file except in compliance
9 * with the License. You may obtain a copy of the License at
10 *
11 * http://www.apache.org/licenses/LICENSE-2.0
12 *
13 * Unless required by applicable law or agreed to in writing,
14 * software distributed under the License is distributed on an
15 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
16 * KIND, either express or implied. See the License for the
17 * specific language governing permissions and limitations
18 * under the License.
19 * ====================================================================
20 *
21 * This software consists of voluntary contributions made by many
22 * individuals on behalf of the Apache Software Foundation. For more
23 * information on the Apache Software Foundation, please see
24 * <http://www.apache.org/>.
25 *
26 */
27 package org.apache.hc.client5.http.impl.cache;
28
29 import static org.mockito.Mockito.mock;
30 import static org.mockito.Mockito.verify;
31 import static org.mockito.Mockito.when;
32
33 import java.io.ByteArrayInputStream;
34
35 import org.apache.hc.core5.http.HttpEntity;
36 import org.apache.hc.core5.http.io.entity.EntityUtils;
37 import org.apache.hc.core5.util.ByteArrayBuffer;
38 import org.junit.jupiter.api.Assertions;
39 import org.junit.jupiter.api.Test;
40
41 class TestCombinedEntity {
42
43 @Test
44 void testCombinedEntityBasics() throws Exception {
45 final HttpEntity httpEntity = mock(HttpEntity.class);
46 when(httpEntity.getContent()).thenReturn(
47 new ByteArrayInputStream(new byte[] { 6, 7, 8, 9, 10 }));
48
49 final ByteArrayBuffer buf = new ByteArrayBuffer(1024);
50 final byte[] tmp = new byte[] { 1, 2, 3, 4, 5 };
51 buf.append(tmp, 0, tmp.length);
52 final CombinedEntity entity = new CombinedEntity(httpEntity, buf);
53 Assertions.assertEquals(-1, entity.getContentLength());
54 Assertions.assertFalse(entity.isRepeatable());
55 Assertions.assertTrue(entity.isStreaming());
56
57 Assertions.assertArrayEquals(new byte[] { 1, 2, 3, 4, 5, 6, 7, 8, 9, 10 }, EntityUtils.toByteArray(entity));
58
59 verify(httpEntity).getContent();
60
61 entity.close();
62
63 verify(httpEntity).close();
64 }
65
66 }