| 165 | } |
| 166 | |
| 167 | void InitializeRendezvous(GrpcServer* grpc_server, ServerDef* server_def, |
| 168 | RemoteRendezvous* remote_rendezvous) { |
| 169 | int rendezvous_id = 0; |
| 170 | auto session_name = tensorflow::strings::StrCat("test_", rendezvous_id); |
| 171 | TF_EXPECT_OK(grpc_server->worker_env()->session_mgr->CreateSession( |
| 172 | session_name, *server_def, true)); |
| 173 | |
| 174 | std::shared_ptr<tensorflow::WorkerSession> worker_session; |
| 175 | TF_EXPECT_OK(grpc_server->worker_env()->session_mgr->WorkerSessionForSession( |
| 176 | session_name, &worker_session)); |
| 177 | |
| 178 | TF_EXPECT_OK(remote_rendezvous->Initialize(worker_session.get())); |
| 179 | } |
| 180 | |
| 181 | TEST_F(NetworksTest, TestStartServer) { |
| 182 | auto* rendezvous_builder = TF_NewRemoteRendezvousBuilder( |
no test coverage detected