Home | History | Annotate | Download | only in webrtc 
      1 /*
      2  * libjingle
      3  * Copyright 2014 Google Inc.
      4  *
      5  * Redistribution and use in source and binary forms, with or without
      6  * modification, are permitted provided that the following conditions are met:
      7  *
      8  *  1. Redistributions of source code must retain the above copyright notice,
      9  *     this list of conditions and the following disclaimer.
     10  *  2. Redistributions in binary form must reproduce the above copyright notice,
     11  *     this list of conditions and the following disclaimer in the documentation
     12  *     and/or other materials provided with the distribution.
     13  *  3. The name of the author may not be used to endorse or promote products
     14  *     derived from this software without specific prior written permission.
     15  *
     16  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR IMPLIED
     17  * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
     18  * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO
     19  * EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
     20  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
     21  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS;
     22  * OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
     23  * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
     24  * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
     25  * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
     26  */
     27 
     28 #ifndef TALK_MEDIA_WEBRTC_WEBRTCVIDEOENGINE2_UNITTEST_H_
     29 #define TALK_MEDIA_WEBRTC_WEBRTCVIDEOENGINE2_UNITTEST_H_
     30 
     31 #include <map>
     32 #include <vector>
     33 
     34 #include "webrtc/call.h"
     35 #include "webrtc/video_receive_stream.h"
     36 #include "webrtc/video_send_stream.h"
     37 
     38 namespace cricket {
     39 class FakeVideoSendStream : public webrtc::VideoSendStream {
     40  public:
     41   FakeVideoSendStream(const webrtc::VideoSendStream::Config& config,
     42                       const std::vector<webrtc::VideoStream>& video_streams);
     43   webrtc::VideoSendStream::Config GetConfig();
     44   std::vector<webrtc::VideoStream> GetVideoStreams();
     45 
     46   bool IsSending();
     47 
     48  private:
     49   virtual webrtc::VideoSendStream::Stats GetStats() const OVERRIDE;
     50 
     51   virtual bool ReconfigureVideoEncoder(
     52       const std::vector<webrtc::VideoStream>& streams,
     53       const void* encoder_specific);
     54 
     55   virtual webrtc::VideoSendStreamInput* Input() OVERRIDE;
     56 
     57   virtual void Start() OVERRIDE;
     58   virtual void Stop() OVERRIDE;
     59 
     60   bool sending_;
     61   webrtc::VideoSendStream::Config config_;
     62   std::vector<webrtc::VideoStream> video_streams_;
     63 };
     64 
     65 class FakeVideoReceiveStream : public webrtc::VideoReceiveStream {
     66  public:
     67   explicit FakeVideoReceiveStream(
     68       const webrtc::VideoReceiveStream::Config& config);
     69 
     70   webrtc::VideoReceiveStream::Config GetConfig();
     71 
     72  private:
     73   virtual webrtc::VideoReceiveStream::Stats GetStats() const OVERRIDE;
     74 
     75   virtual void Start() OVERRIDE;
     76   virtual void Stop() OVERRIDE;
     77   virtual void GetCurrentReceiveCodec(webrtc::VideoCodec* codec);
     78 
     79   webrtc::VideoReceiveStream::Config config_;
     80   bool receiving_;
     81 };
     82 
     83 class FakeCall : public webrtc::Call {
     84  public:
     85   FakeCall();
     86   ~FakeCall();
     87 
     88   void SetVideoCodecs(const std::vector<webrtc::VideoCodec> codecs);
     89 
     90   std::vector<FakeVideoSendStream*> GetVideoSendStreams();
     91   std::vector<FakeVideoReceiveStream*> GetVideoReceiveStreams();
     92 
     93   webrtc::VideoCodec GetEmptyVideoCodec();
     94 
     95   webrtc::VideoCodec GetVideoCodecVp8();
     96   webrtc::VideoCodec GetVideoCodecVp9();
     97 
     98   std::vector<webrtc::VideoCodec> GetDefaultVideoCodecs();
     99 
    100  private:
    101   virtual webrtc::VideoSendStream::Config GetDefaultSendConfig() OVERRIDE;
    102 
    103   virtual webrtc::VideoSendStream* CreateVideoSendStream(
    104       const webrtc::VideoSendStream::Config& config,
    105       const std::vector<webrtc::VideoStream>& video_streams,
    106       const void* encoder_settings) OVERRIDE;
    107 
    108   virtual void DestroyVideoSendStream(
    109       webrtc::VideoSendStream* send_stream) OVERRIDE;
    110 
    111   virtual webrtc::VideoReceiveStream::Config GetDefaultReceiveConfig() OVERRIDE;
    112 
    113   virtual webrtc::VideoReceiveStream* CreateVideoReceiveStream(
    114       const webrtc::VideoReceiveStream::Config& config) OVERRIDE;
    115 
    116   virtual void DestroyVideoReceiveStream(
    117       webrtc::VideoReceiveStream* receive_stream) OVERRIDE;
    118   virtual webrtc::PacketReceiver* Receiver() OVERRIDE;
    119 
    120   virtual uint32_t SendBitrateEstimate() OVERRIDE;
    121   virtual uint32_t ReceiveBitrateEstimate() OVERRIDE;
    122 
    123   std::vector<webrtc::VideoCodec> codecs_;
    124   std::vector<FakeVideoSendStream*> video_send_streams_;
    125   std::vector<FakeVideoReceiveStream*> video_receive_streams_;
    126 };
    127 
    128 class FakeWebRtcVideoChannel2 : public WebRtcVideoChannel2 {
    129  public:
    130   FakeWebRtcVideoChannel2(FakeCall* call,
    131                           WebRtcVideoEngine2* engine,
    132                           VoiceMediaChannel* voice_channel);
    133   virtual ~FakeWebRtcVideoChannel2();
    134 
    135   VoiceMediaChannel* GetVoiceChannel();
    136   FakeCall* GetFakeCall();
    137 
    138  private:
    139   FakeCall* fake_call_;
    140   VoiceMediaChannel* voice_channel_;
    141 };
    142 
    143 class FakeWebRtcVideoMediaChannelFactory : public WebRtcVideoChannelFactory {
    144  public:
    145   FakeWebRtcVideoChannel2* GetFakeChannel(VideoMediaChannel* channel);
    146 
    147  private:
    148   virtual WebRtcVideoChannel2* Create(
    149       WebRtcVideoEngine2* engine,
    150       VoiceMediaChannel* voice_channel) OVERRIDE;
    151 
    152   std::map<VideoMediaChannel*, FakeWebRtcVideoChannel2*> channel_map_;
    153 };
    154 
    155 
    156 }  // namespace cricket
    157 #endif  // TALK_MEDIA_WEBRTC_WEBRTCVIDEOENGINE2_UNITTEST_H_
    158