headless: Fix handling of replies without any parameters

headless/lib/headless_devtools_client_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 "content/public/test/browser_test.h"
 #include "headless/public/domains/page.h"
 #include "headless/public/domains/runtime.h"
 #include "headless/public/headless_browser.h"
@@ skipping 92 matching lines @@
     int value;
     EXPECT_TRUE(result->GetResult()->HasValue());
     EXPECT_TRUE(result->GetResult()->GetValue()->GetAsInteger(&value));
     EXPECT_EQ(168, value);
     FinishAsynchronousTest();
   }
 };
 
 DEVTOOLS_CLIENT_TEST_F(HeadlessDevToolsClientEvalTest);
 
+class HeadlessDevToolsClientCallbackTest : public HeadlessDevToolsClientTest {
+ public:
+  HeadlessDevToolsClientCallbackTest() : first_result_received_(false) {}
+
+  void RunDevToolsClientTest() override {
+    // Null callback without parameters.
+    devtools_client_->GetRuntime()->Run();
+    // Null callback with parameters.
+    devtools_client_->GetRuntime()->Evaluate("true");
+    // Non-null callback without parameters.
+    devtools_client_->GetRuntime()->Disable(
+        base::Bind(&HeadlessDevToolsClientCallbackTest::OnFirstResult,
+                   base::Unretained(this)));
+    // Non-null callback with parameters.
+    devtools_client_->GetRuntime()->Evaluate(
+        "true", base::Bind(&HeadlessDevToolsClientCallbackTest::OnSecondResult,
+                           base::Unretained(this)));
+  }
+
+  void OnFirstResult() {
+    EXPECT_FALSE(first_result_received_);
+    first_result_received_ = true;
+  }
+
+  void OnSecondResult(std::unique_ptr<runtime::EvaluateResult> result) {
+    EXPECT_TRUE(first_result_received_);
+    FinishAsynchronousTest();
+  }
+
+ private:
+  bool first_result_received_;
+};
+
+DEVTOOLS_CLIENT_TEST_F(HeadlessDevToolsClientCallbackTest);
+
 }  // namespace headless