add test: TcpTransport

src/test/java/logicface/TcpTransportTest.java

@@ -0,0 +1,112 @@
+package logicface;
+
+import component.ComponentException;
+import encoding.BlockException;
+import encoding.EncoderException;
+import encoding.SelfEncodingBase;
+import mgmt.MgmtException;
+import org.junit.Test;
+import packet.LpPacket;
+import packet.PacketException;
+import util.ByteHelper;
+
+import java.io.IOException;
+import java.io.InputStream;
+import java.io.OutputStream;
+import java.net.InetSocketAddress;
+import java.net.ServerSocket;
+import java.net.Socket;
+import java.net.SocketAddress;
+import java.nio.channels.SocketChannel;
+import java.util.Arrays;
+
+/*
+ * @Author: Wang Feng
+ * @Description:
+ * @Version: 1.0.0
+ * @Date: 10:38 2021/5/7
+ * @Copyright: MIN-Group；国家重大科技基础设施——未来网络北大实验室；深圳市信息论与未来网络重点实验室
+ */
+public class TcpTransportTest {
+    @Test
+    public void testSend() throws IOException, LogicFaceException, EncoderException, BlockException, MgmtException, ComponentException, PacketException {
+        // 初始化一个tcp连接
+        TcpTransport tcpTransport = new TcpTransport();
+        SocketChannel socketChannel = SocketChannel.open();
+        SocketAddress socketAddress = new InetSocketAddress("127.0.0.1", 60000);
+        socketChannel.connect(socketAddress);
+        tcpTransport.init(socketChannel);
+
+        // 发送数据包
+        LpPacket lpPacket = new LpPacket();
+        lpPacket.setId(1);
+        lpPacket.setFragmentNum(1);
+        lpPacket.setFragmentSeq(0);
+        lpPacket.setValue(new byte[]{4, 5, 6, 7});
+        tcpTransport.send(lpPacket);
+        System.out.println("send: " +
+                Arrays.toString(new SelfEncodingBase().selfWireEncode(lpPacket).getRaw()));
+    }
+
+    @Test
+    public void testReceive() throws IOException, LogicFaceException, EncoderException, BlockException, MgmtException, ComponentException, PacketException {
+        // 初始化一个tcp连接
+        TcpTransport tcpTransport = new TcpTransport();
+        SocketChannel socketChannel = SocketChannel.open();
+        SocketAddress socketAddress = new InetSocketAddress("127.0.0.1", 60000);
+        socketChannel.connect(socketAddress);
+        tcpTransport.init(socketChannel);
+
+        // 发送数据包
+        LpPacket lpPacket = new LpPacket();
+        lpPacket.setId(1);
+        lpPacket.setFragmentNum(1);
+        lpPacket.setFragmentSeq(0);
+        lpPacket.setValue(new byte[]{4, 5, 6, 7});
+        tcpTransport.send(lpPacket);
+        System.out.println("send: " +
+                Arrays.toString(new SelfEncodingBase().selfWireEncode(lpPacket).getRaw()));
+
+        // 接收数据包
+        LpPacket newLp = tcpTransport.receive();
+        if (newLp == null) {
+            System.out.println("lp null");
+        }
+        System.out.println("receive: " +
+                Arrays.toString(new SelfEncodingBase().selfWireEncode(newLp).getRaw()));
+    }
+
+    /**
+     * 开启一个tcp服务器，端口号为60000
+     */
+    @Test
+    public void startTCPServer() {
+        ServerSocket socket = null;
+        OutputStream outputStream = null;
+        try {
+            //建立基站
+            socket = new ServerSocket(60000);
+            //开始开启接收模式,接到后返回客户端的socket对象
+            Socket client = socket.accept();
+
+            byte[] bytes = new byte[1500];
+
+            // 循环接收数据，并打印
+            InputStream inputStream = client.getInputStream();
+            int len = inputStream.read(bytes);
+//            inputStream.close();
+            System.out.println(Arrays.toString(ByteHelper.getLenBytes(bytes, 0, len)));
+
+            //获取向客户端发送消息的对象流
+            outputStream = client.getOutputStream();
+            //向客户端写数据
+            outputStream.write(ByteHelper.getLenBytes(bytes, 0, len));
+
+            inputStream.close();
+            outputStream.close();
+            socket.close();
+        } catch (Exception e) {
+            e.printStackTrace();
+        }
+    }
+}