def test_get_messages():
answer = protobuf_helpers.get_messages(date_pb2)
# Ensure that Date was exported properly.
assert answer['Date'] is date_pb2.Date
# Ensure that no non-Message objects were exported.
for value in answer.values():
assert issubclass(value, message.Message)
from google.api_core.protobuf_helpers import get_messages
from google.cloud.oslogin_v1.proto import common_pb2
from google.cloud.oslogin_v1.proto import oslogin_pb2
_shared_modules = [
http_pb2,
descriptor_pb2,
empty_pb2,
field_mask_pb2,
]
_local_modules = [
common_pb2,
oslogin_pb2,
]
names = []
for module in _shared_modules:
for name, message in get_messages(module).items():
setattr(sys.modules[__name__], name, message)
names.append(name)
for module in _local_modules:
for name, message in get_messages(module).items():
message.__module__ = 'google.cloud.oslogin_v1.types'
setattr(sys.modules[__name__], name, message)
names.append(name)
__all__ = tuple(sorted(names))
import sys
from google.api_core.protobuf_helpers import get_messages
from google.cloud.videointelligence_v1p3beta1.proto import video_intelligence_pb2
from google.longrunning import operations_pb2
from google.protobuf import any_pb2
from google.protobuf import duration_pb2
from google.protobuf import timestamp_pb2
from google.rpc import status_pb2
_shared_modules = [operations_pb2, any_pb2, duration_pb2, timestamp_pb2, status_pb2]
_local_modules = [video_intelligence_pb2]
names = []
for module in _shared_modules: # pragma: NO COVER
for name, message in get_messages(module).items():
setattr(sys.modules[__name__], name, message)
names.append(name)
for module in _local_modules:
for name, message in get_messages(module).items():
message.__module__ = "google.cloud.videointelligence_v1p3beta1.types"
setattr(sys.modules[__name__], name, message)
names.append(name)
__all__ = tuple(sorted(names))
'max_request_batch_size', 'max_request_batch_latency',
'max_lease_duration'],
)
FlowControl.__new__.__defaults__ = (
psutil.virtual_memory().total * 0.2, # max_bytes: 20% of total RAM
float('inf'), # max_messages: no limit
0.8, # resume_threshold: 80%
100, # max_requests: 100
100, # max_request_batch_size: 100
0.01, # max_request_batch_latency: 0.01s
2 * 60 * 60, # max_lease_duration: 2 hours.
)
names = ['BatchSettings', 'FlowControl']
for name, message in get_messages(pubsub_pb2).items():
message.__module__ = 'google.cloud.pubsub_v1.types'
setattr(sys.modules[__name__], name, message)
names.append(name)
for module in (
http_pb2,
pubsub_pb2,
iam_policy_pb2,
policy_pb2,
audit_data_pb2,
descriptor_pb2,
duration_pb2,
empty_pb2,
field_mask_pb2,
[
'max_bytes', 'max_messages', 'resume_threshold', 'max_requests',
'max_request_batch_size', 'max_request_batch_latency'
],
)
FlowControl.__new__.__defaults__ = (
psutil.virtual_memory().total * 0.2, # max_bytes: 20% of total RAM
float('inf'), # max_messages: no limit
0.8, # resume_threshold: 80%
100, # max_requests: 100
100, # max_request_batch_size: 100
0.01, # max_request_batch_latency: 0.01s
)
names = ['BatchSettings', 'FlowControl']
for name, message in get_messages(pubsub_pb2).items():
message.__module__ = 'google.cloud.pubsub_v1.types'
setattr(sys.modules[__name__], name, message)
names.append(name)
for module in (
http_pb2,
pubsub_pb2,
iam_policy_pb2,
policy_pb2,
audit_data_pb2,
descriptor_pb2,
duration_pb2,
empty_pb2,
field_mask_pb2,
timestamp_pb2,
