def test_to_pb(self):
from gcloud.bigtable._generated import bigtable_data_pb2 as data_pb2
from gcloud.bigtable.row import TimestampRange
range_ = TimestampRange()
row_filter = self._makeOne(range_)
pb_val = row_filter.to_pb()
expected_pb = data_pb2.RowFilter(
timestamp_range_filter=data_pb2.TimestampRange())
self.assertEqual(pb_val, expected_pb)
def to_pb(self):
"""Converts the :class:`TimestampRange` to a protobuf.
:rtype: :class:`.data_pb2.TimestampRange`
:returns: The converted current object.
"""
timestamp_range_kwargs = {}
if self.start is not None:
timestamp_range_kwargs['start_timestamp_micros'] = (
_microseconds_from_datetime(self.start))
if self.end is not None:
timestamp_range_kwargs['end_timestamp_micros'] = (
_microseconds_from_datetime(self.end))
return data_pb2.TimestampRange(**timestamp_range_kwargs)
def _to_pb_helper(self, start_micros=None, end_micros=None):
import datetime
from gcloud._helpers import _EPOCH
from gcloud.bigtable._generated import bigtable_data_pb2 as data_pb2
pb_kwargs = {}
start = None
if start_micros is not None:
start = _EPOCH + datetime.timedelta(microseconds=start_micros)
pb_kwargs['start_timestamp_micros'] = start_micros
end = None
if end_micros is not None:
end = _EPOCH + datetime.timedelta(microseconds=end_micros)
pb_kwargs['end_timestamp_micros'] = end_micros
time_range = self._makeOne(start=start, end=end)
expected_pb = data_pb2.TimestampRange(**pb_kwargs)
self.assertEqual(time_range.to_pb(), expected_pb)
