Adding the Previews infobar to pages that show a client LoFi image

components/data_reduction_proxy/content/browser/content_lofi_ui_service_unittest.cc

 // Copyright 2015 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "components/data_reduction_proxy/content/browser/content_lofi_ui_service.h"
 
 #include <stddef.h>
 
 #include <memory>
 
@@ skipping 16 matching lines @@
 
 namespace data_reduction_proxy {
 
 class ContentLoFiUIServiceTest : public content::RenderViewHostTestHarness {
  public:
   ContentLoFiUIServiceTest() : callback_called_(false) {
     // Cannot use IO_MAIN_LOOP with RenderViewHostTestHarness.
     SetThreadBundleOptions(content::TestBrowserThreadBundle::REAL_IO_THREAD);
   }
 
-  void RunTestOnIOThread(base::RunLoop* ui_run_loop) {
+  void RunTestOnIOThread(base::RunLoop* ui_run_loop, bool is_server_lofi) {
     ASSERT_TRUE(ui_run_loop);
     EXPECT_TRUE(
         content::BrowserThread::CurrentlyOn(content::BrowserThread::IO));
 
     net::TestURLRequestContext context(true);
     net::MockClientSocketFactory mock_socket_factory;
     net::TestDelegate delegate;
     context.set_client_socket_factory(&mock_socket_factory);
     context.Init();
 
     content_lofi_ui_service_.reset(new ContentLoFiUIService(
         content::BrowserThread::GetTaskRunnerForThread(
             content::BrowserThread::UI),
         base::Bind(&ContentLoFiUIServiceTest::OnLoFiResponseReceivedCallback,
                    base::Unretained(this))));
 
     std::unique_ptr<net::URLRequest> request =
         CreateRequest(context, &delegate);
 
-    content_lofi_ui_service_->OnLoFiReponseReceived(*request);
+    content_lofi_ui_service_->OnLoFiReponseReceived(*request, is_server_lofi);
 
     content::BrowserThread::PostTask(
         content::BrowserThread::UI, FROM_HERE,
         base::Bind(&base::RunLoop::Quit, base::Unretained(ui_run_loop)));
   }
 
   std::unique_ptr<net::URLRequest> CreateRequest(
       const net::TestURLRequestContext& context,
       net::TestDelegate* delegate) {
     EXPECT_TRUE(
         content::BrowserThread::CurrentlyOn(content::BrowserThread::IO));
 
     std::unique_ptr<net::URLRequest> request =
         context.CreateRequest(GURL("http://www.google.com/"), net::IDLE,
                               delegate, TRAFFIC_ANNOTATION_FOR_TESTS);
 
     content::ResourceRequestInfo::AllocateForTesting(
         request.get(), content::RESOURCE_TYPE_SUB_FRAME, NULL,
         web_contents()->GetMainFrame()->GetProcess()->GetID(), -1,
         web_contents()->GetMainFrame()->GetRoutingID(),
         /*is_main_frame=*/false,
         /*parent_is_main_frame=*/false,
         /*allow_download=*/false,
         /*is_async=*/false, content::SERVER_LOFI_ON);
 
     return request;
   }
 
-  void OnLoFiResponseReceivedCallback(content::WebContents* web_contents) {
+  void OnLoFiResponseReceivedCallback(content::WebContents* web_contents,
+                                      bool is_server_lofi) {
     EXPECT_TRUE(
         content::BrowserThread::CurrentlyOn(content::BrowserThread::UI));
     callback_called_ = true;
+    is_server_lofi_ = is_server_lofi;
   }
 
-  void VerifyOnLoFiResponseReceivedCallback() {
+  void VerifyOnLoFiResponseReceivedCallback(bool is_server_lofi) {

[sclittle, 2017/05/03 22:51:29]

nit: could this |is_server_lofi| argument here be renamed to
|expected_is_server_lofi| here to distinguish it from the class member variable?

[RyanSturm, 2017/05/03 23:20:30]

Done.

     EXPECT_TRUE(
         content::BrowserThread::CurrentlyOn(content::BrowserThread::UI));
     EXPECT_TRUE(callback_called_);
+    EXPECT_EQ(is_server_lofi, is_server_lofi_);
   }
 
  private:
   std::unique_ptr<ContentLoFiUIService> content_lofi_ui_service_;
   bool callback_called_;
+  bool is_server_lofi_;
 };
 
-TEST_F(ContentLoFiUIServiceTest, OnLoFiResponseReceived) {
+TEST_F(ContentLoFiUIServiceTest, OnLoFiResponseReceivedServer) {
   base::RunLoop ui_run_loop;
+  bool is_server_lofi = true;
   content::BrowserThread::PostTask(
       content::BrowserThread::IO, FROM_HERE,
       base::Bind(&ContentLoFiUIServiceTest::RunTestOnIOThread,
-                 base::Unretained(this), &ui_run_loop));
+                 base::Unretained(this), &ui_run_loop, is_server_lofi));
   ui_run_loop.Run();
   base::RunLoop().RunUntilIdle();
-  VerifyOnLoFiResponseReceivedCallback();
+  VerifyOnLoFiResponseReceivedCallback(is_server_lofi);
+}
+
+TEST_F(ContentLoFiUIServiceTest, OnLoFiResponseReceivedClient) {
+  base::RunLoop ui_run_loop;
+  bool is_server_lofi = false;
+  content::BrowserThread::PostTask(
+      content::BrowserThread::IO, FROM_HERE,
+      base::Bind(&ContentLoFiUIServiceTest::RunTestOnIOThread,
+                 base::Unretained(this), &ui_run_loop, is_server_lofi));
+  ui_run_loop.Run();
+  base::RunLoop().RunUntilIdle();
+  VerifyOnLoFiResponseReceivedCallback(is_server_lofi);
 }
 
 }  // namespace data_reduction_proxy