src/content/browser/media/webrtc_internals_unittest.cc

   1 // Copyright (c) 2013 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "base/memory/scoped_ptr.h"
   6 #include "base/message_loop/message_loop.h"
   7 #include "base/values.h"
   8 #include "content/browser/media/webrtc_internals.h"
   9 #include "content/browser/media/webrtc_internals_ui_observer.h"
  10 #include "content/public/test/test_browser_thread.h"
  11 #include "testing/gtest/include/gtest/gtest.h"
  12
  13 namespace content {
  14
  15 namespace {
  16
  17 static const std::string kContraints = "c";
  18 static const std::string kRtcConfiguration = "r";
  19 static const std::string kUrl = "u";
  20
  21 class MockWebRTCInternalsProxy : public WebRTCInternalsUIObserver {
  22  public:
  23   virtual void OnUpdate(const std::string& command,
  24                         const base::Value* value) OVERRIDE {
  25     command_ = command;
  26     if (value)
  27       value_.reset(value->DeepCopy());
  28   }
  29
  30   std::string command() {
  31     return command_;
  32   }
  33
  34   base::Value* value() {
  35     return value_.get();
  36   }
  37
  38  private:
  39    std::string command_;
  40    scoped_ptr<base::Value> value_;
  41 };
  42
  43 class WebRTCInternalsTest : public testing::Test {
  44  public:
  45   WebRTCInternalsTest() : io_thread_(BrowserThread::UI, &io_loop_) {
  46     WebRTCInternals::GetInstance()->ResetForTesting();
  47   }
  48
  49  protected:
  50   void VerifyString(const base::DictionaryValue* dict,
  51                     const std::string& key,
  52                     const std::string& expected) {
  53     std::string actual;
  54     EXPECT_TRUE(dict->GetString(key, &actual));
  55     EXPECT_EQ(expected, actual);
  56   }
  57
  58   void VerifyInt(const base::DictionaryValue* dict,
  59                  const std::string& key,
  60                  int expected) {
  61     int actual;
  62     EXPECT_TRUE(dict->GetInteger(key, &actual));
  63     EXPECT_EQ(expected, actual);
  64   }
  65
  66   void VerifyList(const base::DictionaryValue* dict,
  67                   const std::string& key,
  68                   const base::ListValue& expected) {
  69     const base::ListValue* actual = NULL;
  70     EXPECT_TRUE(dict->GetList(key, &actual));
  71     EXPECT_TRUE(expected.Equals(actual));
  72   }
  73
  74   void VerifyGetUserMediaData(base::Value* actual_data,
  75                               int rid,
  76                               int pid,
  77                               const std::string& origin,
  78                               const std::string& audio,
  79                               const std::string& video) {
  80     base::DictionaryValue* dict = NULL;
  81     EXPECT_TRUE(actual_data->GetAsDictionary(&dict));
  82
  83     VerifyInt(dict, "rid", rid);
  84     VerifyInt(dict, "pid", pid);
  85     VerifyString(dict, "origin", origin);
  86     VerifyString(dict, "audio", audio);
  87     VerifyString(dict, "video", video);
  88   }
  89
  90   base::MessageLoop io_loop_;
  91   TestBrowserThread io_thread_;
  92 };
  93
  94 }  // namespace
  95
  96 TEST_F(WebRTCInternalsTest, AddRemoveObserver) {
  97   scoped_ptr<MockWebRTCInternalsProxy> observer(
  98       new MockWebRTCInternalsProxy());
  99   WebRTCInternals::GetInstance()->AddObserver(observer.get());
 100   WebRTCInternals::GetInstance()->RemoveObserver(observer.get());
 101   WebRTCInternals::GetInstance()->OnAddPeerConnection(
 102       0, 3, 4, kUrl, kRtcConfiguration, kContraints);
 103   EXPECT_EQ("", observer->command());
 104
 105   WebRTCInternals::GetInstance()->OnRemovePeerConnection(3, 4);
 106 }
 107
 108 TEST_F(WebRTCInternalsTest, SendAddPeerConnectionUpdate) {
 109   scoped_ptr<MockWebRTCInternalsProxy> observer(
 110       new MockWebRTCInternalsProxy());
 111   WebRTCInternals::GetInstance()->AddObserver(observer.get());
 112   WebRTCInternals::GetInstance()->OnAddPeerConnection(
 113       0, 1, 2, kUrl, kRtcConfiguration, kContraints);
 114   EXPECT_EQ("addPeerConnection", observer->command());
 115
 116   base::DictionaryValue* dict = NULL;
 117   EXPECT_TRUE(observer->value()->GetAsDictionary(&dict));
 118
 119   VerifyInt(dict, "pid", 1);
 120   VerifyInt(dict, "lid", 2);
 121   VerifyString(dict, "url", kUrl);
 122   VerifyString(dict, "rtcConfiguration", kRtcConfiguration);
 123   VerifyString(dict, "constraints", kContraints);
 124
 125   WebRTCInternals::GetInstance()->RemoveObserver(observer.get());
 126   WebRTCInternals::GetInstance()->OnRemovePeerConnection(1, 2);
 127 }
 128
 129 TEST_F(WebRTCInternalsTest, SendRemovePeerConnectionUpdate) {
 130   scoped_ptr<MockWebRTCInternalsProxy> observer(
 131       new MockWebRTCInternalsProxy());
 132   WebRTCInternals::GetInstance()->AddObserver(observer.get());
 133   WebRTCInternals::GetInstance()->OnAddPeerConnection(
 134       0, 1, 2, kUrl, kRtcConfiguration, kContraints);
 135   WebRTCInternals::GetInstance()->OnRemovePeerConnection(1, 2);
 136   EXPECT_EQ("removePeerConnection", observer->command());
 137
 138   base::DictionaryValue* dict = NULL;
 139   EXPECT_TRUE(observer->value()->GetAsDictionary(&dict));
 140
 141   VerifyInt(dict, "pid", 1);
 142   VerifyInt(dict, "lid", 2);
 143
 144   WebRTCInternals::GetInstance()->RemoveObserver(observer.get());
 145 }
 146
 147 TEST_F(WebRTCInternalsTest, SendUpdatePeerConnectionUpdate) {
 148   scoped_ptr<MockWebRTCInternalsProxy> observer(
 149       new MockWebRTCInternalsProxy());
 150   WebRTCInternals::GetInstance()->AddObserver(observer.get());
 151   WebRTCInternals::GetInstance()->OnAddPeerConnection(
 152       0, 1, 2, kUrl, kRtcConfiguration, kContraints);
 153
 154   const std::string update_type = "fakeType";
 155   const std::string update_value = "fakeValue";
 156   WebRTCInternals::GetInstance()->OnUpdatePeerConnection(
 157       1, 2, update_type, update_value);
 158
 159   EXPECT_EQ("updatePeerConnection", observer->command());
 160
 161   base::DictionaryValue* dict = NULL;
 162   EXPECT_TRUE(observer->value()->GetAsDictionary(&dict));
 163
 164   VerifyInt(dict, "pid", 1);
 165   VerifyInt(dict, "lid", 2);
 166   VerifyString(dict, "type", update_type);
 167   VerifyString(dict, "value", update_value);
 168
 169   std::string time;
 170   EXPECT_TRUE(dict->GetString("time", &time));
 171   EXPECT_FALSE(time.empty());
 172
 173   WebRTCInternals::GetInstance()->OnRemovePeerConnection(1, 2);
 174   WebRTCInternals::GetInstance()->RemoveObserver(observer.get());
 175 }
 176
 177 TEST_F(WebRTCInternalsTest, AddGetUserMedia) {
 178   scoped_ptr<MockWebRTCInternalsProxy> observer(new MockWebRTCInternalsProxy());
 179
 180   // Add one observer before "getUserMedia".
 181   WebRTCInternals::GetInstance()->AddObserver(observer.get());
 182
 183   const int rid = 1;
 184   const int pid = 2;
 185   const std::string audio_constraint = "aaa";
 186   const std::string video_constraint = "vvv";
 187   WebRTCInternals::GetInstance()->OnGetUserMedia(
 188       rid, pid, kUrl, true, true, audio_constraint, video_constraint);
 189
 190   EXPECT_EQ("addGetUserMedia", observer->command());
 191   VerifyGetUserMediaData(
 192       observer->value(), rid, pid, kUrl, audio_constraint, video_constraint);
 193
 194   WebRTCInternals::GetInstance()->RemoveObserver(observer.get());
 195 }
 196
 197 TEST_F(WebRTCInternalsTest, SendAllUpdateWithGetUserMedia) {
 198   const int rid = 1;
 199   const int pid = 2;
 200   const std::string audio_constraint = "aaa";
 201   const std::string video_constraint = "vvv";
 202   WebRTCInternals::GetInstance()->OnGetUserMedia(
 203       rid, pid, kUrl, true, true, audio_constraint, video_constraint);
 204
 205   scoped_ptr<MockWebRTCInternalsProxy> observer(new MockWebRTCInternalsProxy());
 206   // Add one observer after "getUserMedia".
 207   WebRTCInternals::GetInstance()->AddObserver(observer.get());
 208   WebRTCInternals::GetInstance()->UpdateObserver(observer.get());
 209
 210   EXPECT_EQ("addGetUserMedia", observer->command());
 211   VerifyGetUserMediaData(
 212       observer->value(), rid, pid, kUrl, audio_constraint, video_constraint);
 213
 214   WebRTCInternals::GetInstance()->RemoveObserver(observer.get());
 215 }
 216
 217 TEST_F(WebRTCInternalsTest, SendAllUpdatesWithPeerConnectionUpdate) {
 218   const int rid = 0, pid = 1, lid = 2;
 219   const std::string update_type = "fakeType";
 220   const std::string update_value = "fakeValue";
 221
 222   WebRTCInternals::GetInstance()->OnAddPeerConnection(
 223       rid, pid, lid, kUrl, kRtcConfiguration, kContraints);
 224   WebRTCInternals::GetInstance()->OnUpdatePeerConnection(
 225       pid, lid, update_type, update_value);
 226
 227   scoped_ptr<MockWebRTCInternalsProxy> observer(new MockWebRTCInternalsProxy());
 228   WebRTCInternals::GetInstance()->AddObserver(observer.get());
 229
 230   WebRTCInternals::GetInstance()->UpdateObserver(observer.get());
 231
 232   EXPECT_EQ("updateAllPeerConnections", observer->command());
 233
 234   base::ListValue* list = NULL;
 235   EXPECT_TRUE(observer->value()->GetAsList(&list));
 236   EXPECT_EQ(1U, list->GetSize());
 237
 238   base::DictionaryValue* dict = NULL;
 239   EXPECT_TRUE((*list->begin())->GetAsDictionary(&dict));
 240
 241   VerifyInt(dict, "rid", rid);
 242   VerifyInt(dict, "pid", pid);
 243   VerifyInt(dict, "lid", lid);
 244   VerifyString(dict, "url", kUrl);
 245   VerifyString(dict, "rtcConfiguration", kRtcConfiguration);
 246   VerifyString(dict, "constraints", kContraints);
 247
 248   base::ListValue* log = NULL;
 249   EXPECT_TRUE(dict->GetList("log", &log));
 250   EXPECT_EQ(1U, log->GetSize());
 251
 252   EXPECT_TRUE((*log->begin())->GetAsDictionary(&dict));
 253   VerifyString(dict, "type", update_type);
 254   VerifyString(dict, "value", update_value);
 255   std::string time;
 256   EXPECT_TRUE(dict->GetString("time", &time));
 257   EXPECT_FALSE(time.empty());
 258 }
 259
 260 TEST_F(WebRTCInternalsTest, OnAddStats) {
 261   const int rid = 0, pid = 1, lid = 2;
 262   scoped_ptr<MockWebRTCInternalsProxy> observer(new MockWebRTCInternalsProxy());
 263   WebRTCInternals::GetInstance()->AddObserver(observer.get());
 264   WebRTCInternals::GetInstance()->OnAddPeerConnection(
 265       rid, pid, lid, kUrl, kRtcConfiguration, kContraints);
 266
 267   base::ListValue list;
 268   list.AppendString("xxx");
 269   list.AppendString("yyy");
 270   WebRTCInternals::GetInstance()->OnAddStats(pid, lid, list);
 271
 272   EXPECT_EQ("addStats", observer->command());
 273   base::DictionaryValue* dict = NULL;
 274   EXPECT_TRUE(observer->value()->GetAsDictionary(&dict));
 275
 276   VerifyInt(dict, "pid", pid);
 277   VerifyInt(dict, "lid", lid);
 278   VerifyList(dict, "reports", list);
 279 }
 280
 281 TEST_F(WebRTCInternalsTest, AecRecordingFileSelectionCanceled) {
 282   scoped_ptr<MockWebRTCInternalsProxy> observer(new MockWebRTCInternalsProxy());
 283   WebRTCInternals::GetInstance()->AddObserver(observer.get());
 284   WebRTCInternals::GetInstance()->FileSelectionCanceled(NULL);
 285   EXPECT_EQ("aecRecordingFileSelectionCancelled", observer->command());
 286   EXPECT_EQ(NULL, observer->value());
 287 }
 288
 289 }  // namespace content