| 27 | #define IsSameAddrSp DeviceNameUtils::IsSameAddressSpace |
| 28 | |
| 29 | TEST(GrpcChannelTest, IsSameAddressSpace) { |
| 30 | // Same. |
| 31 | EXPECT_TRUE(IsSameAddrSp("/job:mnist/replica:10/task:10/cpu:0", |
| 32 | "/job:mnist/replica:10/task:10/cpu:1")); |
| 33 | EXPECT_TRUE(IsSameAddrSp("/job:mnist/replica:10/task:10/cpu:0", |
| 34 | "/job:mnist/replica:10/task:10/device:GPU:2")); |
| 35 | EXPECT_TRUE(IsSameAddrSp("/job:mnist/replica:10/task:10", |
| 36 | "/job:mnist/replica:10/task:10/device:GPU:2")); |
| 37 | EXPECT_TRUE(IsSameAddrSp("/job:mnist/replica:10/task:10/cpu:1", |
| 38 | "/job:mnist/replica:10/task:10")); |
| 39 | |
| 40 | // Different. |
| 41 | EXPECT_FALSE(IsSameAddrSp("/job:mnist/replica:10/task:9/cpu:0", |
| 42 | "/job:mnist/replica:10/task:10/cpu:0")); |
| 43 | EXPECT_FALSE(IsSameAddrSp("/job:mnist/replica:9/task:10/cpu:0", |
| 44 | "/job:mnist/replica:10/task:10/cpu:0")); |
| 45 | EXPECT_FALSE(IsSameAddrSp("/job:MNIST/replica:10/task:10/cpu:0", |
| 46 | "/job:mnist/replica:10/task:10/cpu:0")); |
| 47 | |
| 48 | // Invalid names. |
| 49 | EXPECT_FALSE(IsSameAddrSp("random_invalid_target", "random_invalid_target")); |
| 50 | EXPECT_FALSE(IsSameAddrSp("/job:/replica:10/task:10/cpu:0", |
| 51 | "/job:/replica:10/task:10/cpu:1")); |
| 52 | EXPECT_FALSE(IsSameAddrSp("/job:mnist/replica:xx/task:10/cpu:0", |
| 53 | "/job:mnist/replica:xx/task:10/cpu:1")); |
| 54 | EXPECT_FALSE(IsSameAddrSp("/job:mnist/replica:10/task:yy/cpu:0", |
| 55 | "/job:mnist/replica:10/task:yy/cpu:1")); |
| 56 | } |
| 57 | |
| 58 | TEST(GrpcChannelTest, HostPorts) { |
| 59 | GrpcChannelSpec spec; |
nothing calls this directly
no test coverage detected