Athena unittests framework

athena/test/athena_test_base.h

+// Copyright 2014 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#ifndef ATHENA_TEST_ATHENA_TEST_BASE_H_
+#define ATHENA_TEST_ATHENA_TEST_BASE_H_
+
+#include "athena/test/athena_test_helper.h"
+#include "base/macros.h"
+#include "base/message_loop/message_loop.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+namespace aura {
+class Window;
+}
+
+namespace athena {
+namespace test {
+
+// A base class for athena unit tests.
+class AthenaTestBase : public testing::Test {
+ public:
+  AthenaTestBase();
+  virtual ~AthenaTestBase();
+
+  // testing::Test:
+  virtual void SetUp() OVERRIDE;
+  virtual void TearDown() OVERRIDE;
+
+ protected:
+  void RunAllPendingInMessageLoop();
+
+  aura::Window* root_window() { return helper_->root_window(); }
+  aura::WindowTreeHost* host() { return helper_->host(); }
+
+ private:
+  bool setup_called_;
+  bool teardown_called_;
+
+  base::MessageLoopForUI message_loop_;
+  scoped_ptr<AthenaTestHelper> helper_;
+
+  DISALLOW_COPY_AND_ASSIGN(AthenaTestBase);
+};
+
+}  // namespace test
+}  // namespace athena
+
+#endif  // ATHENA_TEST_ATHENA_TEST_BASE_H_