// Copyright 2024 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "chrome/browser/ash/arc/tracing/test/overview_tracing_test_base.h"

#include <stdlib.h>
#include <time.h>

#include "ash/constants/ash_switches.h"
#include "base/test/test_file_util.h"
#include "chrome/browser/ash/arc/tracing/test/overview_tracing_test_handler.h"
#include "chrome/test/base/testing_profile.h"
#include "components/exo/surface.h"
#include "components/exo/wm_helper.h"
#include "content/public/test/browser_task_environment.h"

namespace arc {

OverviewTracingTestBase::OverviewTracingTestBase()
    : ChromeAshTestBase(std::make_unique<content::BrowserTaskEnvironment>(
          base::test::TaskEnvironment::TimeSource::MOCK_TIME)) {}

OverviewTracingTestBase::~OverviewTracingTestBase() = default;

void OverviewTracingTestBase::SetUp() {
  arc_app_test_.PreProfileSetUp();
  ChromeAshTestBase::SetUp();
  profile_ = std::make_unique<TestingProfile>();
  arc_app_test_.PostProfileSetUp(profile_.get());

  // WMHelper constructor sets a global instance which the Handler constructor
  // requires.
  wm_helper_ = std::make_unique<exo::WMHelper>();
  download_path_ = base::GetTempDirForTesting();

  base::CommandLine::ForCurrentProcess()->AppendSwitch(
      ash::switches::kEnableArcVm);

  saved_tz_.reset(icu::TimeZone::createDefault());
  const char* system_tz = getenv("TZ");
  if (system_tz) {
    saved_system_tz_ = system_tz;
  } else {
    saved_system_tz_.reset();
  }
}

// static
void OverviewTracingTestBase::SetTimeZone(const char* name) {
  std::unique_ptr<icu::TimeZone> tz{icu::TimeZone::createTimeZone(name)};
  icu::TimeZone::setDefault(*tz);
  setenv("TZ", name, 1);
  tzset();
}

void OverviewTracingTestBase::TearDown() {
  icu::TimeZone::setDefault(*saved_tz_);
  if (saved_system_tz_) {
    setenv("TZ", saved_system_tz_->c_str(), 1);
  } else {
    unsetenv("TZ");
  }
  tzset();

  wm_helper_.reset();

  arc_app_test_.PreProfileTearDown();

  profile_.reset();

  ChromeAshTestBase::TearDown();
  arc_app_test_.PostProfileTearDown();
}

void OverviewTracingTestBase::CommitAndPresentFrames(
    arc::OverviewTracingTestHandler* handler,
    exo::Surface* surface,
    int count,
    base::TimeDelta delta) {
  for (int i = 0; i < count; i++) {
    surface->Commit();
    std::list<exo::Surface::FrameCallback> frame_callbacks;
    std::list<exo::Surface::PresentationCallback> presentation_callbacks;
    surface->AppendSurfaceHierarchyCallbacks(&frame_callbacks,
                                             &presentation_callbacks);
    gfx::PresentationFeedback feedback(handler->SystemTicksNow(),
                                       /*interval=*/base::TimeDelta(),
                                       /*flags=*/0);
    for (auto& cb : presentation_callbacks) {
      cb.Run(feedback);
    }
    FastForwardClockAndTaskQueue(handler, delta);
  }
}

void OverviewTracingTestBase::FastForwardClockAndTaskQueue(
    arc::OverviewTracingTestHandler* handler,
    base::TimeDelta delta) {
  handler->set_now(handler->Now() + delta);
  task_environment()->FastForwardBy(delta);
}

}  // namespace arc