| Index: chrome/browser/ui/cocoa/media_picker/desktop_media_picker_controller_unittest.mm
|
| diff --git a/chrome/browser/ui/cocoa/media_picker/desktop_media_picker_controller_unittest.mm b/chrome/browser/ui/cocoa/media_picker/desktop_media_picker_controller_unittest.mm
|
| index 7c518b99fca3d82e9059f58446afd99f99aee742..3901f95adbe1a4f234691bd15a9ffe159d667681 100644
|
| --- a/chrome/browser/ui/cocoa/media_picker/desktop_media_picker_controller_unittest.mm
|
| +++ b/chrome/browser/ui/cocoa/media_picker/desktop_media_picker_controller_unittest.mm
|
| @@ -7,6 +7,7 @@
|
| #include "base/bind.h"
|
| #include "base/strings/utf_string_conversions.h"
|
| #include "base/run_loop.h"
|
| +#include "chrome/browser/media/desktop_media_list_observer.h"
|
| #import "chrome/browser/ui/cocoa/cocoa_test_helper.h"
|
| #include "content/public/test/test_browser_thread_bundle.h"
|
| #include "testing/gtest_mac.h"
|
| @@ -35,15 +36,16 @@
|
| }
|
| @end
|
|
|
| -class FakeDesktopMediaPickerModel : public DesktopMediaPickerModel {
|
| +class FakeDesktopMediaList : public DesktopMediaList {
|
| public:
|
| - FakeDesktopMediaPickerModel() : observer_(NULL) {
|
| + FakeDesktopMediaList() : observer_(NULL) {
|
| }
|
|
|
| void AddSource(int id) {
|
| - Source source(
|
| - content::DesktopMediaID(content::DesktopMediaID::TYPE_WINDOW, id),
|
| - base::Int64ToString16(id));
|
| + Source source;
|
| + source.id =
|
| + content::DesktopMediaID(content::DesktopMediaID::TYPE_WINDOW, id);
|
| + source.name = base::Int64ToString16(id);
|
|
|
| sources_.push_back(source);
|
| observer_->OnSourceAdded(sources_.size() - 1);
|
| @@ -64,7 +66,7 @@ class FakeDesktopMediaPickerModel : public DesktopMediaPickerModel {
|
| observer_->OnSourceNameChanged(index);
|
| }
|
|
|
| - // DesktopMediaPickerModel implementation:
|
| + // DesktopMediaList implementation:
|
| virtual void SetUpdatePeriod(base::TimeDelta period) OVERRIDE {
|
| }
|
|
|
| @@ -75,7 +77,7 @@ class FakeDesktopMediaPickerModel : public DesktopMediaPickerModel {
|
| content::DesktopMediaID::Id dialog_id) OVERRIDE {
|
| }
|
|
|
| - virtual void StartUpdating(Observer* observer) OVERRIDE {
|
| + virtual void StartUpdating(DesktopMediaListObserver* observer) OVERRIDE {
|
| observer_ = observer;
|
|
|
| SkBitmap bitmap;
|
| @@ -85,29 +87,30 @@ class FakeDesktopMediaPickerModel : public DesktopMediaPickerModel {
|
| thumbnail_ = gfx::ImageSkia::CreateFrom1xBitmap(bitmap);
|
| }
|
|
|
| - virtual int source_count() const OVERRIDE {
|
| + virtual int GetSourceCount() const OVERRIDE {
|
| return sources_.size();
|
| }
|
|
|
| - virtual const Source& source(int index) const OVERRIDE {
|
| + virtual const Source& GetSource(int index) const OVERRIDE {
|
| return sources_[index];
|
| }
|
|
|
| private:
|
| std::vector<Source> sources_;
|
| - Observer* observer_;
|
| + DesktopMediaListObserver* observer_;
|
| gfx::ImageSkia thumbnail_;
|
| };
|
|
|
| class DesktopMediaPickerControllerTest : public CocoaTest {
|
| public:
|
| - DesktopMediaPickerControllerTest() : callback_called_(false), model_(NULL) {
|
| + DesktopMediaPickerControllerTest()
|
| + : callback_called_(false), media_list_(NULL) {
|
| }
|
|
|
| virtual void SetUp() OVERRIDE {
|
| CocoaTest::SetUp();
|
|
|
| - model_ = new FakeDesktopMediaPickerModel();
|
| + media_list_ = new FakeDesktopMediaList();
|
|
|
| DesktopMediaPicker::DoneCallback callback =
|
| base::Bind(&DesktopMediaPickerControllerTest::OnResult,
|
| @@ -115,9 +118,9 @@ class DesktopMediaPickerControllerTest : public CocoaTest {
|
|
|
| controller_.reset(
|
| [[DesktopMediaPickerController alloc]
|
| - initWithModel:scoped_ptr<DesktopMediaPickerModel>(model_)
|
| - callback:callback
|
| - appName:ASCIIToUTF16("Screenshare Test")]);
|
| + initWithMediaList:scoped_ptr<DesktopMediaList>(media_list_)
|
| + callback:callback
|
| + appName:ASCIIToUTF16("Screenshare Test")]);
|
| }
|
|
|
| virtual void TearDown() OVERRIDE {
|
| @@ -142,16 +145,16 @@ class DesktopMediaPickerControllerTest : public CocoaTest {
|
| content::TestBrowserThreadBundle thread_bundle_;
|
| bool callback_called_;
|
| content::DesktopMediaID source_reported_;
|
| - FakeDesktopMediaPickerModel* model_;
|
| + FakeDesktopMediaList* media_list_;
|
| base::scoped_nsobject<DesktopMediaPickerController> controller_;
|
| };
|
|
|
| TEST_F(DesktopMediaPickerControllerTest, ShowAndDismiss) {
|
| [controller_ showWindow:nil];
|
|
|
| - model_->AddSource(0);
|
| - model_->AddSource(1);
|
| - model_->SetSourceThumbnail(1);
|
| + media_list_->AddSource(0);
|
| + media_list_->AddSource(1);
|
| + media_list_->SetSourceThumbnail(1);
|
|
|
| NSArray* items = [controller_ items];
|
| EXPECT_EQ(2U, [items count]);
|
| @@ -164,10 +167,10 @@ TEST_F(DesktopMediaPickerControllerTest, ShowAndDismiss) {
|
| TEST_F(DesktopMediaPickerControllerTest, ClickOK) {
|
| [controller_ showWindow:nil];
|
|
|
| - model_->AddSource(0);
|
| - model_->SetSourceThumbnail(0);
|
| - model_->AddSource(1);
|
| - model_->SetSourceThumbnail(1);
|
| + media_list_->AddSource(0);
|
| + media_list_->SetSourceThumbnail(0);
|
| + media_list_->AddSource(1);
|
| + media_list_->SetSourceThumbnail(1);
|
|
|
| EXPECT_EQ(2U, [[controller_ items] count]);
|
| EXPECT_FALSE([[controller_ okButton] isEnabled]);
|
| @@ -179,16 +182,16 @@ TEST_F(DesktopMediaPickerControllerTest, ClickOK) {
|
|
|
| [[controller_ okButton] performClick:nil];
|
| EXPECT_TRUE(WaitForCallback());
|
| - EXPECT_EQ(model_->source(1).id, source_reported_);
|
| + EXPECT_EQ(media_list_->GetSource(1).id, source_reported_);
|
| }
|
|
|
| TEST_F(DesktopMediaPickerControllerTest, ClickCancel) {
|
| [controller_ showWindow:nil];
|
|
|
| - model_->AddSource(0);
|
| - model_->SetSourceThumbnail(0);
|
| - model_->AddSource(1);
|
| - model_->SetSourceThumbnail(1);
|
| + media_list_->AddSource(0);
|
| + media_list_->SetSourceThumbnail(0);
|
| + media_list_->AddSource(1);
|
| + media_list_->SetSourceThumbnail(1);
|
|
|
| [[controller_ cancelButton] performClick:nil];
|
| EXPECT_TRUE(WaitForCallback());
|
| @@ -198,10 +201,10 @@ TEST_F(DesktopMediaPickerControllerTest, ClickCancel) {
|
| TEST_F(DesktopMediaPickerControllerTest, CloseWindow) {
|
| [controller_ showWindow:nil];
|
|
|
| - model_->AddSource(0);
|
| - model_->SetSourceThumbnail(0);
|
| - model_->AddSource(1);
|
| - model_->SetSourceThumbnail(1);
|
| + media_list_->AddSource(0);
|
| + media_list_->SetSourceThumbnail(0);
|
| + media_list_->AddSource(1);
|
| + media_list_->SetSourceThumbnail(1);
|
|
|
| [controller_ close];
|
| EXPECT_TRUE(WaitForCallback());
|
| @@ -211,42 +214,42 @@ TEST_F(DesktopMediaPickerControllerTest, CloseWindow) {
|
| TEST_F(DesktopMediaPickerControllerTest, UpdateThumbnail) {
|
| [controller_ showWindow:nil];
|
|
|
| - model_->AddSource(0);
|
| - model_->SetSourceThumbnail(0);
|
| - model_->AddSource(1);
|
| - model_->SetSourceThumbnail(1);
|
| + media_list_->AddSource(0);
|
| + media_list_->SetSourceThumbnail(0);
|
| + media_list_->AddSource(1);
|
| + media_list_->SetSourceThumbnail(1);
|
|
|
| NSArray* items = [controller_ items];
|
| EXPECT_EQ(2U, [items count]);
|
| NSUInteger version = [[items objectAtIndex:0] imageVersion];
|
|
|
| - model_->SetSourceThumbnail(0);
|
| + media_list_->SetSourceThumbnail(0);
|
| EXPECT_NE(version, [[items objectAtIndex:0] imageVersion]);
|
| }
|
|
|
| TEST_F(DesktopMediaPickerControllerTest, UpdateName) {
|
| [controller_ showWindow:nil];
|
|
|
| - model_->AddSource(0);
|
| - model_->SetSourceThumbnail(0);
|
| - model_->AddSource(1);
|
| - model_->SetSourceThumbnail(1);
|
| + media_list_->AddSource(0);
|
| + media_list_->SetSourceThumbnail(0);
|
| + media_list_->AddSource(1);
|
| + media_list_->SetSourceThumbnail(1);
|
|
|
| NSArray* items = [controller_ items];
|
| EXPECT_EQ(2U, [items count]);
|
| NSUInteger version = [[items objectAtIndex:0] imageVersion];
|
|
|
| - model_->SetSourceThumbnail(0);
|
| + media_list_->SetSourceThumbnail(0);
|
| EXPECT_NE(version, [[items objectAtIndex:0] imageVersion]);
|
| }
|
|
|
| TEST_F(DesktopMediaPickerControllerTest, RemoveSource) {
|
| [controller_ showWindow:nil];
|
|
|
| - model_->AddSource(0);
|
| - model_->AddSource(1);
|
| - model_->AddSource(2);
|
| - model_->SetSourceName(1, ASCIIToUTF16("foo"));
|
| + media_list_->AddSource(0);
|
| + media_list_->AddSource(1);
|
| + media_list_->AddSource(2);
|
| + media_list_->SetSourceName(1, ASCIIToUTF16("foo"));
|
|
|
| NSArray* items = [controller_ items];
|
| EXPECT_EQ(3U, [items count]);
|
|
|