Reland of Injectable Obj-C video codecs

webrtc/sdk/objc/Framework/UnitTests/objc_video_encoder_factory_tests.mm

Index: webrtc/sdk/objc/Framework/UnitTests/objc_video_encoder_factory_tests.mm
diff --git a/webrtc/sdk/objc/Framework/UnitTests/objc_video_encoder_factory_tests.mm b/webrtc/sdk/objc/Framework/UnitTests/objc_video_encoder_factory_tests.mm
new file mode 100644
index 0000000000000000000000000000000000000000..11fa3fce9df351c7a9f0bc2b1d40382000de9c13
--- /dev/null
+++ b/webrtc/sdk/objc/Framework/UnitTests/objc_video_encoder_factory_tests.mm
@@ -0,0 +1,134 @@
+/*
+ *  Copyright 2017 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.
+ */
+
+#import <Foundation/Foundation.h>
+#import <OCMock/OCMock.h>
+
+#include "webrtc/sdk/objc/Framework/Classes/VideoToolbox/objc_video_encoder_factory.h"
+
+#import "WebRTC/RTCVideoCodec.h"
+#import "WebRTC/RTCVideoCodecFactory.h"
+#import "WebRTC/RTCVideoFrameBuffer.h"
+#include "webrtc/modules/include/module_common_types.h"
+#include "webrtc/modules/video_coding/include/video_codec_interface.h"
+#include "webrtc/modules/video_coding/include/video_error_codes.h"
+#include "webrtc/rtc_base/gunit.h"
+#include "webrtc/sdk/objc/Framework/Classes/Video/objc_frame_buffer.h"
+
+id<RTCVideoEncoderFactory> CreateEncoderFactoryReturning(int return_code) {
+  id encoderMock = OCMProtocolMock(@protocol(RTCVideoEncoder));
+  OCMStub([encoderMock startEncodeWithSettings:[OCMArg any] numberOfCores:1])
+      .andReturn(return_code);
+  OCMStub([encoderMock encode:[OCMArg any] codecSpecificInfo:[OCMArg any] frameTypes:[OCMArg any]])
+      .andReturn(return_code);
+  OCMStub([encoderMock releaseEncoder]).andReturn(return_code);
+  OCMStub([encoderMock setBitrate:0 framerate:0]).andReturn(return_code == WEBRTC_VIDEO_CODEC_OK);
+
+  id encoderFactoryMock = OCMProtocolMock(@protocol(RTCVideoEncoderFactory));
+  RTCVideoCodecInfo *supported =
+      [[RTCVideoCodecInfo alloc] initWithPayload:0 name:@"H264" parameters:@{}];
+  OCMStub([encoderFactoryMock supportedCodecs]).andReturn(@[ supported ]);
+  OCMStub([encoderFactoryMock createEncoder:[OCMArg any]]).andReturn(encoderMock);
+  return encoderFactoryMock;
+}
+
+id<RTCVideoEncoderFactory> CreateOKEncoderFactory() {
+  return CreateEncoderFactoryReturning(WEBRTC_VIDEO_CODEC_OK);
+}
+
+id<RTCVideoEncoderFactory> CreateErrorEncoderFactory() {
+  return CreateEncoderFactoryReturning(WEBRTC_VIDEO_CODEC_ERROR);
+}
+
+webrtc::VideoEncoder *GetObjCEncoder(id<RTCVideoEncoderFactory> factory) {
+  webrtc::ObjCVideoEncoderFactory encoder_factory(factory);
+  cricket::VideoCodec codec("H264");
+  return encoder_factory.CreateVideoEncoder(codec);
+}
+
+#pragma mark -
+
+TEST(ObjCVideoEncoderFactoryTest, InitEncodeReturnsOKOnSuccess) {
+  webrtc::VideoEncoder *encoder = GetObjCEncoder(CreateOKEncoderFactory());
+
+  auto settings = new webrtc::VideoCodec();
+  EXPECT_EQ(encoder->InitEncode(settings, 1, 0), WEBRTC_VIDEO_CODEC_OK);
+}
+
+TEST(ObjCVideoEncoderFactoryTest, InitEncodeReturnsErrorOnFail) {
+  webrtc::VideoEncoder *encoder = GetObjCEncoder(CreateErrorEncoderFactory());
+
+  auto settings = new webrtc::VideoCodec();
+  EXPECT_EQ(encoder->InitEncode(settings, 1, 0), WEBRTC_VIDEO_CODEC_ERROR);
+}
+
+TEST(ObjCVideoEncoderFactoryTest, EncodeReturnsOKOnSuccess) {
+  webrtc::VideoEncoder *encoder = GetObjCEncoder(CreateOKEncoderFactory());
+
+  CVPixelBufferRef pixel_buffer;
+  CVPixelBufferCreate(kCFAllocatorDefault, 640, 480, kCVPixelFormatType_32ARGB, nil, &pixel_buffer);
+  rtc::scoped_refptr<webrtc::VideoFrameBuffer> buffer =
+      new rtc::RefCountedObject<webrtc::ObjCFrameBuffer>(
+          [[RTCCVPixelBuffer alloc] initWithPixelBuffer:pixel_buffer]);
+  webrtc::VideoFrame frame(buffer, webrtc::kVideoRotation_0, 0);
+  webrtc::CodecSpecificInfo info;
+  info.codecType = webrtc::kVideoCodecH264;
+  info.codec_name = "H264";
+  std::vector<webrtc::FrameType> frame_types;
+
+  EXPECT_EQ(encoder->Encode(frame, &info, &frame_types), WEBRTC_VIDEO_CODEC_OK);
+}
+
+TEST(ObjCVideoEncoderFactoryTest, EncodeReturnsErrorOnFail) {
+  webrtc::VideoEncoder *encoder = GetObjCEncoder(CreateErrorEncoderFactory());
+
+  CVPixelBufferRef pixel_buffer;
+  CVPixelBufferCreate(kCFAllocatorDefault, 640, 480, kCVPixelFormatType_32ARGB, nil, &pixel_buffer);
+  rtc::scoped_refptr<webrtc::VideoFrameBuffer> buffer =
+      new rtc::RefCountedObject<webrtc::ObjCFrameBuffer>(
+          [[RTCCVPixelBuffer alloc] initWithPixelBuffer:pixel_buffer]);
+  webrtc::VideoFrame frame(buffer, webrtc::kVideoRotation_0, 0);
+  webrtc::CodecSpecificInfo info;
+  info.codecType = webrtc::kVideoCodecH264;
+  info.codec_name = "H264";
+  std::vector<webrtc::FrameType> frame_types;
+
+  EXPECT_EQ(encoder->Encode(frame, &info, &frame_types), WEBRTC_VIDEO_CODEC_ERROR);
+}
+
+TEST(ObjCVideoEncoderFactoryTest, ReleaseEncodeReturnsOKOnSuccess) {
+  webrtc::VideoEncoder *encoder = GetObjCEncoder(CreateOKEncoderFactory());
+
+  EXPECT_EQ(encoder->Release(), WEBRTC_VIDEO_CODEC_OK);
+}
+
+TEST(ObjCVideoEncoderFactoryTest, ReleaseEncodeReturnsErrorOnFail) {
+  webrtc::VideoEncoder *encoder = GetObjCEncoder(CreateErrorEncoderFactory());
+
+  EXPECT_EQ(encoder->Release(), WEBRTC_VIDEO_CODEC_ERROR);
+}
+
+TEST(ObjCVideoEncoderFactoryTest, SetChannelParametersAlwaysReturnsOK) {
+  webrtc::VideoEncoder *encoder = GetObjCEncoder(CreateErrorEncoderFactory());
+
+  EXPECT_EQ(encoder->SetChannelParameters(1, 1), WEBRTC_VIDEO_CODEC_OK);
+}
+
+TEST(ObjCVideoEncoderFactoryTest, SetRatesReturnsOKOnSuccess) {
+  webrtc::VideoEncoder *encoder = GetObjCEncoder(CreateOKEncoderFactory());
+
+  EXPECT_EQ(encoder->SetRates(0, 0), WEBRTC_VIDEO_CODEC_OK);
+}
+
+TEST(ObjCVideoEncoderFactoryTest, SetRatesReturnsErrorOnFail) {
+  webrtc::VideoEncoder *encoder = GetObjCEncoder(CreateErrorEncoderFactory());
+
+  EXPECT_EQ(encoder->SetRates(0, 0), WEBRTC_VIDEO_CODEC_ERROR);
+}