| 31 | } // namespace |
| 32 | |
| 33 | TEST(RPCFactoryRegistryTest, TestBasic) { |
| 34 | EXPECT_EQ(RPCFactoryRegistry::Global()->Get("NON-EXISTENT"), nullptr); |
| 35 | auto factory1 = RPCFactoryRegistry::Global()->Get("TEST FACTORY 1"); |
| 36 | EXPECT_NE(factory1, nullptr); |
| 37 | auto factory2 = RPCFactoryRegistry::Global()->Get("TEST FACTORY 2"); |
| 38 | EXPECT_NE(factory2, nullptr); |
| 39 | } |
| 40 | |
| 41 | } // namespace tensorflow |