Github user HuangWHWHW commented on a diff in the pull request:
https://github.com/apache/flink/pull/1077#discussion_r38282466
--- Diff:
flink-staging/flink-streaming/flink-streaming-core/src/test/java/org/apache/flink/streaming/api/functions/SocketTextStreamFunctionTest.java
---
@@ -0,0 +1,237 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements. See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License. You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.apache.flink.streaming.api.functions;
+
+import
org.apache.flink.streaming.api.functions.source.SocketTextStreamFunction;
+import org.junit.Test;
+
+import java.io.ByteArrayInputStream;
+import java.lang.reflect.Field;
+import java.net.Socket;
+import java.util.ArrayList;
+import java.util.List;
+
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertTrue;
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.when;
+
+
+public class SocketTextStreamFunctionTest {
+ //Actual text
+ /*
+ Lorem ipsum dolor sit amet, consectetur adipiscing elit. Cras sagittis
nisl non euismod fermentum. Curabitur lacinia vehicula enim quis tristique.
Suspendisse imperdiet arcu sed bibendum vulputate. Sed vitae nisl vitae turpis
dapibus lacinia in id elit. Integer lorem dolor, porttitor ut nisi in,
tincidunt sodales leo. Aliquam tristique dui sit amet odio bibendum, et rutrum
turpis auctor. Morbi sit amet mollis augue, ac rutrum velit. Vestibulum
suscipit finibus sapien, et congue enim laoreet consequat.
+
+ Integer aliquam metus iaculis risus hendrerit maximus. Suspendisse
vestibulum nibh ac mauris cursus molestie sit amet vel turpis. Nulla et posuere
orci. Aliquam dui quam, posuere vitae erat vitae, finibus commodo ipsum.
Aliquam eu dui quis arcu porttitor sollicitudin. Integer sodales finibus
ullamcorper. Praesent et felis tempor, laoreet libero eget, consequat nisl.
Aenean molestie rutrum lorem, ac cursus nisl dapibus vitae.
+
+ Quisque sodales dui et sem bibendum semper. Pellentesque luctus leo
nec lacus euismod pellentesque. Phasellus a metus dignissim risus auctor
lacinia. Class aptent taciti sociosqu ad litora torquent per conubia nostra,
per inceptos himenaeos. Aenean consectetur bibendum imperdiet. Etiam dignissim
rutrum enim, non volutpat nisi condimentum sed. Quisque condimentum ultrices
est sit amet facilisis.
+
+ Ut vitae volutpat odio. Sed eget vestibulum libero, eu tincidunt
lorem. Nam pretium nulla nisl. Maecenas fringilla nunc ut turpis consectetur,
et fringilla sem placerat. Etiam nec scelerisque nisi, at sodales ligula.
Aliquam euismod faucibus egestas. Curabitur eget enim quam. Praesent convallis
mattis lobortis. Pellentesque a consectetur nisl. Duis molestie diam est. Nam a
malesuada augue. Vivamus enim massa, luctus ac elit ut, vestibulum laoreet
nulla. Curabitur pellentesque vel mi eget tempus. Donec cursus et leo quis
viverra.
+
+ In ac imperdiet ex, nec aliquet erat. Nullam sit amet enim in dolor
finibus convallis id eu nibh. Fusce aliquam convallis orci aliquam.
+ */
+ // Generated 5 paragraphs, 290 words, 2000 bytes of Lorem Ipsum
+
+ private static final String content = "Lorem ipsum dolor sit amet,
consectetur adipiscing elit. Cras sagittis nisl non euismod fermentum. " +
+ "Curabitur lacinia vehicula enim quis tristique. Suspendisse
imperdiet arcu sed bibendum vulputate. " +
+ "Sed vitae nisl vitae turpis dapibus lacinia in id elit.
Integer lorem dolor, " +
+ "porttitor ut nisi in, tincidunt sodales leo. Aliquam
tristique dui sit amet odio bibendum, " +
+ "et rutrum turpis auctor. Morbi sit amet mollis augue, ac
rutrum velit. " +
+ "Vestibulum suscipit finibus sapien, et congue enim laoreet
consequat.\r\nInteger " +
+ "aliquam metus iaculis risus hendrerit maximus. Suspendisse
vestibulum nibh ac " +
+ "mauris cursus molestie sit amet vel turpis. Nulla et posuere
orci. " +
+ "Aliquam dui quam, posuere vitae erat vitae, finibus commodo
ipsum. Aliquam " +
+ "eu dui quis arcu porttitor sollicitudin. Integer sodales
finibus ullamcorper. " +
+ "Praesent et felis tempor, laoreet libero eget, consequat
nisl. " +
+ "Aenean molestie rutrum lorem, ac cursus nisl dapibus
vitae.\r\nQuisque sodales " +
+ "dui et sem bibendum semper. Pellentesque luctus leo nec lacus
euismod pellentesque. " +
+ "Phasellus a metus dignissim risus auctor lacinia. Class " +
+ "aptent taciti sociosqu ad litora torquent per conubia nostra,
per inceptos himenaeos" +
+ ". Aenean consectetur bibendum imperdiet. Etiam dignissim
rutrum enim, " +
+ "non volutpat nisi condimentum sed. Quisque condimentum
ultrices est sit amet " +
+ "facilisis.\r\nUt vitae volutpat odio. Sed eget vestibulum
libero, eu " +
+ "tincidunt lorem. Nam pretium nulla nisl. Maecenas fringilla
nunc ut turpis consectetur, " +
+ "et fringilla sem placerat. Etiam nec scelerisque nisi, at
sodales ligula. Aliquam " +
+ "euismod faucibus egestas. Curabitur eget enim quam. Praesent
convallis mattis lobortis. " +
+ "Pellentesque a consectetur nisl. Duis molestie diam est. Nam
a malesuada augue. " +
+ "Vivamus enim massa, luctus ac elit ut, vestibulum laoreet
nulla. " +
+ "Curabitur pellentesque vel mi eget tempus. Donec cursus et
leo quis viverra.\r\nIn ac imperdiet ex, " +
+ "nec aliquet erat. Nullam sit amet enim in dolor finibus
convallis id eu nibh. Fusce aliquam convallis orci aliquam.";
+ private static final byte[] data = content.getBytes();
+
+ @Test
+ public void testNewLineDelimitedOldApiWithChar() throws Exception {
+ List<String> actualList = new ArrayList<>();
+
+ Socket socket = mock(Socket.class);
+ when(socket.getInputStream()).thenReturn(new
ByteArrayInputStream(data));
+ when(socket.isClosed()).thenReturn(false);
+ when(socket.isConnected()).thenReturn(true);
+
+ SocketTextStreamFunction source = new SocketTextStreamFunction("",
0, '\n', 0);
+ Field field =
SocketTextStreamFunction.class.getDeclaredField("isRunning");
+ field.setAccessible(true);
+ field.set(source, true);
--- End diff --
I do not think it is good to use reflection since I did something the same
as you before and they told don`t do this.
Why not set up a socket server in your test?
---
If your project is set up for it, you can reply to this email and have your
reply appear on GitHub as well. If your project does not have this feature
enabled and wishes so, or if the feature is enabled but not working, please
contact infrastructure at infrastruct...@apache.org or file a JIRA ticket
with INFRA.
---
