diff options
Diffstat (limited to 'tensorflow/core/distributed_runtime/rpc/grpc_channel_test.cc')
| -rw-r--r-- | tensorflow/core/distributed_runtime/rpc/grpc_channel_test.cc | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/tensorflow/core/distributed_runtime/rpc/grpc_channel_test.cc b/tensorflow/core/distributed_runtime/rpc/grpc_channel_test.cc index a17acc85b3..f07a5a0974 100644 --- a/tensorflow/core/distributed_runtime/rpc/grpc_channel_test.cc +++ b/tensorflow/core/distributed_runtime/rpc/grpc_channel_test.cc @@ -150,10 +150,15 @@ TEST(GrpcChannelTest, NewHostPortGrpcChannelValidation) { EXPECT_TRUE(NewHostPortGrpcChannel("127.0.0.1:2222", &mock_ptr).ok()); EXPECT_TRUE(NewHostPortGrpcChannel("example.com:2222", &mock_ptr).ok()); EXPECT_TRUE(NewHostPortGrpcChannel("fqdn.example.com.:2222", &mock_ptr).ok()); + EXPECT_TRUE(NewHostPortGrpcChannel("[2002:a9c:258e::]:2222", &mock_ptr).ok()); + EXPECT_TRUE(NewHostPortGrpcChannel("[::]:2222", &mock_ptr).ok()); EXPECT_FALSE(NewHostPortGrpcChannel("example.com/abc:2222", &mock_ptr).ok()); EXPECT_FALSE(NewHostPortGrpcChannel("127.0.0.1:2222/", &mock_ptr).ok()); EXPECT_FALSE(NewHostPortGrpcChannel("example.com/abc:", &mock_ptr).ok()); + EXPECT_FALSE(NewHostPortGrpcChannel("[::]/:2222", &mock_ptr).ok()); + EXPECT_FALSE(NewHostPortGrpcChannel("[::]:2222/", &mock_ptr).ok()); + EXPECT_FALSE(NewHostPortGrpcChannel("[::]:", &mock_ptr).ok()); } } // namespace tensorflow |