Mojo: Armed Watchers

ios/web/webui/mojo_facade_unittest.mm

 // 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.
 
 #import "ios/web/webui/mojo_facade.h"
 
 #include <memory>
 
 #import "base/mac/scoped_nsobject.h"
 #include "base/strings/string_number_conversions.h"
@@ skipping 181 matching lines @@
   int watch_id = 0;
   EXPECT_TRUE(base::StringToInt(watch_id_as_string, &watch_id));
 
   // Start waiting for the watch callback.
   __block bool callback_received = false;
   NSString* expected_script =
       [NSString stringWithFormat:@"__crWeb.mojo.signalWatch(%d, %d)",
                                  callback_id, MOJO_RESULT_OK];
   [[[evaluator() expect] andDo:^(NSInvocation*) {
     callback_received = true;
+
+    // Cancel the watch immediately to ensure there are no additional
+    // notifications.
+    NSDictionary* cancel_watch = @{
+      @"name" : @"support.cancelWatch",
+      @"args" : @{
+        @"watchId" : @(watch_id),
+      },
+    };
+    std::string result_as_string =
+        facade()->HandleMojoMessage(GetJson(cancel_watch));
+    EXPECT_TRUE(result_as_string.empty());
   }] executeJavaScript:expected_script completionHandler:nil];
 
   // Write to the other end of the pipe.
   NSDictionary* write = @{
     @"name" : @"core.writeMessage",
     @"args" : @{
       @"handle" : handle1,
       @"handles" : @[],
       @"flags" : @(MOJO_WRITE_MESSAGE_FLAG_NONE),
       @"buffer" : @{@"0" : @0}
@@ skipping 59 matching lines @@
   NSDictionary* message = GetObject(facade()->HandleMojoMessage(GetJson(read)));
   EXPECT_TRUE([message isKindOfClass:[NSDictionary class]]);
   EXPECT_TRUE(message);
   NSArray* expected_message = @[ @9, @2, @216 ];  // 2008 does not fit 8-bit.
   EXPECT_NSEQ(expected_message, message[@"buffer"]);
   EXPECT_FALSE([message[@"handles"] count]);
   EXPECT_EQ(MOJO_RESULT_OK, [message[@"result"] unsignedIntValue]);
 }
 
 }  // namespace web