Index: ppapi/tests/test_udp_socket_private.cc |
diff --git a/ppapi/tests/test_udp_socket_private.cc b/ppapi/tests/test_udp_socket_private.cc |
index 995b53a71a8519743b135016e533aaf23e239bea..1380135944ebee93c71c540fb62639006e4ada71 100644 |
--- a/ppapi/tests/test_udp_socket_private.cc |
+++ b/ppapi/tests/test_udp_socket_private.cc |
@@ -54,25 +54,21 @@ bool TestUDPSocketPrivate::Init() { |
} |
void TestUDPSocketPrivate::RunTests(const std::string& filter) { |
- RUN_TEST_FORCEASYNC_AND_NOT(Connect, filter); |
- RUN_TEST_FORCEASYNC_AND_NOT(ConnectFailure, filter); |
- RUN_TEST_FORCEASYNC_AND_NOT(Broadcast, filter); |
- RUN_TEST_FORCEASYNC_AND_NOT(SetSocketFeatureErrors, filter); |
+ RUN_CALLBACK_TEST(TestUDPSocketPrivate, Connect, filter); |
+ RUN_CALLBACK_TEST(TestUDPSocketPrivate, ConnectFailure, filter); |
+ RUN_CALLBACK_TEST(TestUDPSocketPrivate, Broadcast, filter); |
+ RUN_CALLBACK_TEST(TestUDPSocketPrivate, SetSocketFeatureErrors, filter); |
} |
std::string TestUDPSocketPrivate::GetLocalAddress( |
PP_NetAddress_Private* address) { |
pp::TCPSocketPrivate socket(instance_); |
- TestCompletionCallback callback(instance_->pp_instance(), force_async_); |
- int32_t rv = socket.Connect(host_.c_str(), port_, callback.GetCallback()); |
- if (force_async_ && rv != PP_OK_COMPLETIONPENDING) |
- return ReportError("PPB_TCPSocket_Private::Connect force_async", rv); |
- if (rv == PP_OK_COMPLETIONPENDING) |
- rv = callback.WaitForResult(); |
- if (rv != PP_OK) |
- return ReportError("PPB_TCPSocket_Private::Connect", rv); |
- if (!socket.GetLocalAddress(address)) |
- return "PPB_TCPSocket_Private::GetLocalAddress: Failed"; |
+ TestCompletionCallback callback(instance_->pp_instance(), callback_type()); |
+ callback.WaitForResult( |
+ socket.Connect(host_.c_str(), port_, callback.GetCallback())); |
+ CHECK_CALLBACK_BEHAVIOR(callback); |
+ ASSERT_EQ(PP_OK, callback.result()); |
+ ASSERT_TRUE(socket.GetLocalAddress(address)); |
socket.Disconnect(); |
PASS(); |
} |
@@ -94,14 +90,10 @@ std::string TestUDPSocketPrivate::SetBroadcastOptions( |
std::string TestUDPSocketPrivate::BindUDPSocket( |
pp::UDPSocketPrivate* socket, |
PP_NetAddress_Private* address) { |
- TestCompletionCallback callback(instance_->pp_instance(), force_async_); |
- int32_t rv = socket->Bind(address, callback.GetCallback()); |
- if (force_async_ && rv != PP_OK_COMPLETIONPENDING) |
- return ReportError("PPB_UDPSocket_Private::Bind force_async", rv); |
- if (rv == PP_OK_COMPLETIONPENDING) |
- rv = callback.WaitForResult(); |
- if (rv != PP_OK) |
- return ReportError("PPB_UDPSocket_Private::Bind", rv); |
+ TestCompletionCallback callback(instance_->pp_instance(), callback_type()); |
+ callback.WaitForResult(socket->Bind(address, callback.GetCallback())); |
+ CHECK_CALLBACK_BEHAVIOR(callback); |
+ ASSERT_EQ(PP_OK, callback.result()); |
PASS(); |
} |
@@ -130,16 +122,11 @@ std::string TestUDPSocketPrivate::LookupPortAndBindUDPSocket( |
std::string TestUDPSocketPrivate::BindUDPSocketFailure( |
pp::UDPSocketPrivate* socket, |
PP_NetAddress_Private *address) { |
- TestCompletionCallback callback(instance_->pp_instance(), force_async_); |
- int32_t rv = socket->Bind(address, callback.GetCallback()); |
- if (force_async_ && rv != PP_OK_COMPLETIONPENDING) |
- return ReportError("PPB_UDPSocket_Private::Bind force_async", rv); |
- if (rv == PP_OK_COMPLETIONPENDING) |
- rv = callback.WaitForResult(); |
- if (rv == PP_OK) |
- return ReportError("PPB_UDPSocket_Private::Bind", rv); |
- if (socket->GetBoundAddress(address)) |
- return "PPB_UDPSocket_Private::GetBoundAddress: Failed"; |
+ TestCompletionCallback callback(instance_->pp_instance(), callback_type()); |
+ callback.WaitForResult(socket->Bind(address, callback.GetCallback())); |
+ CHECK_CALLBACK_BEHAVIOR(callback); |
+ ASSERT_NE(PP_OK, callback.result()); |
+ ASSERT_FALSE(socket->GetBoundAddress(address)); |
PASS(); |
} |
@@ -148,14 +135,12 @@ std::string TestUDPSocketPrivate::ReadSocket(pp::UDPSocketPrivate* socket, |
size_t size, |
std::string* message) { |
std::vector<char> buffer(size); |
- TestCompletionCallback callback(instance_->pp_instance(), force_async_); |
- int32_t rv = socket->RecvFrom(&buffer[0], size, callback.GetCallback()); |
- if (force_async_ && rv != PP_OK_COMPLETIONPENDING) |
- return ReportError("PPB_UDPSocket_Private::RecvFrom force_async", rv); |
- if (rv == PP_OK_COMPLETIONPENDING) |
- rv = callback.WaitForResult(); |
- if (rv < 0 || size != static_cast<size_t>(rv)) |
- return ReportError("PPB_UDPSocket_Private::RecvFrom", rv); |
+ TestCompletionCallback callback(instance_->pp_instance(), callback_type()); |
+ callback.WaitForResult( |
+ socket->RecvFrom(&buffer[0], size, callback.GetCallback())); |
+ CHECK_CALLBACK_BEHAVIOR(callback); |
+ ASSERT_FALSE(callback.result() < 0); |
+ ASSERT_EQ(size, static_cast<size_t>(callback.result())); |
message->assign(buffer.begin(), buffer.end()); |
PASS(); |
} |
@@ -164,20 +149,16 @@ std::string TestUDPSocketPrivate::PassMessage(pp::UDPSocketPrivate* target, |
pp::UDPSocketPrivate* source, |
PP_NetAddress_Private* address, |
const std::string& message) { |
- TestCompletionCallback callback(instance_->pp_instance(), force_async_); |
+ TestCompletionCallback callback(instance_->pp_instance(), callback_type()); |
int32_t rv = source->SendTo(message.c_str(), message.size(), address, |
callback.GetCallback()); |
- if (force_async_ && rv != PP_OK_COMPLETIONPENDING) |
- return ReportError("PPB_UDPSocket_Private::SendTo force_async", rv); |
- |
std::string str; |
ASSERT_SUBTEST_SUCCESS(ReadSocket(target, address, message.size(), &str)); |
- if (rv == PP_OK_COMPLETIONPENDING) |
- rv = callback.WaitForResult(); |
- if (rv < 0 || message.size() != static_cast<size_t>(rv)) |
- return ReportError("PPB_UDPSocket_Private::SendTo", rv); |
- |
+ callback.WaitForResult(rv); |
+ CHECK_CALLBACK_BEHAVIOR(callback); |
+ ASSERT_FALSE(callback.result() < 0); |
+ ASSERT_EQ(message.size(), static_cast<size_t>(callback.result())); |
ASSERT_EQ(message, str); |
PASS(); |
} |