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_