public:

TestEventListenerImpl(const string& dump_root) : dump_root(dump_root) {}

::grpc::Status SendEvents(

::grpc::ServerContext* context,

::grpc::ServerReaderWriter< ::tensorflow::EventReply,

::tensorflow::Event>* stream);

string dump_root;

public:

GrpcTestServerClientPair(const int server_port);

virtual ~GrpcTestServerClientPair() {}

// Keep sending requests to the test server until the first success.

// This is necessary because the server may take a certain amount of time

// to start up and become responsive.

//

// Returns: A boolean indicating whether a successful response is obtained

// within the limit of maximum number of attempts.

bool PollTillFirstRequestSucceeds();

string dump_root;

int server_port;

string test_server_url;

private:

std::unique_ptr<Tensor> prep_tensor_;

const int kMaxAttempts = 100;

const int kSleepDurationMicros = 100 * 1000;