viz_test_suite.cc 1.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051
  1. // Copyright 2017 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. #include "components/viz/test/viz_test_suite.h"
  5. #include "base/threading/thread_id_name_manager.h"
  6. #include "components/viz/test/paths.h"
  7. #include "ui/events/platform/platform_event_source.h"
  8. #include "ui/gl/test/gl_surface_test_support.h"
  9. namespace viz {
  10. std::unique_ptr<base::test::TaskEnvironment> VizTestSuite::task_environment_;
  11. VizTestSuite::VizTestSuite(int argc, char** argv)
  12. : base::TestSuite(argc, argv) {}
  13. VizTestSuite::~VizTestSuite() = default;
  14. // static
  15. void VizTestSuite::RunUntilIdle() {
  16. CHECK(task_environment_);
  17. task_environment_->RunUntilIdle();
  18. }
  19. void VizTestSuite::Initialize() {
  20. base::TestSuite::Initialize();
  21. // Must be initialized after time outs are initialized in by the TestSuite.
  22. CHECK(!task_environment_);
  23. task_environment_ = std::make_unique<base::test::TaskEnvironment>(
  24. base::test::TaskEnvironment::MainThreadType::UI);
  25. platform_event_source_ = ui::PlatformEventSource::CreateDefault();
  26. gl::GLSurfaceTestSupport::InitializeOneOff();
  27. Paths::RegisterPathProvider();
  28. base::ThreadIdNameManager::GetInstance()->SetName("Main");
  29. base::DiscardableMemoryAllocator::SetInstance(&discardable_memory_allocator_);
  30. }
  31. void VizTestSuite::Shutdown() {
  32. platform_event_source_.reset();
  33. task_environment_.reset();
  34. base::TestSuite::Shutdown();
  35. }
  36. } // namespace viz