Index: mojo/services/network/udp_socket_apptest.cc |
diff --git a/mojo/services/network/udp_socket_apptest.cc b/mojo/services/network/udp_socket_apptest.cc |
index 62f3cdd0c525321cb214dad4fce919ef4b80cf72..f7fe6c32b58902a72c31a7e69305955590e17730 100644 |
--- a/mojo/services/network/udp_socket_apptest.cc |
+++ b/mojo/services/network/udp_socket_apptest.cc |
@@ -10,9 +10,9 @@ |
#include "mojo/services/network/udp_socket_wrapper.h" |
#include "net/base/net_errors.h" |
#include "testing/gtest/include/gtest/gtest.h" |
-#include "third_party/mojo/src/mojo/public/cpp/application/application_connection.h" |
#include "third_party/mojo/src/mojo/public/cpp/application/application_impl.h" |
#include "third_party/mojo/src/mojo/public/cpp/application/application_test_base.h" |
+#include "third_party/mojo/src/mojo/public/cpp/application/connect.h" |
#include "third_party/mojo/src/mojo/public/cpp/bindings/callback.h" |
namespace mojo { |
@@ -322,9 +322,8 @@ class UDPSocketAppTest : public test::ApplicationTestBase { |
void SetUp() override { |
ApplicationTestBase::SetUp(); |
- ApplicationConnection* connection = |
- application_impl()->ConnectToApplication("mojo:network_service"); |
- connection->ConnectToService(&network_service_); |
+ ConnectToService(application_impl()->shell(), "mojo:network_service", |
+ GetProxy(&network_service_)); |
network_service_->CreateUDPSocket(GetProxy(&socket_)); |
} |