| OLD | NEW |
| (Empty) |
| 1 // Copyright 2014 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 <vector> | |
| 6 | |
| 7 #include "base/memory/scoped_ptr.h" | |
| 8 #include "base/message_loop/message_loop.h" | |
| 9 #include "testing/gtest/include/gtest/gtest.h" | |
| 10 #include "third_party/skia/include/core/SkCanvas.h" | |
| 11 #include "third_party/skia/include/core/SkColor.h" | |
| 12 #include "third_party/skia/include/core/SkImageInfo.h" | |
| 13 #include "ui/ozone/platform/dri/dri_buffer.h" | |
| 14 #include "ui/ozone/platform/dri/dri_surface.h" | |
| 15 #include "ui/ozone/platform/dri/dri_surface_factory.h" | |
| 16 #include "ui/ozone/platform/dri/dri_window_delegate_impl.h" | |
| 17 #include "ui/ozone/platform/dri/dri_window_delegate_manager.h" | |
| 18 #include "ui/ozone/platform/dri/hardware_display_controller.h" | |
| 19 #include "ui/ozone/platform/dri/screen_manager.h" | |
| 20 #include "ui/ozone/platform/dri/test/mock_dri_wrapper.h" | |
| 21 #include "ui/ozone/public/surface_factory_ozone.h" | |
| 22 #include "ui/ozone/public/surface_ozone_canvas.h" | |
| 23 | |
| 24 namespace { | |
| 25 | |
| 26 // Mode of size 6x4. | |
| 27 const drmModeModeInfo kDefaultMode = | |
| 28 {0, 6, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0, 0, {'\0'}}; | |
| 29 | |
| 30 const uint32_t kDefaultCrtc = 1; | |
| 31 const uint32_t kDefaultConnector = 2; | |
| 32 | |
| 33 class MockScreenManager : public ui::ScreenManager { | |
| 34 public: | |
| 35 MockScreenManager(ui::DriWrapper* dri, | |
| 36 ui::ScanoutBufferGenerator* buffer_generator) | |
| 37 : ScreenManager(dri, buffer_generator), | |
| 38 dri_(dri) {} | |
| 39 virtual ~MockScreenManager() {} | |
| 40 | |
| 41 // Normally we'd use DRM to figure out the controller configuration. But we | |
| 42 // can't use DRM in unit tests, so we just create a fake configuration. | |
| 43 virtual void ForceInitializationOfPrimaryDisplay() override { | |
| 44 ConfigureDisplayController( | |
| 45 kDefaultCrtc, kDefaultConnector, gfx::Point(), kDefaultMode); | |
| 46 } | |
| 47 | |
| 48 private: | |
| 49 ui::DriWrapper* dri_; // Not owned. | |
| 50 | |
| 51 DISALLOW_COPY_AND_ASSIGN(MockScreenManager); | |
| 52 }; | |
| 53 | |
| 54 } // namespace | |
| 55 | |
| 56 class DriSurfaceFactoryTest : public testing::Test { | |
| 57 public: | |
| 58 DriSurfaceFactoryTest() {} | |
| 59 | |
| 60 virtual void SetUp() override; | |
| 61 virtual void TearDown() override; | |
| 62 protected: | |
| 63 scoped_ptr<base::MessageLoop> message_loop_; | |
| 64 scoped_ptr<ui::MockDriWrapper> dri_; | |
| 65 scoped_ptr<ui::DriBufferGenerator> buffer_generator_; | |
| 66 scoped_ptr<MockScreenManager> screen_manager_; | |
| 67 scoped_ptr<ui::DriSurfaceFactory> factory_; | |
| 68 scoped_ptr<ui::DriWindowDelegateManager> window_delegate_manager_; | |
| 69 | |
| 70 private: | |
| 71 DISALLOW_COPY_AND_ASSIGN(DriSurfaceFactoryTest); | |
| 72 }; | |
| 73 | |
| 74 void DriSurfaceFactoryTest::SetUp() { | |
| 75 message_loop_.reset(new base::MessageLoopForUI); | |
| 76 dri_.reset(new ui::MockDriWrapper(3)); | |
| 77 buffer_generator_.reset(new ui::DriBufferGenerator(dri_.get())); | |
| 78 screen_manager_.reset(new MockScreenManager(dri_.get(), | |
| 79 buffer_generator_.get())); | |
| 80 window_delegate_manager_.reset(new ui::DriWindowDelegateManager()); | |
| 81 factory_.reset(new ui::DriSurfaceFactory( | |
| 82 dri_.get(), screen_manager_.get(), window_delegate_manager_.get())); | |
| 83 | |
| 84 scoped_ptr<ui::DriWindowDelegate> window_delegate( | |
| 85 new ui::DriWindowDelegateImpl(ui::DriSurfaceFactory::kDefaultWidgetHandle, | |
| 86 screen_manager_.get())); | |
| 87 window_delegate->Initialize(); | |
| 88 window_delegate_manager_->AddWindowDelegate( | |
| 89 ui::DriSurfaceFactory::kDefaultWidgetHandle, window_delegate.Pass()); | |
| 90 } | |
| 91 | |
| 92 void DriSurfaceFactoryTest::TearDown() { | |
| 93 scoped_ptr<ui::DriWindowDelegate> delegate = | |
| 94 window_delegate_manager_->RemoveWindowDelegate( | |
| 95 ui::DriSurfaceFactory::kDefaultWidgetHandle); | |
| 96 delegate->Shutdown(); | |
| 97 factory_.reset(); | |
| 98 message_loop_.reset(); | |
| 99 } | |
| 100 | |
| 101 TEST_F(DriSurfaceFactoryTest, FailInitialization) { | |
| 102 dri_->fail_init(); | |
| 103 EXPECT_EQ(ui::DriSurfaceFactory::FAILED, factory_->InitializeHardware()); | |
| 104 } | |
| 105 | |
| 106 TEST_F(DriSurfaceFactoryTest, SuccessfulInitialization) { | |
| 107 EXPECT_EQ(ui::DriSurfaceFactory::INITIALIZED, | |
| 108 factory_->InitializeHardware()); | |
| 109 } | |
| 110 | |
| 111 TEST_F(DriSurfaceFactoryTest, SuccessfulWidgetRealization) { | |
| 112 EXPECT_EQ(ui::DriSurfaceFactory::INITIALIZED, | |
| 113 factory_->InitializeHardware()); | |
| 114 | |
| 115 EXPECT_TRUE(factory_->CreateCanvasForWidget( | |
| 116 ui::DriSurfaceFactory::kDefaultWidgetHandle)); | |
| 117 } | |
| 118 | |
| 119 TEST_F(DriSurfaceFactoryTest, SetCursorImage) { | |
| 120 EXPECT_EQ(ui::DriSurfaceFactory::INITIALIZED, | |
| 121 factory_->InitializeHardware()); | |
| 122 | |
| 123 scoped_ptr<ui::SurfaceOzoneCanvas> surf = factory_->CreateCanvasForWidget( | |
| 124 ui::DriSurfaceFactory::kDefaultWidgetHandle); | |
| 125 EXPECT_TRUE(surf); | |
| 126 | |
| 127 SkBitmap image; | |
| 128 SkImageInfo info = SkImageInfo::Make( | |
| 129 6, 4, kN32_SkColorType, kPremul_SkAlphaType); | |
| 130 image.allocPixels(info); | |
| 131 image.eraseColor(SK_ColorWHITE); | |
| 132 | |
| 133 std::vector<SkBitmap> cursor_bitmaps; | |
| 134 cursor_bitmaps.push_back(image); | |
| 135 factory_->SetHardwareCursor(ui::DriSurfaceFactory::kDefaultWidgetHandle, | |
| 136 cursor_bitmaps, | |
| 137 gfx::Point(4, 2), | |
| 138 0); | |
| 139 | |
| 140 SkBitmap cursor; | |
| 141 // Buffers 0 and 1 are the cursor buffers. | |
| 142 cursor.setInfo(dri_->buffers()[1]->getCanvas()->imageInfo()); | |
| 143 EXPECT_TRUE(dri_->buffers()[1]->getCanvas()->readPixels(&cursor, 0, 0)); | |
| 144 | |
| 145 // Check that the frontbuffer is displaying the right image as set above. | |
| 146 for (int i = 0; i < cursor.height(); ++i) { | |
| 147 for (int j = 0; j < cursor.width(); ++j) { | |
| 148 if (j < info.width() && i < info.height()) | |
| 149 EXPECT_EQ(SK_ColorWHITE, cursor.getColor(j, i)); | |
| 150 else | |
| 151 EXPECT_EQ(static_cast<SkColor>(SK_ColorTRANSPARENT), | |
| 152 cursor.getColor(j, i)); | |
| 153 } | |
| 154 } | |
| 155 } | |
| OLD | NEW |