Add checks for events in DeepScanningDialogDelegate browsertests

Bug: 1059780
Change-Id: I75f74fcd70eee7fbd61a6a32aa9104e148b4d9fd
Reviewed-on: https://chromium-review.googlesource.com/c/chromium/src/+/2102637Reviewed-by: Daniel Rubery <drubery@chromium.org>
Commit-Queue: Dominique Fauteux-Chapleau <domfc@chromium.org>
Cr-Commit-Position: refs/heads/master@{#750643}

chrome/browser/extensions/api/safe_browsing_private/safe_browsing_private_event_router.cc

@@ -659,6 +659,12 @@ void SafeBrowsingPrivateEventRouter::SetCloudPolicyClientForTesting(
   client_ = client;
 }
 
+void SafeBrowsingPrivateEventRouter::SetBinaryUploadServiceForTesting(
+    safe_browsing::BinaryUploadService* binary_upload_service) {
+  DCHECK_EQ(nullptr, binary_upload_service_);
+  binary_upload_service_ = binary_upload_service;
+}
+
 void SafeBrowsingPrivateEventRouter::InitRealtimeReportingClient() {
   // If already initialized, do nothing.
   if (client_) {

chrome/browser/extensions/api/safe_browsing_private/safe_browsing_private_event_router.h

@@ -194,6 +194,9 @@ class SafeBrowsingPrivateEventRouter : public KeyedService {
 
   void SetCloudPolicyClientForTesting(policy::CloudPolicyClient* client);
 
+  void SetBinaryUploadServiceForTesting(
+      safe_browsing::BinaryUploadService* binary_upload_service);
+
  protected:
   // Callback to report safe browsing event through real-time reporting channel,
   // if the browser is authorized to do so. Declared as protected to be called

chrome/browser/safe_browsing/cloud_content_scanning/deep_scanning_browsertest_base.cc

@@ -83,6 +83,9 @@ void DeepScanningBrowserTestBase::TearDownOnMainThread() {
   SetWaitPolicy(DelayDeliveryUntilVerdictValues::DELAY_NONE);
   SetAllowPasswordProtectedFilesPolicy(
       AllowPasswordProtectedFilesValues::ALLOW_UPLOADS_AND_DOWNLOADS);
+  SetBlockUnsupportedFileTypesPolicy(
+      BlockUnsupportedFiletypesValues::BLOCK_UNSUPPORTED_FILETYPES_NONE);
+  SetUnsafeEventsReportingPolicy(false);
 }
 
 void DeepScanningBrowserTestBase::SetDlpPolicy(
@@ -115,6 +118,11 @@ void DeepScanningBrowserTestBase::SetBlockUnsupportedFileTypesPolicy(
       prefs::kBlockUnsupportedFiletypes, state);
 }
 
+void DeepScanningBrowserTestBase::SetUnsafeEventsReportingPolicy(bool report) {
+  g_browser_process->local_state()->SetBoolean(
+      prefs::kUnsafeEventsReportingEnabled, report);
+}
+
 void DeepScanningBrowserTestBase::SetUpDelegate() {
   SetDMTokenForTesting(policy::DMToken::CreateValidTokenForTesting(kDmToken));
   DeepScanningDialogDelegate::SetFactoryForTesting(base::BindRepeating(

chrome/browser/safe_browsing/cloud_content_scanning/deep_scanning_browsertest_base.h

@@ -28,6 +28,7 @@ class DeepScanningBrowserTestBase : public InProcessBrowserTest {
       AllowPasswordProtectedFilesValues state);
   void SetBlockUnsupportedFileTypesPolicy(
       BlockUnsupportedFiletypesValues state);
+  void SetUnsafeEventsReportingPolicy(bool report);
 
   // Sets up a FakeDeepScanningDialogDelegate to use this class's StatusCallback
   // and EncryptionStatusCallback. Also sets up a test DM token.