1313// limitations under the License.
1414
1515#include " generator/integration_tests/golden/golden_thing_admin_connection.h"
16+ #include " google/cloud/grpc_options.h"
1617#include " google/cloud/polling_policy.h"
1718#include " google/cloud/testing_util/async_sequencer.h"
1819#include " google/cloud/testing_util/is_proto_equal.h"
1920#include " google/cloud/testing_util/scoped_log.h"
2021#include " google/cloud/testing_util/status_matchers.h"
2122#include " generator/integration_tests/golden/golden_thing_admin_options.h"
23+ #include " generator/integration_tests/golden/internal/golden_thing_admin_connection_impl.h"
24+ #include " generator/integration_tests/golden/internal/golden_thing_admin_option_defaults.h"
2225#include " generator/integration_tests/golden/mocks/mock_golden_thing_admin_stub.h"
2326#include < google/protobuf/text_format.h>
2427#include < gmock/gmock.h>
@@ -30,6 +33,9 @@ namespace golden {
3033GOOGLE_CLOUD_CPP_INLINE_NAMESPACE_BEGIN
3134namespace {
3235
36+ using ::google::cloud::golden_internal::GoldenThingAdminConnectionImpl;
37+ using ::google::cloud::golden_internal::GoldenThingAdminDefaultOptions;
38+ using ::google::cloud::golden_internal::GoldenThingAdminStub;
3339using ::google::cloud::golden_internal::MockGoldenThingAdminStub;
3440using ::google::cloud::testing_util::AsyncSequencer;
3541using ::google::cloud::testing_util::IsProtoEqual;
@@ -43,7 +49,7 @@ using ::testing::HasSubstr;
4349using ::testing::Return;
4450
4551std::shared_ptr<golden::GoldenThingAdminConnection> CreateTestingConnection (
46- std::shared_ptr<golden_internal:: GoldenThingAdminStub> mock) {
52+ std::shared_ptr<GoldenThingAdminStub> mock) {
4753 golden::GoldenThingAdminLimitedErrorCountRetryPolicy retry (
4854 /* maximum_failures=*/ 2 );
4955 ExponentialBackoffPolicy backoff (
@@ -53,12 +59,14 @@ std::shared_ptr<golden::GoldenThingAdminConnection> CreateTestingConnection(
5359 GenericPollingPolicy<golden::GoldenThingAdminLimitedErrorCountRetryPolicy,
5460 ExponentialBackoffPolicy>
5561 polling (retry, backoff);
56- Options options;
57- options.set <golden::GoldenThingAdminRetryPolicyOption>(retry.clone ());
58- options.set <golden::GoldenThingAdminBackoffPolicyOption>(backoff.clone ());
59- options.set <golden::GoldenThingAdminPollingPolicyOption>(polling.clone ());
60- return golden_internal::MakeGoldenThingAdminConnection (std::move (mock),
61- std::move (options));
62+ auto options = GoldenThingAdminDefaultOptions (
63+ Options{}
64+ .set <golden::GoldenThingAdminRetryPolicyOption>(retry.clone ())
65+ .set <golden::GoldenThingAdminBackoffPolicyOption>(backoff.clone ())
66+ .set <golden::GoldenThingAdminPollingPolicyOption>(polling.clone ()));
67+ auto background = internal::MakeBackgroundThreadsFactory (options)();
68+ return std::make_shared<GoldenThingAdminConnectionImpl>(
69+ std::move (background), std::move (mock), std::move (options));
6270}
6371
6472google::longrunning::Operation CreateStartingOperation () {
0 commit comments