1   /*
2    *  Licensed to the Apache Software Foundation (ASF) under one
3    *  or more contributor license agreements.  See the NOTICE file
4    *  distributed with this work for additional information
5    *  regarding copyright ownership.  The ASF licenses this file
6    *  to you under the Apache License, Version 2.0 (the
7    *  "License"); you may not use this file except in compliance
8    *  with the License.  You may obtain a copy of the License at
9    *
10   *    http://www.apache.org/licenses/LICENSE-2.0
11   *
12   *  Unless required by applicable law or agreed to in writing,
13   *  software distributed under the License is distributed on an
14   *  "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
15   *  KIND, either express or implied.  See the License for the
16   *  specific language governing permissions and limitations
17   *  under the License.
18   *
19   */
20  package org.apache.mina.transport.socket.nio;
21  
22  import java.net.InetSocketAddress;
23  import java.net.SocketAddress;
24  
25  import org.apache.mina.core.future.ConnectFuture;
26  import org.apache.mina.core.service.IoConnector;
27  import org.apache.mina.core.service.IoHandler;
28  import org.apache.mina.transport.AbstractTrafficControlTest;
29  
30  /**
31   * Tests suspending and resuming reads and writes for the datagram
32   * transport type.
33   *
34   * @author The Apache MINA Project (dev@mina.apache.org)
35   * @version $Id: DatagramTrafficControlTest.java 671827 2008-06-26 08:49:48Z jvermillard $
36   */
37  public class DatagramTrafficControlTest extends AbstractTrafficControlTest {
38  
39      public DatagramTrafficControlTest() {
40          super(new NioDatagramAcceptor());
41      }
42  
43      @Override
44      protected ConnectFuture connect(int port, IoHandler handler)
45              throws Exception {
46          IoConnector connector = new NioDatagramConnector();
47          connector.setHandler(handler);
48          return connector.connect(new InetSocketAddress("localhost", port));
49      }
50  
51      @Override
52      protected SocketAddress createServerSocketAddress(int port) {
53          return new InetSocketAddress(port);
54      }
55  
56      @Override
57      protected int getPort(SocketAddress address) {
58          return ((InetSocketAddress) address).getPort();
59      }
60  }