| Index: views/test/views_test_base.h
|
| diff --git a/views/test/views_test_base.h b/views/test/views_test_base.h
|
| index 4eb111ac4ae767453700821fbff45e4f8feba7b0..0d27c3f7c45e93674971f11f5c997813374085fc 100644
|
| --- a/views/test/views_test_base.h
|
| +++ b/views/test/views_test_base.h
|
| @@ -9,10 +9,13 @@
|
| #include "testing/gtest/include/gtest/gtest.h"
|
|
|
| #include "base/message_loop.h"
|
| +#include "base/scoped_ptr.h"
|
| #include "views/test/test_views_delegate.h"
|
|
|
| namespace views {
|
|
|
| +class TestViewsDelegate;
|
| +
|
| // A base class for views unit test. It creates a message loop necessary
|
| // to drive UI events and takes care of OLE initialization for windows.
|
| class ViewsTestBase : public testing::Test {
|
| @@ -21,18 +24,25 @@ class ViewsTestBase : public testing::Test {
|
| virtual ~ViewsTestBase();
|
|
|
| // testing::Test:
|
| - virtual void TearDown();
|
| + virtual void SetUp() OVERRIDE;
|
| + virtual void TearDown() OVERRIDE;
|
|
|
| void RunPendingMessages() {
|
| message_loop_.RunAllPending();
|
| }
|
|
|
| protected:
|
| - TestViewsDelegate& views_delegate() { return views_delegate_; }
|
| + TestViewsDelegate& views_delegate() const { return *views_delegate_.get(); }
|
| +
|
| + void set_views_delegate(TestViewsDelegate* views_delegate) {
|
| + views_delegate_.reset(views_delegate);
|
| + }
|
|
|
| private:
|
| MessageLoopForUI message_loop_;
|
| - TestViewsDelegate views_delegate_;
|
| + scoped_ptr<TestViewsDelegate> views_delegate_;
|
| + bool setup_called_;
|
| + bool teardown_called_;
|
|
|
| DISALLOW_COPY_AND_ASSIGN(ViewsTestBase);
|
| };
|
|
|