athena/test/base/athena_test_base.h

   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 #ifndef ATHENA_TEST_BASE_ATHENA_TEST_BASE_H_
   6 #define ATHENA_TEST_BASE_ATHENA_TEST_BASE_H_
   7
   8 #include "athena/test/base/athena_test_helper.h"
   9 #include "base/macros.h"
  10 #include "base/message_loop/message_loop.h"
  11 #include "testing/gtest/include/gtest/gtest.h"
  12
  13 namespace aura {
  14 class Window;
  15 class WindowDelegate;
  16 }
  17
  18 namespace athena {
  19 namespace test {
  20
  21 // A base class for athena unit tests.
  22 class AthenaTestBase : public testing::Test {
  23  public:
  24   AthenaTestBase();
  25   ~AthenaTestBase() override;
  26
  27   // testing::Test:
  28   void SetUp() override;
  29   void TearDown() override;
  30
  31  protected:
  32   void RunAllPendingInMessageLoop();
  33
  34   aura::Window* root_window() { return helper_->GetRootWindow(); }
  35   aura::WindowTreeHost* host() { return helper_->GetHost(); }
  36
  37  private:
  38   bool setup_called_;
  39   bool teardown_called_;
  40
  41   base::MessageLoopForUI message_loop_;
  42   scoped_ptr<AthenaTestHelper> helper_;
  43
  44   DISALLOW_COPY_AND_ASSIGN(AthenaTestBase);
  45 };
  46
  47 }  // namespace test
  48 }  // namespace athena
  49
  50 #endif  // ATHENA_TEST_BASE_ATHENA_TEST_BASE_H_