Pepper WebSocket API: Implement new design Chrome IPC

ppapi/proxy/websocket_resource_unittest.cc

Index: ppapi/proxy/websocket_resource_unittest.cc
diff --git a/ppapi/proxy/websocket_resource_unittest.cc b/ppapi/proxy/websocket_resource_unittest.cc
new file mode 100644
index 0000000000000000000000000000000000000000..61e892520fee19687e7c136a9b7127a4a289796f
--- /dev/null
+++ b/ppapi/proxy/websocket_resource_unittest.cc
@@ -0,0 +1,168 @@
+// Copyright (c) 2012 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 "base/memory/ref_counted.h"
+#include "base/message_loop.h"
+#include "ppapi/c/pp_errors.h"
+#include "ppapi/c/ppb_websocket.h"
+#include "ppapi/c/ppb_var.h"
+#include "ppapi/proxy/websocket_resource.h"
+#include "ppapi/proxy/ppapi_messages.h"
+#include "ppapi/proxy/ppapi_proxy_test.h"
+#include "ppapi/shared_impl/ppb_var_shared.h"
+#include "ppapi/shared_impl/scoped_pp_resource.h"
+#include "ppapi/shared_impl/scoped_pp_var.h"
+#include "ppapi/shared_impl/tracked_callback.h"
+#include "ppapi/shared_impl/var.h"
+#include "ppapi/thunk/thunk.h"
+
+namespace ppapi {
+namespace proxy {
+
+namespace {
+
+typedef PluginProxyTest WebSocketResourceTest;
+
+bool g_callback_called;
+int32_t g_callback_result;
+const PPB_Var* ppb_var_ = NULL;
+
+void Callback(void* user_data, int32_t result) {
+  g_callback_called = true;
+  g_callback_result = result;
+}
+
+PP_CompletionCallback MakeCallback() {
+  g_callback_called = false;
+  g_callback_result = PP_OK;
+  return PP_MakeCompletionCallback(Callback, NULL);
+}
+
+PP_Var MakeStringVar(const std::string& string) {
+  if (!ppb_var_)
+    ppb_var_ = ppapi::PPB_Var_Shared::GetVarInterface1_1();
+  return ppb_var_->VarFromUtf8(string.c_str(), string.length());
+}
+
+}  // namespace
+
+
+// Does a test of Connect().
+TEST_F(WebSocketResourceTest, Connect) {
+  const PPB_WebSocket_1_0* websocket_iface =
+      thunk::GetPPB_WebSocket_1_0_Thunk();
+
+  std::string url("ws://ws.google.com");
+  std::string protocol0("x-foo");
+  std::string protocol1("x-bar");
+  PP_Var url_var = MakeStringVar(url);
+  PP_Var protocols[] = { MakeStringVar(protocol0), MakeStringVar(protocol1) };
+
+  ScopedPPResource res(ScopedPPResource::PassRef(),
+                       websocket_iface->Create(pp_instance()));
+
+  int32_t result =
+      websocket_iface->Connect(res, url_var, protocols, 2, MakeCallback());
+  ASSERT_EQ(PP_OK_COMPLETIONPENDING, result);
+
+  // Should be sent a "Connect" message.
+  ResourceMessageCallParams params;
+  IPC::Message msg;
+  ASSERT_TRUE(sink().GetFirstResourceCallMatching(
+      PpapiHostMsg_WebSocket_Connect::ID, &params, &msg));
+  PpapiHostMsg_WebSocket_Connect::Schema::Param p;
+  PpapiHostMsg_WebSocket_Connect::Read(&msg, &p);
+  EXPECT_EQ(url, p.a);
+  EXPECT_EQ(protocol0, p.b[0]);
+  EXPECT_EQ(protocol1, p.b[1]);
+
+  // Synthesize a response.
+  ResourceMessageReplyParams reply_params(params.pp_resource(),
+                                          params.sequence());
+  reply_params.set_result(PP_OK);
+  ASSERT_TRUE(plugin_dispatcher()->OnMessageReceived(
+      PpapiPluginMsg_ResourceReply(reply_params,
+          PpapiPluginMsg_WebSocket_ConnectReply(url, protocol1))));
+
+  EXPECT_EQ(PP_OK, g_callback_result);
+  EXPECT_EQ(true, g_callback_called);
+}
+
+// Does a test for unsolicited replies.
+TEST_F(WebSocketResourceTest, UnsolicitedReplies) {
+  const PPB_WebSocket_1_0* websocket_iface =
+      thunk::GetPPB_WebSocket_1_0_Thunk();
+
+  ScopedPPResource res(ScopedPPResource::PassRef(),
+                       websocket_iface->Create(pp_instance()));
+
+  // Check if BufferedAmountReply is handled.
+  ResourceMessageReplyParams reply_params(res, 0);
+  reply_params.set_result(PP_OK);
+  ASSERT_TRUE(plugin_dispatcher()->OnMessageReceived(
+      PpapiPluginMsg_ResourceReply(
+          reply_params,
+          PpapiPluginMsg_WebSocket_BufferedAmountReply(19760227u))));
+
+  uint64_t amount = websocket_iface->GetBufferedAmount(res);
+  EXPECT_EQ(19760227u, amount);
+
+  // Check if StateReply is handled.
+  ASSERT_TRUE(plugin_dispatcher()->OnMessageReceived(
+      PpapiPluginMsg_ResourceReply(
+          reply_params,
+          PpapiPluginMsg_WebSocket_StateReply(
+              static_cast<int32_t>(PP_WEBSOCKETREADYSTATE_CLOSING)))));
+
+  PP_WebSocketReadyState state = websocket_iface->GetReadyState(res);
+  EXPECT_EQ(PP_WEBSOCKETREADYSTATE_CLOSING, state);
+}
+
+TEST_F(WebSocketResourceTest, MessageError) {
+  const PPB_WebSocket_1_0* websocket_iface =
+      thunk::GetPPB_WebSocket_1_0_Thunk();
+
+  std::string url("ws://ws.google.com");
+  PP_Var url_var = MakeStringVar(url);
+
+  ScopedPPResource res(ScopedPPResource::PassRef(),
+                       websocket_iface->Create(pp_instance()));
+
+  // Establish the connection virtually.
+  int32_t result =
+      websocket_iface->Connect(res, url_var, NULL, 0, MakeCallback());
+  ASSERT_EQ(PP_OK_COMPLETIONPENDING, result);
+
+  ResourceMessageCallParams params;
+  IPC::Message msg;
+  ASSERT_TRUE(sink().GetFirstResourceCallMatching(
+      PpapiHostMsg_WebSocket_Connect::ID, &params, &msg));
+
+  ResourceMessageReplyParams connect_reply_params(params.pp_resource(),
+                                                  params.sequence());
+  connect_reply_params.set_result(PP_OK);
+  ASSERT_TRUE(plugin_dispatcher()->OnMessageReceived(
+      PpapiPluginMsg_ResourceReply(connect_reply_params,
+          PpapiPluginMsg_WebSocket_ConnectReply(url, std::string()))));
+
+  EXPECT_EQ(PP_OK, g_callback_result);
+  EXPECT_EQ(true, g_callback_called);
+
+  PP_Var message;
+  result = websocket_iface->ReceiveMessage(res, &message, MakeCallback());
+  EXPECT_EQ(false, g_callback_called);
+
+  // Synthesize a WebSocket_ErrorReply message.
+  ResourceMessageReplyParams error_reply_params(res, 0);
+  error_reply_params.set_result(PP_OK);
+  ASSERT_TRUE(plugin_dispatcher()->OnMessageReceived(
+      PpapiPluginMsg_ResourceReply(error_reply_params,
+          PpapiPluginMsg_WebSocket_ErrorReply())));
+
+  EXPECT_EQ(PP_ERROR_FAILED, g_callback_result);
+  EXPECT_EQ(true, g_callback_called);
+}
+
+}  // namespace proxy
+}  // namespace ppapi