From a8bec8d8e74e95998d3a18d7b7fed41b4c570e37 Mon Sep 17 00:00:00 2001
From: mandermo <mandermo@webrtc.org>
Date: Wed, 26 Oct 2016 01:47:07 -0700
Subject: [PATCH] Testing of VideoFileRenderer with byte frames

BUG=webrtc:6545

Review-Url: https://codereview.webrtc.org/2415563002
Cr-Commit-Position: refs/heads/master@{#14781}
---
 webrtc/examples/BUILD.gn                      |  5 +-
 .../apprtc/test/VideoFileRendererTest.java    | 78 +++++++++++++++++++
 2 files changed, 82 insertions(+), 1 deletion(-)
 create mode 100644 webrtc/examples/androidtests/src/org/appspot/apprtc/test/VideoFileRendererTest.java

diff --git a/webrtc/examples/BUILD.gn b/webrtc/examples/BUILD.gn
index b0b530fdd0..d78c28774a 100644
--- a/webrtc/examples/BUILD.gn
+++ b/webrtc/examples/BUILD.gn
@@ -93,7 +93,10 @@ if (is_android) {
     apk_name = "AppRTCMobileTest"
     android_manifest = "androidtests/AndroidManifest.xml"
 
-    java_files = [ "androidtests/src/org/appspot/apprtc/test/PeerConnectionClientTest.java" ]
+    java_files = [
+      "androidtests/src/org/appspot/apprtc/test/PeerConnectionClientTest.java",
+      "androidtests/src/org/appspot/apprtc/test/VideoFileRendererTest.java",
+    ]
 
     apk_under_test = ":AppRTCMobile"
 
diff --git a/webrtc/examples/androidtests/src/org/appspot/apprtc/test/VideoFileRendererTest.java b/webrtc/examples/androidtests/src/org/appspot/apprtc/test/VideoFileRendererTest.java
new file mode 100644
index 0000000000..5cf3e9f6f0
--- /dev/null
+++ b/webrtc/examples/androidtests/src/org/appspot/apprtc/test/VideoFileRendererTest.java
@@ -0,0 +1,78 @@
+/*
+ *  Copyright 2016 The WebRTC Project Authors. All rights reserved.
+ *
+ *  Use of this source code is governed by a BSD-style license
+ *  that can be found in the LICENSE file in the root of the source
+ *  tree. An additional intellectual property rights grant can be found
+ *  in the file PATENTS.  All contributing project authors may
+ *  be found in the AUTHORS file in the root of the source tree.
+ */
+
+package org.webrtc;
+
+import android.test.InstrumentationTestCase;
+import android.test.suitebuilder.annotation.SmallTest;
+
+import java.io.File;
+import java.io.IOException;
+import java.io.RandomAccessFile;
+import java.lang.Thread;
+import java.nio.charset.StandardCharsets;
+import java.nio.ByteBuffer;
+import java.util.ArrayList;
+import java.util.Random;
+
+public class VideoFileRendererTest extends InstrumentationTestCase {
+  @SmallTest
+  public void testYuvRenderingToFile() throws InterruptedException, IOException {
+    EglBase eglBase = EglBase.create();
+    final String videoOutPath = "/sdcard/chromium_tests_root/testvideoout.y4m";
+    int frameWidth = 4;
+    int frameHeight = 4;
+    VideoFileRenderer videoFileRenderer =
+        new VideoFileRenderer(videoOutPath, frameWidth, frameHeight, eglBase.getEglBaseContext());
+
+    String[] frames = {
+        "THIS IS JUST SOME TEXT x", "THE SECOND FRAME qwerty.", "HERE IS THE THRID FRAME!"};
+
+    for (String frameStr : frames) {
+      int[] planeSizes = {
+          frameWidth * frameWidth, frameWidth * frameHeight / 4, frameWidth * frameHeight / 4};
+
+      byte[] frameBytes = frameStr.getBytes(StandardCharsets.US_ASCII);
+      ByteBuffer[] yuvPlanes = new ByteBuffer[3];
+      int pos = 0;
+      for (int i = 0; i < 3; i++) {
+        yuvPlanes[i] = ByteBuffer.allocateDirect(planeSizes[i]);
+        yuvPlanes[i].put(frameBytes, pos, planeSizes[i]);
+        pos += planeSizes[i];
+      }
+
+      int[] yuvStrides = {frameWidth, frameWidth / 2, frameWidth / 2};
+
+      VideoRenderer.I420Frame frame =
+          new VideoRenderer.I420Frame(frameWidth, frameHeight, 0, yuvStrides, yuvPlanes, 0);
+
+      videoFileRenderer.renderFrame(frame);
+    }
+    videoFileRenderer.release();
+
+    RandomAccessFile writtenFile = new RandomAccessFile(videoOutPath, "r");
+    try {
+      int length = (int) writtenFile.length();
+      byte[] data = new byte[length];
+      writtenFile.readFully(data);
+      String fileContent = new String(data, StandardCharsets.US_ASCII);
+      String expected = "YUV4MPEG2 C420 W4 H4 Ip F30:1 A1:1\n"
+          + "FRAME\n"
+          + "THIS IS JUST SOME TEXT xFRAME\n"
+          + "THE SECOND FRAME qwerty.FRAME\n"
+          + "HERE IS THE THRID FRAME!";
+      assertEquals(expected, fileContent);
+    } finally {
+      writtenFile.close();
+    }
+
+    new File(videoOutPath).delete();
+  }
+}