Class MockGenerativeServiceConnection (2.30.0-rc)

A class to mock GenerativeServiceConnection.

Application developers may want to test their code with simulated responses, including errors, from an object of type GenerativeServiceClient. To do so, construct an object of type GenerativeServiceClient 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
Type Description
Options

virtual GenerateContent(google::ai::generativelanguage::v1::GenerateContentRequest const &)

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

Parameter
Name Description
request google::ai::generativelanguage::v1::GenerateContentRequest const &
Returns
Type Description
StatusOr< google::ai::generativelanguage::v1::GenerateContentResponse >

virtual StreamGenerateContent(google::ai::generativelanguage::v1::GenerateContentRequest const &)

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

Parameter
Name Description
request google::ai::generativelanguage::v1::GenerateContentRequest const &
Returns
Type Description
StreamRange< google::ai::generativelanguage::v1::GenerateContentResponse >

virtual EmbedContent(google::ai::generativelanguage::v1::EmbedContentRequest const &)

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

Parameter
Name Description
request google::ai::generativelanguage::v1::EmbedContentRequest const &
Returns
Type Description
StatusOr< google::ai::generativelanguage::v1::EmbedContentResponse >

virtual BatchEmbedContents(google::ai::generativelanguage::v1::BatchEmbedContentsRequest const &)

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

Parameter
Name Description
request google::ai::generativelanguage::v1::BatchEmbedContentsRequest const &
Returns
Type Description
StatusOr< google::ai::generativelanguage::v1::BatchEmbedContentsResponse >

virtual CountTokens(google::ai::generativelanguage::v1::CountTokensRequest const &)

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

Parameter
Name Description
request google::ai::generativelanguage::v1::CountTokensRequest const &
Returns
Type Description
StatusOr< google::ai::generativelanguage::v1::CountTokensResponse >