WebUI: Add JavaScript lifecycle-control to WebUIMessageHandler.

chrome/browser/ui/webui/settings/chromeos/easy_unlock_settings_handler_unittest.cc

 // Copyright 2016 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 "chrome/browser/ui/webui/settings/chromeos/easy_unlock_settings_handler.h"
 
 #include <memory>
 
 #include "base/memory/ptr_util.h"
 #include "chrome/browser/signin/easy_unlock_service.h"
@@ skipping 178 matching lines @@
   fake_easy_unlock_service()->set_is_allowed(false);
   handler.reset(EasyUnlockSettingsHandler::Create(data_source, profile()));
   EXPECT_FALSE(handler.get());
 }
 
 TEST_F(EasyUnlockSettingsHandlerTest, EnabledStatus) {
   std::unique_ptr<EasyUnlockSettingsHandler> handler;
   handler.reset(new TestEasyUnlockSettingsHandler(profile()));
   handler->set_web_ui(web_ui());
 
-  // Test the C++ -> JS push path.
-  handler->SendEnabledStatus();
-  VerifyEnabledStatusCallback(1U, false);
-
-  fake_easy_unlock_service()->set_is_enabled(true);
-  handler->SendEnabledStatus();
-  VerifyEnabledStatusCallback(2U, true);
-
   // Test the JS -> C++ -> JS callback path.
   base::ListValue list_args;
   list_args.Append(new base::StringValue("test-callback-id"));
   handler->HandleGetEnabledStatus(&list_args);
 
-  EXPECT_EQ(3U, web_ui()->call_data().size());
+  EXPECT_EQ(1U, web_ui()->call_data().size());
 
   const content::TestWebUI::CallData& data = *web_ui()->call_data().back();
   EXPECT_EQ("cr.webUIResponse", data.function_name());
 
   std::string callback_id;
   ASSERT_TRUE(data.arg1()->GetAsString(&callback_id));
   EXPECT_EQ("test-callback-id", callback_id);
 
   bool enabled_status = false;
   ASSERT_TRUE(data.arg3()->GetAsBoolean(&enabled_status));
-  EXPECT_TRUE(enabled_status);
+  EXPECT_FALSE(enabled_status);
+
+  // Test the C++ -> JS push path.
+  handler->SendEnabledStatus();
+  VerifyEnabledStatusCallback(2U, false);
+
+  fake_easy_unlock_service()->set_is_enabled(true);
+  handler->SendEnabledStatus();
+  VerifyEnabledStatusCallback(3U, true);
 }
 
 TEST_F(EasyUnlockSettingsHandlerTest, TurnOffFlowStatus) {
   std::unique_ptr<EasyUnlockSettingsHandler> handler;
   handler.reset(new TestEasyUnlockSettingsHandler(profile()));
   handler->set_web_ui(web_ui());
 
   // Send an initial status query to turn on service observer.
   base::ListValue list_args1;
   list_args1.Append(new base::StringValue("test-callback-id-1"));
@@ skipping 21 matching lines @@
 
   base::ListValue list_args4;
   list_args4.Append(new base::StringValue("test-callback-id-4"));
   handler->HandleGetTurnOffFlowStatus(&list_args4);
   VerifyTurnOffFlowStatusWebUIResponse(7U, "test-callback-id-4",
                                        "server-error");
 }
 
 }  // namespace settings
 }  // namespace chromeos