From 07ea8b588068bc147ed5daab1eb093b319e1a016 Mon Sep 17 00:00:00 2001
From: sergei <spylogsster@gmail.com>
Date: Tue, 18 Apr 2023 16:04:12 +0300
Subject: [PATCH] Fix tests

---
 browser/brave_vpn/brave_vpn_service_factory.cc | 3 ++-
 test/base/testing_brave_browser_process.cc     | 1 -
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/browser/brave_vpn/brave_vpn_service_factory.cc b/browser/brave_vpn/brave_vpn_service_factory.cc
index 4785a592196..aa616891b83 100644
--- a/browser/brave_vpn/brave_vpn_service_factory.cc
+++ b/browser/brave_vpn/brave_vpn_service_factory.cc
@@ -68,7 +68,8 @@ BraveVpnServiceFactory::~BraveVpnServiceFactory() = default;
 
 KeyedService* BraveVpnServiceFactory::BuildServiceInstanceFor(
     content::BrowserContext* context) const {
-  if (!brave_vpn::IsAllowedForContext(context)) {
+  if (!brave_vpn::IsAllowedForContext(context) ||
+      !g_brave_browser_process->brave_vpn_os_connection_api()) {
     return nullptr;
   }
 
diff --git a/test/base/testing_brave_browser_process.cc b/test/base/testing_brave_browser_process.cc
index 6823d85e767..f9e24201a46 100644
--- a/test/base/testing_brave_browser_process.cc
+++ b/test/base/testing_brave_browser_process.cc
@@ -155,7 +155,6 @@ TestingBraveBrowserProcess::brave_farbling_service() {
 #if BUILDFLAG(ENABLE_BRAVE_VPN)
 brave_vpn::BraveVPNOSConnectionAPI*
 TestingBraveBrowserProcess::brave_vpn_os_connection_api() {
-  NOTREACHED();
   return nullptr;
 }
 #endif