Discharge7443/brave-core
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Enable leo opt in perf tests (#23464)

Browse Source 
* Enable leo in perf tests

* Add tests
This commit is contained in:
Mikhail
2024-05-09 00:50:37 +07:00
committed by GitHub
parent caa7784a5c
commit 874c70851e
3 changed files with 24 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
browser/perf/brave_perf_features_processor.cc
+12
View File
@@ -10,6 +10,7 @@
#include "brave/browser/brave_browser_process.h"
#include "brave/browser/brave_rewards/rewards_service_factory.h"
#include "brave/browser/perf/brave_perf_switches.h"
#include "brave/components/ai_chat/core/common/buildflags/buildflags.h"
#include "brave/components/brave_ads/core/public/prefs/pref_names.h"
#include "brave/components/brave_news/common/pref_names.h"
#include "brave/components/brave_rewards/browser/rewards_service.h"
@@ -24,6 +25,10 @@
#include "brave/components/speedreader/speedreader_pref_names.h"
#endif
#if BUILDFLAG(ENABLE_AI_CHAT)
#include "brave/components/ai_chat/core/common/pref_names.h"
#endif // BUILDFLAG(ENABLE_AI_CHAT)
namespace {
void FakeCallback(brave_rewards::mojom::CreateRewardsWalletResult) {}
@@ -78,6 +83,13 @@ void MaybeEnableBraveFeatureForPerfTesting(Profile* profile) {
profile->GetPrefs()->SetBoolean(speedreader::kSpeedreaderPrefEnabled, true);
#endif
#if BUILDFLAG(ENABLE_AI_CHAT)
profile->GetPrefs()->SetTime(ai_chat::prefs::kLastAcceptedDisclaimer,
base::Time::Now());
profile->GetPrefs()->SetBoolean(
ai_chat::prefs::kBraveChatAutocompleteProviderEnabled, true);
#endif // BUILDFLAG(ENABLE_AI_CHAT)
// Adblock
EnableAdblockCookieList(profile->GetWeakPtr());
}
browser/perf/brave_perf_features_processor_browsertest.cc
+11
View File
@@ -7,6 +7,7 @@
#include "brave/browser/brave_news/brave_news_controller_factory.h"
#include "brave/browser/brave_rewards/rewards_service_factory.h"
#include "brave/browser/perf/brave_perf_switches.h"
#include "brave/components/ai_chat/core/common/buildflags/buildflags.h"
#include "brave/components/brave_ads/core/public/prefs/pref_names.h"
#include "brave/components/brave_news/browser/brave_news_controller.h"
#include "brave/components/brave_rewards/browser/rewards_service.h"
@@ -23,6 +24,10 @@
#include "brave/components/speedreader/speedreader_service.h"
#endif
#if BUILDFLAG(ENABLE_AI_CHAT)
#include "brave/components/ai_chat/core/browser/utils.h"
#endif // BUILDFLAG(ENABLE_AI_CHAT)
namespace {
class TestRewardsServiceObserver
: public brave_rewards::RewardsServiceObserver {
@@ -93,4 +98,10 @@ IN_PROC_BROWSER_TEST_F(BraveSpeedFeatureProcessorBrowserTest, Default) {
EXPECT_TRUE(HasOptedInToNotificationAds());
EXPECT_TRUE(BraveNewsAreEnabled());
WaitForRewardsServiceInitialized();
#if BUILDFLAG(ENABLE_AI_CHAT)
auto* prefs = browser()->profile()->GetPrefs();
EXPECT_TRUE(ai_chat::IsAIChatEnabled(prefs));
EXPECT_TRUE(ai_chat::HasUserOptedIn(prefs));
#endif // BUILDFLAG(ENABLE_AI_CHAT)
}