| OLD | NEW |
| (Empty) |
| 1 // Copyright (c) 2011 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 "views/test/views_test_base.h" | |
| 6 | |
| 7 #if defined(OS_WIN) | |
| 8 #include <ole2.h> | |
| 9 #endif | |
| 10 | |
| 11 #if defined(USE_AURA) | |
| 12 #include "ui/aura/desktop.h" | |
| 13 #include "ui/aura/test/test_stacking_client.h" | |
| 14 #endif | |
| 15 | |
| 16 namespace views { | |
| 17 | |
| 18 ViewsTestBase::ViewsTestBase() | |
| 19 : setup_called_(false), | |
| 20 teardown_called_(false) { | |
| 21 #if defined(OS_WIN) | |
| 22 OleInitialize(NULL); | |
| 23 #endif | |
| 24 #if defined(USE_AURA) | |
| 25 new aura::test::TestStackingClient; | |
| 26 #endif | |
| 27 } | |
| 28 | |
| 29 ViewsTestBase::~ViewsTestBase() { | |
| 30 #if defined(OS_WIN) | |
| 31 OleUninitialize(); | |
| 32 #endif | |
| 33 CHECK(setup_called_) | |
| 34 << "You have overridden SetUp but never called super class's SetUp"; | |
| 35 CHECK(teardown_called_) | |
| 36 << "You have overrideen TearDown but never called super class's TearDown"; | |
| 37 } | |
| 38 | |
| 39 void ViewsTestBase::SetUp() { | |
| 40 testing::Test::SetUp(); | |
| 41 setup_called_ = true; | |
| 42 if (!views_delegate_.get()) | |
| 43 views_delegate_.reset(new TestViewsDelegate()); | |
| 44 } | |
| 45 | |
| 46 void ViewsTestBase::TearDown() { | |
| 47 // Flush the message loop because we have pending release tasks | |
| 48 // and these tasks if un-executed would upset Valgrind. | |
| 49 RunPendingMessages(); | |
| 50 teardown_called_ = true; | |
| 51 views_delegate_.reset(); | |
| 52 testing::Test::TearDown(); | |
| 53 } | |
| 54 | |
| 55 void ViewsTestBase::RunPendingMessages() { | |
| 56 #if defined(USE_AURA) | |
| 57 message_loop_.RunAllPendingWithDispatcher( | |
| 58 aura::Desktop::GetInstance()->GetDispatcher()); | |
| 59 #else | |
| 60 message_loop_.RunAllPending(); | |
| 61 #endif | |
| 62 } | |
| 63 | |
| 64 } // namespace views | |
| OLD | NEW |