Fix TabSocket.onmessage whoch should be able to retrieve all messages

headless/lib/headless_web_contents_browsertest.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 <memory>
 #include <string>
 #include <vector>
 
 #include "base/base64.h"
 #include "base/json/json_writer.h"
@@ skipping 16 matching lines @@
 #include "url/gurl.h"
 
 #if BUILDFLAG(ENABLE_BASIC_PRINTING)
 #include "base/strings/string_number_conversions.h"
 #include "pdf/pdf.h"
 #include "printing/pdf_render_settings.h"
 #include "printing/units.h"
 #include "ui/gfx/geometry/rect.h"
 #endif
 
+using testing::ElementsAre;
 using testing::UnorderedElementsAre;
 
 namespace headless {
 
 class HeadlessWebContentsTest : public HeadlessBrowserTest {};
 
 IN_PROC_BROWSER_TEST_F(HeadlessWebContentsTest, Navigation) {
   EXPECT_TRUE(embedded_test_server()->Start());
 
   HeadlessBrowserContext* browser_context =
@@ skipping 281 matching lines @@
         R"(window.TabSocket.onmessage =
             function(message) {
               window.TabSocket.send('Embedder sent us: ' + message);
             };
           )");
 
     HeadlessTabSocket* headless_tab_socket =
         web_contents_->GetHeadlessTabSocket();
     DCHECK(headless_tab_socket);
 
-    headless_tab_socket->SendMessageToTab("Hello");
+    headless_tab_socket->SendMessageToTab("One");
+    headless_tab_socket->SendMessageToTab("Two");
+    headless_tab_socket->SendMessageToTab("Three");
     headless_tab_socket->SetListener(this);
   }
 
   void OnMessageFromTab(const std::string& message) override {
-    EXPECT_EQ("Embedder sent us: Hello", message);
-    FinishAsynchronousTest();
+    messages_.push_back(message);
+    if (messages_.size() == 3u) {
+      EXPECT_THAT(messages_,
+                  ElementsAre("Embedder sent us: One", "Embedder sent us: Two",
+                              "Embedder sent us: Three"));
+      FinishAsynchronousTest();
+    }
   }
 
   HeadlessWebContents::Builder::TabSocketType GetTabSocketType() override {
     return HeadlessWebContents::Builder::TabSocketType::MAIN_WORLD;
   }
+
+ private:
+  std::vector<std::string> messages_;
 };
 
 HEADLESS_ASYNC_DEVTOOLED_TEST_F(HeadlessMainWorldTabSocketTest);
 
 class HeadlessMainWorldTabSocketNotThereTest
     : public HeadlessAsyncDevTooledBrowserTest,
       public HeadlessTabSocket::Listener {
  public:
   void SetUp() override {
     options()->mojo_service_names.insert("headless::TabSocket");
@@ skipping 187 matching lines @@
     return HeadlessWebContents::Builder::TabSocketType::MAIN_WORLD;
   }
 
  private:
   std::string main_frame_id_;
 };
 
 HEADLESS_ASYNC_DEVTOOLED_TEST_F(HeadlessIsolatedWorldTabSocketNotThereTest);
 
 }  // namespace headless