Testing of FileVideoCapturer

webrtc/examples/androidtests/src/org/appspot/apprtc/test/FileVideoCapturerTest.java

+/*
+ *  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.appspot.apprtc.test;
+
+import android.test.InstrumentationTestCase;
+import android.test.suitebuilder.annotation.LargeTest;
+import android.test.suitebuilder.annotation.MediumTest;
+import android.test.suitebuilder.annotation.SmallTest;
+
+import org.webrtc.FileVideoCapturer;
+import org.webrtc.VideoCapturer;
+
+import java.io.IOException;
+import java.lang.Thread;
+import java.util.ArrayList;
+
+public class FileVideoCapturerTest extends InstrumentationTestCase {
+  private static class Frame {
+    public byte[] data;
+    public int width;
+    public int height;
+  }
+
+  public class MockCapturerObserver implements VideoCapturer.CapturerObserver {
+    private final ArrayList<Frame> frameDatas = new ArrayList<Frame>();
+
+    @Override
+    public void onCapturerStarted(boolean success) {
+      assertTrue(success);
+    }
+
+    @Override
+    public void onCapturerStopped() {
+      // Empty on purpose.
+    }
+
+    @Override
+    public synchronized void onByteBufferFrameCaptured(
+        byte[] data, int width, int height, int rotation, long timeStamp) {
+      Frame frame = new Frame();
+      frame.data = data;
+      frame.width = width;
+      frame.height = height;
+      assertTrue(data.length != 0);
+      frameDatas.add(frame);
+      notify();
+    }
+
+    @Override
+    public void onTextureFrameCaptured(int width, int height, int oesTextureId,
+        float[] transformMatrix, int rotation, long timestamp) {
+      // Empty on purpose.
+    }
+
+    public synchronized ArrayList<Frame> getMinimumFramesBlocking(int minFrames)
+        throws InterruptedException {
+      while (frameDatas.size() < minFrames) {
+        wait();
+      }
+      return new ArrayList<Frame>(frameDatas);
+    }
+  }
+
+  @SmallTest
+  public void testVideoCaptureFromFile() throws InterruptedException, IOException {
+    final int FRAME_WIDTH = 4;
+    final int FRAME_HEIGHT = 4;
+    final FileVideoCapturer fileVideoCapturer = new FileVideoCapturer(
+        "/sdcard/chromium_tests_root/webrtc/examples/androidtests/src/org/appspot/apprtc/test/capturetestvideo.y4m");
+    final MockCapturerObserver capturerObserver = new MockCapturerObserver();
+    fileVideoCapturer.initialize(null, null, capturerObserver);
+    fileVideoCapturer.startCapture(FRAME_WIDTH, FRAME_HEIGHT, 33);
+
+    final String[] expectedFrames = {"54484953204953204A55535420534F4D5845542020547845",
+        "544845205345434F4E44204652414D457220747179772E65",
+        "4845524520495320544845205448524941444D2045462152"};
+
+    final ArrayList<Frame> frameDatas;
+    frameDatas = capturerObserver.getMinimumFramesBlocking(expectedFrames.length);
+
+    assertTrue(expectedFrames.length == frameDatas.size());

[magjed_webrtc, 2016/10/26 11:20:00]

nit: Use assertEquals instead

+
+    fileVideoCapturer.stopCapture();
+    fileVideoCapturer.dispose();
+
+    for (int i = 0; i < expectedFrames.length; ++i) {
+      Frame frame = frameDatas.get(i);
+
+      assertEquals(FRAME_WIDTH, frame.width);
+      assertEquals(FRAME_HEIGHT, frame.height);
+      assertEquals(FRAME_WIDTH * FRAME_HEIGHT * 3 / 2, frame.data.length);
+
+      StringBuilder builder = new StringBuilder();
+      for (byte b : frame.data) {
+        builder.append(String.format("%02X", b));
+      }
+
+      assertEquals(expectedFrames[i], builder.toString());
+    }
+  }
+}