Class MockPredictionServiceConnection (2.21.0)

A class to mock PredictionServiceConnection.

Application developers may want to test their code with simulated responses, including errors, from an object of type PredictionServiceClient. To do so, construct an object of type PredictionServiceClient with an instance of this class. Then use the Google Test framework functions to program the behavior of this mock.

See Also

This example for how to test your application with GoogleTest. While the example showcases types from the BigQuery library, the underlying principles apply for any pair of *Client and *Connection.

Functions

virtual options()

This function is implemented using gMock's MOCK_METHOD(). Consult the gMock documentation to use this mock in your tests.

Returns
TypeDescription
Options

virtual Predict(google::cloud::aiplatform::v1::PredictRequest const &)

This function is implemented using gMock's MOCK_METHOD(). Consult the gMock documentation to use this mock in your tests.

Parameter
NameDescription
request google::cloud::aiplatform::v1::PredictRequest const &
Returns
TypeDescription
StatusOr< google::cloud::aiplatform::v1::PredictResponse >

virtual RawPredict(google::cloud::aiplatform::v1::RawPredictRequest const &)

This function is implemented using gMock's MOCK_METHOD(). Consult the gMock documentation to use this mock in your tests.

Parameter
NameDescription
request google::cloud::aiplatform::v1::RawPredictRequest const &
Returns
TypeDescription
StatusOr< google::api::HttpBody >

virtual DirectPredict(google::cloud::aiplatform::v1::DirectPredictRequest const &)

This function is implemented using gMock's MOCK_METHOD(). Consult the gMock documentation to use this mock in your tests.

Parameter
NameDescription
request google::cloud::aiplatform::v1::DirectPredictRequest const &
Returns
TypeDescription
StatusOr< google::cloud::aiplatform::v1::DirectPredictResponse >

virtual DirectRawPredict(google::cloud::aiplatform::v1::DirectRawPredictRequest const &)

This function is implemented using gMock's MOCK_METHOD(). Consult the gMock documentation to use this mock in your tests.

Parameter
NameDescription
request google::cloud::aiplatform::v1::DirectRawPredictRequest const &
Returns
TypeDescription
StatusOr< google::cloud::aiplatform::v1::DirectRawPredictResponse >

virtual AsyncStreamingPredict()

This function is implemented using gMock's MOCK_METHOD(). Consult the gMock documentation to use this mock in your tests.

Returns
TypeDescription
std::unique_ptr<::google::cloud::AsyncStreamingReadWriteRpc< google::cloud::aiplatform::v1::StreamingPredictRequest, google::cloud::aiplatform::v1::StreamingPredictResponse > >

virtual ServerStreamingPredict(google::cloud::aiplatform::v1::StreamingPredictRequest const &)

This function is implemented using gMock's MOCK_METHOD(). Consult the gMock documentation to use this mock in your tests.

Parameter
NameDescription
request google::cloud::aiplatform::v1::StreamingPredictRequest const &
Returns
TypeDescription
StreamRange< google::cloud::aiplatform::v1::StreamingPredictResponse >

virtual AsyncStreamingRawPredict()

This function is implemented using gMock's MOCK_METHOD(). Consult the gMock documentation to use this mock in your tests.

Returns
TypeDescription
std::unique_ptr<::google::cloud::AsyncStreamingReadWriteRpc< google::cloud::aiplatform::v1::StreamingRawPredictRequest, google::cloud::aiplatform::v1::StreamingRawPredictResponse > >

virtual Explain(google::cloud::aiplatform::v1::ExplainRequest const &)

This function is implemented using gMock's MOCK_METHOD(). Consult the gMock documentation to use this mock in your tests.

Parameter
NameDescription
request google::cloud::aiplatform::v1::ExplainRequest const &
Returns
TypeDescription
StatusOr< google::cloud::aiplatform::v1::ExplainResponse >

virtual StreamGenerateContent(google::cloud::aiplatform::v1::GenerateContentRequest const &)

This function is implemented using gMock's MOCK_METHOD(). Consult the gMock documentation to use this mock in your tests.

Parameter
NameDescription
request google::cloud::aiplatform::v1::GenerateContentRequest const &
Returns
TypeDescription
StreamRange< google::cloud::aiplatform::v1::GenerateContentResponse >