Index: third_party/mojo/src/mojo/public/cpp/application/tests/service_registry_unittest.cc |
diff --git a/third_party/mojo/src/mojo/public/cpp/application/tests/service_registry_unittest.cc b/third_party/mojo/src/mojo/public/cpp/application/tests/service_registry_unittest.cc |
index 283bf144034028e66ea2ecd1ccbad197333be5af..e0c5a9d3b9010863de890c1552a9a87148da6cd6 100644 |
--- a/third_party/mojo/src/mojo/public/cpp/application/tests/service_registry_unittest.cc |
+++ b/third_party/mojo/src/mojo/public/cpp/application/tests/service_registry_unittest.cc |
@@ -4,19 +4,19 @@ |
#include "mojo/public/cpp/application/lib/service_registry.h" |
-#include "mojo/public/cpp/application/lib/service_connector.h" |
+#include "mojo/public/cpp/application/service_connector.h" |
#include "testing/gtest/include/gtest/gtest.h" |
namespace mojo { |
namespace internal { |
namespace { |
-class TestConnector : public ServiceConnectorBase { |
+class TestConnector : public ServiceConnector { |
public: |
- TestConnector(const std::string& name, int* delete_count) |
- : ServiceConnectorBase(name), delete_count_(delete_count) {} |
+ explicit TestConnector(int* delete_count) : delete_count_(delete_count) {} |
~TestConnector() override { (*delete_count_)++; } |
- void ConnectToService(const std::string& name, |
+ void ConnectToService(ApplicationConnection* application_connection, |
+ const std::string& interface_name, |
ScopedMessagePipeHandle client_handle) override {} |
private: |
@@ -29,32 +29,37 @@ TEST(ServiceRegistryTest, Ownership) { |
// Destruction. |
{ |
ServiceRegistry registry; |
- registry.AddServiceConnector(new TestConnector("TC1", &delete_count)); |
+ registry.SetServiceConnectorForName(new TestConnector(&delete_count), |
+ "TC1"); |
} |
EXPECT_EQ(1, delete_count); |
// Removal. |
{ |
ServiceRegistry registry; |
- ServiceConnectorBase* c = new TestConnector("TC1", &delete_count); |
- registry.AddServiceConnector(c); |
- registry.RemoveServiceConnector(c); |
+ ServiceConnector* c = new TestConnector(&delete_count); |
+ registry.SetServiceConnectorForName(c, "TC1"); |
+ registry.RemoveServiceConnectorForName("TC1"); |
EXPECT_EQ(2, delete_count); |
} |
// Multiple. |
{ |
ServiceRegistry registry; |
- registry.AddServiceConnector(new TestConnector("TC1", &delete_count)); |
- registry.AddServiceConnector(new TestConnector("TC2", &delete_count)); |
+ registry.SetServiceConnectorForName(new TestConnector(&delete_count), |
+ "TC1"); |
+ registry.SetServiceConnectorForName(new TestConnector(&delete_count), |
+ "TC2"); |
} |
EXPECT_EQ(4, delete_count); |
// Re-addition. |
{ |
ServiceRegistry registry; |
- registry.AddServiceConnector(new TestConnector("TC1", &delete_count)); |
- registry.AddServiceConnector(new TestConnector("TC1", &delete_count)); |
+ registry.SetServiceConnectorForName(new TestConnector(&delete_count), |
+ "TC1"); |
+ registry.SetServiceConnectorForName(new TestConnector(&delete_count), |
+ "TC1"); |
EXPECT_EQ(5, delete_count); |
} |
EXPECT_EQ(6, delete_count); |