查看本类的 API文档回源码主页即时通讯网 - 即时通讯开发者社区!
1   /*
2    * Copyright 2020 The Netty Project
3    *
4    * The Netty Project licenses this file to you under the Apache License,
5    * version 2.0 (the "License"); you may not use this file except in compliance
6    * with the License. You may obtain a copy of the License at:
7    *
8    *   https://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, WITHOUT
12   * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
13   * License for the specific language governing permissions and limitations
14   * under the License.
15   */
16  package io.netty.channel.unix.tests;
17  
18  import io.netty.buffer.ByteBuf;
19  import io.netty.buffer.Unpooled;
20  import io.netty.buffer.UnpooledByteBufAllocator;
21  import io.netty.buffer.UnpooledDirectByteBuf;
22  import io.netty.channel.unix.IovArray;
23  import org.junit.jupiter.api.Test;
24  
25  import static org.junit.jupiter.api.Assertions.assertEquals;
26  import static org.junit.jupiter.api.Assertions.assertNotEquals;
27  import static org.junit.jupiter.api.Assertions.assertTrue;
28  
29  public abstract class IovArrayTest {
30  
31      @Test
32      public void testNotFailsWihtoutMemoryAddress() {
33          ByteBuf buffer = new NoMemoryAddressByteBuf(128);
34          IovArray array = new IovArray(buffer);
35  
36          ByteBuf buf = Unpooled.directBuffer().writeZero(8);
37          ByteBuf buf2 = new NoMemoryAddressByteBuf(8).writeZero(8);
38          assertTrue(array.add(buf, 0, buf.readableBytes()));
39          assertTrue(array.add(buf, 0, buf2.readableBytes()));
40          assertEquals(2, array.count());
41          assertEquals(16, array.size());
42          assertTrue(buf.release());
43          assertTrue(buf2.release());
44          assertNotEquals(-1, array.memoryAddress(0));
45          array.release();
46          assertEquals(0, buffer.refCnt());
47      }
48  
49      private static final class NoMemoryAddressByteBuf extends UnpooledDirectByteBuf {
50  
51          NoMemoryAddressByteBuf(int capacity) {
52              super(UnpooledByteBufAllocator.DEFAULT, capacity, Integer.MAX_VALUE);
53          }
54  
55          @Override
56          public boolean hasMemoryAddress() {
57              return false;
58          }
59  
60          @Override
61          public long memoryAddress() {
62              throw new UnsupportedOperationException();
63          }
64      }
65  }