Test N/A values in all Progress parser implementations

Author: Euklios

src/test/java/net/bramp/ffmpeg/fixtures/Progresses.java

@@ -10,12 +10,20 @@ private Progresses() {
   }
 
   public static final ImmutableList<String> allFiles =
-      ImmutableList.of("ffmpeg-progress-0", "ffmpeg-progress-1", "ffmpeg-progress-2");
+      ImmutableList.of(
+              "ffmpeg-progress-0",
+              "ffmpeg-progress-1",
+              "ffmpeg-progress-2");
+
+  public static final ImmutableList<String> naProgressFile = ImmutableList.of("ffmpeg-progress-na");
 
   public static final ImmutableList<Progress> allProgresses =
       ImmutableList.of(
           new Progress(5, 0.0f, 800, 48, 512000000, 0, 0, 1.01f, Progress.Status.CONTINUE),
           new Progress(118, 23.4f, -1, -1, 5034667000L, 0, 0, -1, Progress.Status.CONTINUE),
-          new Progress(
-              132, 23.1f, 1935500, 1285168, 5312000000L, 0, 0, 0.929f, Progress.Status.END));
+          new Progress(132, 23.1f, 1935500, 1285168, 5312000000L, 0, 0, 0.929f, Progress.Status.END));
+
+  public static final ImmutableList<Progress> naProgresses = ImmutableList.of(
+          new Progress(0, 0.0f, -1, -1, -1, 0, 0, -1, Progress.Status.END)
+  );
 }

src/test/java/net/bramp/ffmpeg/progress/StreamProgressParserTest.java

@@ -6,7 +6,6 @@
 
 import java.io.IOException;
 import java.io.InputStream;
-import java.util.List;
 import net.bramp.ffmpeg.fixtures.Progresses;
 import org.junit.Test;
 
@@ -23,6 +22,18 @@ public void testNormal() throws IOException {
     InputStream inputStream = combineResource(Progresses.allFiles);
     parser.processStream(inputStream);
 
-    assertThat(listener.progesses, equalTo((List<Progress>) Progresses.allProgresses));
+    assertThat(listener.progesses, equalTo(Progresses.allProgresses));
+  }
+
+  @Test
+  public void testNaProgressPackets() throws IOException {
+    listener.reset();
+
+    StreamProgressParser parser = new StreamProgressParser(listener);
+
+    InputStream inputStream = combineResource(Progresses.naProgressFile);
+    parser.processStream(inputStream);
+
+    assertThat(listener.progesses, equalTo(Progresses.naProgresses));
   }
 }

src/test/java/net/bramp/ffmpeg/progress/TcpProgressParserTest.java

@@ -12,7 +12,6 @@
 import java.io.OutputStream;
 import java.net.Socket;
 import java.net.URISyntaxException;
-import java.util.List;
 import net.bramp.ffmpeg.fixtures.Progresses;
 import org.junit.Test;
 
@@ -44,12 +43,37 @@ public void testNormal() throws IOException, InterruptedException, URISyntaxExce
     parser.stop();
 
     assertThat(bytes, greaterThan(0L));
-    assertThat(progesses, equalTo((List<Progress>) Progresses.allProgresses));
+    assertThat(progesses, equalTo(Progresses.allProgresses));
+  }
+
+
+
+  @Test
+  public void testNaProgressPackets() throws IOException, InterruptedException, URISyntaxException {
+    parser.start();
+
+    Socket client = new Socket(uri.getHost(), uri.getPort());
+    assertTrue("Socket is connected", client.isConnected());
+
+    InputStream inputStream = combineResource(Progresses.naProgressFile);
+    OutputStream outputStream = client.getOutputStream();
+
+    long bytes = ByteStreams.copy(inputStream, outputStream);
+
+    // HACK, but give the TcpProgressParser thread time to actually handle the connection/data
+    // before the client is closed, and the parser is stopped.
+    Thread.sleep(100);
+
+    client.close();
+    parser.stop();
+
+    assertThat(bytes, greaterThan(0L));
+    assertThat(progesses, equalTo(Progresses.naProgresses));
   }
 
   @Test
   public void testPrematureDisconnect()
-      throws IOException, InterruptedException, URISyntaxException {
+      throws IOException {
     parser.start();
     new Socket(uri.getHost(), uri.getPort()).close();
     parser.stop();

src/test/java/net/bramp/ffmpeg/progress/UdpProgressParserTest.java

@@ -24,7 +24,7 @@ public ProgressParser newParser(ProgressListener listener)
   }
 
   @Test
-  public void testNormal() throws IOException, InterruptedException, URISyntaxException {
+  public void testNormal() throws IOException, InterruptedException {
     parser.start();
 
     final InetAddress addr = InetAddress.getByName(uri.getHost());
@@ -41,10 +41,35 @@ public void testNormal() throws IOException, InterruptedException, URISyntaxExce
       }
     }
 
+    Thread.sleep(1000); // HACK: Wait a short while to avoid closing the receiving socket
+
+    parser.stop();
+
+    assertThat(progesses, equalTo(Progresses.allProgresses));
+  }
+
+  @Test
+  public void testNaProgressPackets() throws IOException, InterruptedException, URISyntaxException {
+    parser.start();
+
+    final InetAddress addr = InetAddress.getByName(uri.getHost());
+    final int port = uri.getPort();
+
+    try (DatagramSocket socket = new DatagramSocket()) {
+      // Load each Progress Fixture, and send in a single datagram packet
+      for (String progressFixture : Progresses.naProgressFile) {
+        InputStream inputStream = loadResource(progressFixture);
+        byte[] bytes = ByteStreams.toByteArray(inputStream);
+
+        DatagramPacket packet = new DatagramPacket(bytes, bytes.length, addr, port);
+        socket.send(packet);
+      }
+    }
+
     Thread.sleep(100); // HACK: Wait a short while to avoid closing the receiving socket
 
     parser.stop();
 
-    assertThat(progesses, equalTo((List<Progress>) Progresses.allProgresses));
+    assertThat(progesses, equalTo(Progresses.naProgresses));
   }
 }

src/test/resources/net/bramp/ffmpeg/fixtures/ffmpeg-progress-na

@@ -0,0 +1,11 @@
+frame=0
+fps=0
+stream_0_0_q=-1.0
+bitrate=N/A
+total_size=N/A
+out_time_ms=N/A
+out_time=N/A
+dup_frames=0
+drop_frames=0
+speed=N/A
+progress=end