chromium/chrome/browser/optimization_guide/model_execution/model_execution_validation_browsertest.cc

// 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 "base/test/metrics/histogram_tester.h"
#include "base/test/scoped_feature_list.h"
#include "build/chromeos_buildflags.h"
#include "chrome/browser/optimization_guide/browser_test_util.h"
#include "chrome/browser/optimization_guide/model_validator_keyed_service.h"
#include "chrome/browser/signin/identity_test_environment_profile_adaptor.h"
#include "chrome/browser/ui/browser.h"
#include "chrome/test/base/in_process_browser_test.h"
#include "components/optimization_guide/core/optimization_guide_constants.h"
#include "components/optimization_guide/core/optimization_guide_features.h"
#include "components/optimization_guide/core/optimization_guide_switches.h"
#include "components/optimization_guide/core/optimization_guide_util.h"
#include "components/optimization_guide/proto/string_value.pb.h"
#include "components/signin/public/identity_manager/account_capabilities_test_mutator.h"
#include "content/public/test/browser_test.h"
#include "content/public/test/browser_test_utils.h"
#include "net/dns/mock_host_resolver.h"
#include "testing/gmock/include/gmock/gmock.h"

namespace optimization_guide {

class ModelExecutionValidationBrowserTestBase : public InProcessBrowserTest { … };

class ModelExecutionValidationBrowserTest
    : public ModelExecutionValidationBrowserTestBase { … };

// TODO(b/318433299, crbug.com/1520214): Flaky on linux-chromeos, Win and Mac.
#if BUILDFLAG(IS_CHROMEOS) || BUILDFLAG(IS_WIN) || BUILDFLAG(IS_MAC)
#define MAYBE_ModelExecutionSuccess …
#else
#define MAYBE_ModelExecutionSuccess …
#endif
IN_PROC_BROWSER_TEST_F(ModelExecutionValidationBrowserTest,
                       MAYBE_ModelExecutionSuccess) { … }

// TODO(b/318433299, crbug.com/1520214): Flaky on linux-chromeos and win
#if BUILDFLAG(IS_CHROMEOS) || BUILDFLAG(IS_WIN)
#define MAYBE_ModelExecutionFailsServerFailure …
#else
#define MAYBE_ModelExecutionFailsServerFailure …
#endif
IN_PROC_BROWSER_TEST_F(ModelExecutionValidationBrowserTest,
                       MAYBE_ModelExecutionFailsServerFailure) { … }

}  // namespace optimization_guide