Introduced base class for WebRTC browser tests, cleaned up duplication.

chrome/browser/media/webrtc_browsertest_base.cc

Index: chrome/browser/media/webrtc_browsertest_base.cc
diff --git a/chrome/browser/media/webrtc_browsertest_base.cc b/chrome/browser/media/webrtc_browsertest_base.cc
new file mode 100644
index 0000000000000000000000000000000000000000..8028fc5decd4b11f7d9f071c003e204bc89e251f
--- /dev/null
+++ b/chrome/browser/media/webrtc_browsertest_base.cc
@@ -0,0 +1,129 @@
+// Copyright 2013 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#include "webrtc_browsertest_base.h"

[tommi (sloooow) - chröme, 2013/07/25 14:09:09]

full path

[phoglund_chromium, 2013/07/25 15:22:46]

Done.

[phoglund_chromium, 2013/07/25 15:22:46]

Done.

+
+#include "base/strings/stringprintf.h"
+#include "chrome/browser/chrome_notification_types.h"
+#include "chrome/browser/infobars/infobar.h"
+#include "chrome/browser/infobars/infobar_service.h"
+#include "chrome/browser/media/media_stream_infobar_delegate.h"
+#include "chrome/browser/media/webrtc_browsertest_common.h"
+#include "chrome/browser/ui/browser.h"
+#include "chrome/test/base/ui_test_utils.h"
+#include "content/public/browser/notification_service.h"
+#include "content/public/test/browser_test_utils.h"
+
+const char kAudioVideoCallConstraints[] = "'{audio: true, video: true}'";
+const char kAudioOnlyCallConstraints[] = "'{audio: true}'";
+const char kVideoOnlyCallConstraints[] = "'{video: true}'";
+const char kFailedWithErrorPermissionDenied[] =
+    "failed-with-error-PERMISSION_DENIED";
+const char kOkGotStream[] = "ok-got-stream";
+
+// Convenience method which executes the provided javascript in the context
+// of the provided web contents and returns what it evaluated to.
+static std::string ExecuteJavascript(const std::string& javascript,
+                                     content::WebContents* tab_contents) {
+  std::string result;
+  EXPECT_TRUE(content::ExecuteScriptAndExtractString(
+      tab_contents, javascript, &result));
+  return result;
+}
+
+MediaStreamInfoBarDelegate* WebrtcTestBase::GetUserMediaAndWaitForInfobar(
+    content::WebContents* tab_contents,
+    const std::string& constraints) {
+  content::WindowedNotificationObserver infobar_added(
+      chrome::NOTIFICATION_TAB_CONTENTS_INFOBAR_ADDED,
+      content::NotificationService::AllSources());
+
+  // Request user media: this will launch the media stream info bar.
+  GetUserMedia(tab_contents, constraints);
+
+  // Wait for the bar to pop up, then return it
+  infobar_added.Wait();
+  content::Details<InfoBarAddedDetails> details(infobar_added.details());
+  MediaStreamInfoBarDelegate* media_infobar =
+      details.ptr()->AsMediaStreamInfoBarDelegate();
+  return media_infobar;
+}
+
+void WebrtcTestBase::CloseInfobarInTab(content::WebContents* tab_contents,
+                                       MediaStreamInfoBarDelegate* infobar) {
+  content::WindowedNotificationObserver infobar_removed(
+      chrome::NOTIFICATION_TAB_CONTENTS_INFOBAR_REMOVED,
+      content::NotificationService::AllSources());
+
+  InfoBarService* infobar_service =
+      InfoBarService::FromWebContents(tab_contents);
+  infobar_service->RemoveInfoBar(infobar);
+
+  infobar_removed.Wait();
+}
+
+void WebrtcTestBase::GetUserMediaAndAccept(content::WebContents* tab_contents) {
+  GetUserMediaWithSpecificConstraintsAndAccept(tab_contents,
+                                             kAudioVideoCallConstraints);

[tommi (sloooow) - chröme, 2013/07/25 14:09:09]

indent

[phoglund_chromium, 2013/07/25 15:22:46]

Done.

+}
+
+void WebrtcTestBase::GetUserMediaWithSpecificConstraintsAndAccept(
+    content::WebContents* tab_contents, const std::string& constraints) {
+  MediaStreamInfoBarDelegate* media_infobar =
+      GetUserMediaAndWaitForInfobar(tab_contents, constraints);
+
+  media_infobar->Accept();
+
+  CloseInfobarInTab(tab_contents, media_infobar);
+
+  // Wait for WebRTC to call the success callback.
+  EXPECT_TRUE(PollingWaitUntil(
+      "obtainGetUserMediaResult()", kOkGotStream, tab_contents));
+}
+
+void WebrtcTestBase::GetUserMediaAndDeny(content::WebContents* tab_contents) {
+  return GetUserMediaWithSpecificConstraintsAndDeny(tab_contents,
+                                         kAudioVideoCallConstraints);

[tommi (sloooow) - chröme, 2013/07/25 14:09:09]

indent

[phoglund_chromium, 2013/07/25 15:22:46]

Done.

+}
+
+void WebrtcTestBase::GetUserMediaWithSpecificConstraintsAndDeny(
+    content::WebContents* tab_contents,
+    const std::string& constraints) {
+  MediaStreamInfoBarDelegate* media_infobar =
+      GetUserMediaAndWaitForInfobar(tab_contents, constraints);
+
+  media_infobar->Cancel();
+
+  CloseInfobarInTab(tab_contents, media_infobar);
+
+  // Wait for WebRTC to call the fail callback.
+  EXPECT_TRUE(PollingWaitUntil("obtainGetUserMediaResult()",
+                               kFailedWithErrorPermissionDenied,
+                               tab_contents));
+}
+
+void WebrtcTestBase::GetUserMediaAndDismiss(
+    content::WebContents* tab_contents) {
+  MediaStreamInfoBarDelegate* media_infobar =
+      GetUserMediaAndWaitForInfobar(tab_contents, kAudioVideoCallConstraints);
+
+  media_infobar->InfoBarDismissed();
+
+  CloseInfobarInTab(tab_contents, media_infobar);
+
+  // A dismiss should be treated like a deny.
+  EXPECT_TRUE(PollingWaitUntil("obtainGetUserMediaResult()",
+                               kFailedWithErrorPermissionDenied,
+                               tab_contents));
+}
+
+void WebrtcTestBase::GetUserMedia(content::WebContents* tab_contents,
+                                  const std::string& constraints) {
+  // Request user media: this will launch the media stream info bar.
+  EXPECT_EQ("ok-requested",
+            ExecuteJavascript(
+                base::StringPrintf("getUserMedia(%s);", constraints.c_str()),
+                tab_contents));
+}
+