Beispiel #1
0
def test_client_withDEFAULT_CLIENT_INFO():
    client_info = gapic_v1.client_info.ClientInfo()

    with mock.patch.object(transports.CampaignBudgetServiceTransport,
                           '_prep_wrapped_messages') as prep:
        client = CampaignBudgetServiceClient(
            credentials=ga_credentials.AnonymousCredentials(),
            client_info=client_info,
        )
        prep.assert_called_once_with(client_info)

    with mock.patch.object(transports.CampaignBudgetServiceTransport,
                           '_prep_wrapped_messages') as prep:
        transport_class = CampaignBudgetServiceClient.get_transport_class()
        transport = transport_class(
            credentials=ga_credentials.AnonymousCredentials(),
            client_info=client_info,
        )
        prep.assert_called_once_with(client_info)
Beispiel #2
0
def test_campaign_budget_service_client_get_transport_class():
    transport = CampaignBudgetServiceClient.get_transport_class()
    assert transport == transports.CampaignBudgetServiceGrpcTransport

    transport = CampaignBudgetServiceClient.get_transport_class("grpc")
    assert transport == transports.CampaignBudgetServiceGrpcTransport