View Javadoc
1   /*
2    * Copyright 2012 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.netty5.testsuite.transport.socket;
17  
18  import io.netty5.bootstrap.Bootstrap;
19  import io.netty5.buffer.api.BufferAllocator;
20  import io.netty5.channel.ChannelOption;
21  import io.netty5.channel.socket.SocketProtocolFamily;
22  import io.netty5.testsuite.transport.AbstractComboTestsuiteTest;
23  import io.netty5.testsuite.transport.TestsuitePermutation;
24  import io.netty5.util.NetUtil;
25  
26  import java.net.InetSocketAddress;
27  import java.net.SocketAddress;
28  import java.util.List;
29  
30  public abstract class AbstractDatagramTest extends AbstractComboTestsuiteTest<Bootstrap, Bootstrap> {
31      @Override
32      protected List<TestsuitePermutation.BootstrapComboFactory<Bootstrap, Bootstrap>> newFactories() {
33          return SocketTestPermutation.INSTANCE.datagram(socketProtocolFamily());
34      }
35  
36      @Override
37      protected void configure(Bootstrap b1, Bootstrap b2,
38                               BufferAllocator bufferAllocator) {
39          b1.option(ChannelOption.BUFFER_ALLOCATOR, bufferAllocator);
40          b2.option(ChannelOption.BUFFER_ALLOCATOR, bufferAllocator);
41      }
42  
43      protected SocketAddress newSocketAddress() {
44          SocketProtocolFamily family = socketProtocolFamily();
45          switch (family) {
46              case INET:
47                  return new InetSocketAddress(NetUtil.LOCALHOST4, 0);
48              case INET6:
49                  return new InetSocketAddress(NetUtil.LOCALHOST6, 0);
50              case UNIX:
51                  return SocketTestPermutation.newDomainSocketAddress();
52              default:
53                  throw new AssertionError();
54          }
55      }
56  
57      protected SocketProtocolFamily protocolFamily() {
58          return SocketProtocolFamily.INET;
59      }
60  
61      protected SocketProtocolFamily groupProtocolFamily() {
62          return protocolFamily();
63      }
64  
65      protected SocketProtocolFamily socketProtocolFamily() {
66          return protocolFamily();
67      }
68  }