670 lines
23 KiB
Python
670 lines
23 KiB
Python
|
# Copyright 2015 Google Inc. All rights reserved.
|
||
|
|
#
|
||
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
||
|
|
# you may not use this file except in compliance with the License.
|
||
|
|
# You may obtain a copy of the License at
|
||
|
|
#
|
||
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
||
|
|
#
|
||
|
|
# Unless required by applicable law or agreed to in writing, software
|
||
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
||
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||
|
|
# See the License for the specific language governing permissions and
|
||
|
|
# limitations under the License.
|
||
|
|
|
||
|
|
|
||
|
|
import unittest2
|
||
|
|
|
||
|
|
|
||
|
|
class Test__timedelta_to_duration_pb(unittest2.TestCase):
|
||
|
|
|
||
|
|
def _callFUT(self, *args, **kwargs):
|
||
|
|
from gcloud.bigtable.column_family import _timedelta_to_duration_pb
|
||
|
|
return _timedelta_to_duration_pb(*args, **kwargs)
|
||
|
|
|
||
|
|
def test_it(self):
|
||
|
|
import datetime
|
||
|
|
from google.protobuf import duration_pb2
|
||
|
|
|
||
|
|
seconds = microseconds = 1
|
||
|
|
timedelta_val = datetime.timedelta(seconds=seconds,
|
||
|
|
microseconds=microseconds)
|
||
|
|
result = self._callFUT(timedelta_val)
|
||
|
|
self.assertTrue(isinstance(result, duration_pb2.Duration))
|
||
|
|
self.assertEqual(result.seconds, seconds)
|
||
|
|
self.assertEqual(result.nanos, 1000 * microseconds)
|
||
|
|
|
||
|
|
def test_with_negative_microseconds(self):
|
||
|
|
import datetime
|
||
|
|
from google.protobuf import duration_pb2
|
||
|
|
|
||
|
|
seconds = 1
|
||
|
|
microseconds = -5
|
||
|
|
timedelta_val = datetime.timedelta(seconds=seconds,
|
||
|
|
microseconds=microseconds)
|
||
|
|
result = self._callFUT(timedelta_val)
|
||
|
|
self.assertTrue(isinstance(result, duration_pb2.Duration))
|
||
|
|
self.assertEqual(result.seconds, seconds - 1)
|
||
|
|
self.assertEqual(result.nanos, 10**9 + 1000 * microseconds)
|
||
|
|
|
||
|
|
def test_with_negative_seconds(self):
|
||
|
|
import datetime
|
||
|
|
from google.protobuf import duration_pb2
|
||
|
|
|
||
|
|
seconds = -1
|
||
|
|
microseconds = 5
|
||
|
|
timedelta_val = datetime.timedelta(seconds=seconds,
|
||
|
|
microseconds=microseconds)
|
||
|
|
result = self._callFUT(timedelta_val)
|
||
|
|
self.assertTrue(isinstance(result, duration_pb2.Duration))
|
||
|
|
self.assertEqual(result.seconds, seconds + 1)
|
||
|
|
self.assertEqual(result.nanos, -(10**9 - 1000 * microseconds))
|
||
|
|
|
||
|
|
|
||
|
|
class Test__duration_pb_to_timedelta(unittest2.TestCase):
|
||
|
|
|
||
|
|
def _callFUT(self, *args, **kwargs):
|
||
|
|
from gcloud.bigtable.column_family import _duration_pb_to_timedelta
|
||
|
|
return _duration_pb_to_timedelta(*args, **kwargs)
|
||
|
|
|
||
|
|
def test_it(self):
|
||
|
|
import datetime
|
||
|
|
from google.protobuf import duration_pb2
|
||
|
|
|
||
|
|
seconds = microseconds = 1
|
||
|
|
duration_pb = duration_pb2.Duration(seconds=seconds,
|
||
|
|
nanos=1000 * microseconds)
|
||
|
|
timedelta_val = datetime.timedelta(seconds=seconds,
|
||
|
|
microseconds=microseconds)
|
||
|
|
result = self._callFUT(duration_pb)
|
||
|
|
self.assertTrue(isinstance(result, datetime.timedelta))
|
||
|
|
self.assertEqual(result, timedelta_val)
|
||
|
|
|
||
|
|
|
||
|
|
class TestMaxVersionsGCRule(unittest2.TestCase):
|
||
|
|
|
||
|
|
def _getTargetClass(self):
|
||
|
|
from gcloud.bigtable.column_family import MaxVersionsGCRule
|
||
|
|
return MaxVersionsGCRule
|
||
|
|
|
||
|
|
def _makeOne(self, *args, **kwargs):
|
||
|
|
return self._getTargetClass()(*args, **kwargs)
|
||
|
|
|
||
|
|
def test___eq__type_differ(self):
|
||
|
|
gc_rule1 = self._makeOne(10)
|
||
|
|
gc_rule2 = object()
|
||
|
|
self.assertNotEqual(gc_rule1, gc_rule2)
|
||
|
|
|
||
|
|
def test___eq__same_value(self):
|
||
|
|
gc_rule1 = self._makeOne(2)
|
||
|
|
gc_rule2 = self._makeOne(2)
|
||
|
|
self.assertEqual(gc_rule1, gc_rule2)
|
||
|
|
|
||
|
|
def test___ne__same_value(self):
|
||
|
|
gc_rule1 = self._makeOne(99)
|
||
|
|
gc_rule2 = self._makeOne(99)
|
||
|
|
comparison_val = (gc_rule1 != gc_rule2)
|
||
|
|
self.assertFalse(comparison_val)
|
||
|
|
|
||
|
|
def test_to_pb(self):
|
||
|
|
max_num_versions = 1337
|
||
|
|
gc_rule = self._makeOne(max_num_versions=max_num_versions)
|
||
|
|
pb_val = gc_rule.to_pb()
|
||
|
|
expected = _GcRulePB(max_num_versions=max_num_versions)
|
||
|
|
self.assertEqual(pb_val, expected)
|
||
|
|
|
||
|
|
|
||
|
|
class TestMaxAgeGCRule(unittest2.TestCase):
|
||
|
|
|
||
|
|
def _getTargetClass(self):
|
||
|
|
from gcloud.bigtable.column_family import MaxAgeGCRule
|
||
|
|
return MaxAgeGCRule
|
||
|
|
|
||
|
|
def _makeOne(self, *args, **kwargs):
|
||
|
|
return self._getTargetClass()(*args, **kwargs)
|
||
|
|
|
||
|
|
def test___eq__type_differ(self):
|
||
|
|
max_age = object()
|
||
|
|
gc_rule1 = self._makeOne(max_age=max_age)
|
||
|
|
gc_rule2 = object()
|
||
|
|
self.assertNotEqual(gc_rule1, gc_rule2)
|
||
|
|
|
||
|
|
def test___eq__same_value(self):
|
||
|
|
max_age = object()
|
||
|
|
gc_rule1 = self._makeOne(max_age=max_age)
|
||
|
|
gc_rule2 = self._makeOne(max_age=max_age)
|
||
|
|
self.assertEqual(gc_rule1, gc_rule2)
|
||
|
|
|
||
|
|
def test___ne__same_value(self):
|
||
|
|
max_age = object()
|
||
|
|
gc_rule1 = self._makeOne(max_age=max_age)
|
||
|
|
gc_rule2 = self._makeOne(max_age=max_age)
|
||
|
|
comparison_val = (gc_rule1 != gc_rule2)
|
||
|
|
self.assertFalse(comparison_val)
|
||
|
|
|
||
|
|
def test_to_pb(self):
|
||
|
|
import datetime
|
||
|
|
from google.protobuf import duration_pb2
|
||
|
|
|
||
|
|
max_age = datetime.timedelta(seconds=1)
|
||
|
|
duration = duration_pb2.Duration(seconds=1)
|
||
|
|
gc_rule = self._makeOne(max_age=max_age)
|
||
|
|
pb_val = gc_rule.to_pb()
|
||
|
|
self.assertEqual(pb_val, _GcRulePB(max_age=duration))
|
||
|
|
|
||
|
|
|
||
|
|
class TestGCRuleUnion(unittest2.TestCase):
|
||
|
|
|
||
|
|
def _getTargetClass(self):
|
||
|
|
from gcloud.bigtable.column_family import GCRuleUnion
|
||
|
|
return GCRuleUnion
|
||
|
|
|
||
|
|
def _makeOne(self, *args, **kwargs):
|
||
|
|
return self._getTargetClass()(*args, **kwargs)
|
||
|
|
|
||
|
|
def test_constructor(self):
|
||
|
|
rules = object()
|
||
|
|
rule_union = self._makeOne(rules)
|
||
|
|
self.assertTrue(rule_union.rules is rules)
|
||
|
|
|
||
|
|
def test___eq__(self):
|
||
|
|
rules = object()
|
||
|
|
gc_rule1 = self._makeOne(rules)
|
||
|
|
gc_rule2 = self._makeOne(rules)
|
||
|
|
self.assertEqual(gc_rule1, gc_rule2)
|
||
|
|
|
||
|
|
def test___eq__type_differ(self):
|
||
|
|
rules = object()
|
||
|
|
gc_rule1 = self._makeOne(rules)
|
||
|
|
gc_rule2 = object()
|
||
|
|
self.assertNotEqual(gc_rule1, gc_rule2)
|
||
|
|
|
||
|
|
def test___ne__same_value(self):
|
||
|
|
rules = object()
|
||
|
|
gc_rule1 = self._makeOne(rules)
|
||
|
|
gc_rule2 = self._makeOne(rules)
|
||
|
|
comparison_val = (gc_rule1 != gc_rule2)
|
||
|
|
self.assertFalse(comparison_val)
|
||
|
|
|
||
|
|
def test_to_pb(self):
|
||
|
|
import datetime
|
||
|
|
from google.protobuf import duration_pb2
|
||
|
|
from gcloud.bigtable.column_family import MaxAgeGCRule
|
||
|
|
from gcloud.bigtable.column_family import MaxVersionsGCRule
|
||
|
|
|
||
|
|
max_num_versions = 42
|
||
|
|
rule1 = MaxVersionsGCRule(max_num_versions)
|
||
|
|
pb_rule1 = _GcRulePB(max_num_versions=max_num_versions)
|
||
|
|
|
||
|
|
max_age = datetime.timedelta(seconds=1)
|
||
|
|
rule2 = MaxAgeGCRule(max_age)
|
||
|
|
pb_rule2 = _GcRulePB(
|
||
|
|
max_age=duration_pb2.Duration(seconds=1))
|
||
|
|
|
||
|
|
rule3 = self._makeOne(rules=[rule1, rule2])
|
||
|
|
pb_rule3 = _GcRulePB(
|
||
|
|
union=_GcRuleUnionPB(rules=[pb_rule1, pb_rule2]))
|
||
|
|
|
||
|
|
gc_rule_pb = rule3.to_pb()
|
||
|
|
self.assertEqual(gc_rule_pb, pb_rule3)
|
||
|
|
|
||
|
|
def test_to_pb_nested(self):
|
||
|
|
import datetime
|
||
|
|
from google.protobuf import duration_pb2
|
||
|
|
from gcloud.bigtable.column_family import MaxAgeGCRule
|
||
|
|
from gcloud.bigtable.column_family import MaxVersionsGCRule
|
||
|
|
|
||
|
|
max_num_versions1 = 42
|
||
|
|
rule1 = MaxVersionsGCRule(max_num_versions1)
|
||
|
|
pb_rule1 = _GcRulePB(max_num_versions=max_num_versions1)
|
||
|
|
|
||
|
|
max_age = datetime.timedelta(seconds=1)
|
||
|
|
rule2 = MaxAgeGCRule(max_age)
|
||
|
|
pb_rule2 = _GcRulePB(
|
||
|
|
max_age=duration_pb2.Duration(seconds=1))
|
||
|
|
|
||
|
|
rule3 = self._makeOne(rules=[rule1, rule2])
|
||
|
|
pb_rule3 = _GcRulePB(
|
||
|
|
union=_GcRuleUnionPB(rules=[pb_rule1, pb_rule2]))
|
||
|
|
|
||
|
|
max_num_versions2 = 1337
|
||
|
|
rule4 = MaxVersionsGCRule(max_num_versions2)
|
||
|
|
pb_rule4 = _GcRulePB(max_num_versions=max_num_versions2)
|
||
|
|
|
||
|
|
rule5 = self._makeOne(rules=[rule3, rule4])
|
||
|
|
pb_rule5 = _GcRulePB(
|
||
|
|
union=_GcRuleUnionPB(rules=[pb_rule3, pb_rule4]))
|
||
|
|
|
||
|
|
gc_rule_pb = rule5.to_pb()
|
||
|
|
self.assertEqual(gc_rule_pb, pb_rule5)
|
||
|
|
|
||
|
|
|
||
|
|
class TestGCRuleIntersection(unittest2.TestCase):
|
||
|
|
|
||
|
|
def _getTargetClass(self):
|
||
|
|
from gcloud.bigtable.column_family import GCRuleIntersection
|
||
|
|
return GCRuleIntersection
|
||
|
|
|
||
|
|
def _makeOne(self, *args, **kwargs):
|
||
|
|
return self._getTargetClass()(*args, **kwargs)
|
||
|
|
|
||
|
|
def test_constructor(self):
|
||
|
|
rules = object()
|
||
|
|
rule_intersection = self._makeOne(rules)
|
||
|
|
self.assertTrue(rule_intersection.rules is rules)
|
||
|
|
|
||
|
|
def test___eq__(self):
|
||
|
|
rules = object()
|
||
|
|
gc_rule1 = self._makeOne(rules)
|
||
|
|
gc_rule2 = self._makeOne(rules)
|
||
|
|
self.assertEqual(gc_rule1, gc_rule2)
|
||
|
|
|
||
|
|
def test___eq__type_differ(self):
|
||
|
|
rules = object()
|
||
|
|
gc_rule1 = self._makeOne(rules)
|
||
|
|
gc_rule2 = object()
|
||
|
|
self.assertNotEqual(gc_rule1, gc_rule2)
|
||
|
|
|
||
|
|
def test___ne__same_value(self):
|
||
|
|
rules = object()
|
||
|
|
gc_rule1 = self._makeOne(rules)
|
||
|
|
gc_rule2 = self._makeOne(rules)
|
||
|
|
comparison_val = (gc_rule1 != gc_rule2)
|
||
|
|
self.assertFalse(comparison_val)
|
||
|
|
|
||
|
|
def test_to_pb(self):
|
||
|
|
import datetime
|
||
|
|
from google.protobuf import duration_pb2
|
||
|
|
from gcloud.bigtable.column_family import MaxAgeGCRule
|
||
|
|
from gcloud.bigtable.column_family import MaxVersionsGCRule
|
||
|
|
|
||
|
|
max_num_versions = 42
|
||
|
|
rule1 = MaxVersionsGCRule(max_num_versions)
|
||
|
|
pb_rule1 = _GcRulePB(max_num_versions=max_num_versions)
|
||
|
|
|
||
|
|
max_age = datetime.timedelta(seconds=1)
|
||
|
|
rule2 = MaxAgeGCRule(max_age)
|
||
|
|
pb_rule2 = _GcRulePB(
|
||
|
|
max_age=duration_pb2.Duration(seconds=1))
|
||
|
|
|
||
|
|
rule3 = self._makeOne(rules=[rule1, rule2])
|
||
|
|
pb_rule3 = _GcRulePB(
|
||
|
|
intersection=_GcRuleIntersectionPB(
|
||
|
|
rules=[pb_rule1, pb_rule2]))
|
||
|
|
|
||
|
|
gc_rule_pb = rule3.to_pb()
|
||
|
|
self.assertEqual(gc_rule_pb, pb_rule3)
|
||
|
|
|
||
|
|
def test_to_pb_nested(self):
|
||
|
|
import datetime
|
||
|
|
from google.protobuf import duration_pb2
|
||
|
|
from gcloud.bigtable.column_family import MaxAgeGCRule
|
||
|
|
from gcloud.bigtable.column_family import MaxVersionsGCRule
|
||
|
|
|
||
|
|
max_num_versions1 = 42
|
||
|
|
rule1 = MaxVersionsGCRule(max_num_versions1)
|
||
|
|
pb_rule1 = _GcRulePB(max_num_versions=max_num_versions1)
|
||
|
|
|
||
|
|
max_age = datetime.timedelta(seconds=1)
|
||
|
|
rule2 = MaxAgeGCRule(max_age)
|
||
|
|
pb_rule2 = _GcRulePB(
|
||
|
|
max_age=duration_pb2.Duration(seconds=1))
|
||
|
|
|
||
|
|
rule3 = self._makeOne(rules=[rule1, rule2])
|
||
|
|
pb_rule3 = _GcRulePB(
|
||
|
|
intersection=_GcRuleIntersectionPB(
|
||
|
|
rules=[pb_rule1, pb_rule2]))
|
||
|
|
|
||
|
|
max_num_versions2 = 1337
|
||
|
|
rule4 = MaxVersionsGCRule(max_num_versions2)
|
||
|
|
pb_rule4 = _GcRulePB(max_num_versions=max_num_versions2)
|
||
|
|
|
||
|
|
rule5 = self._makeOne(rules=[rule3, rule4])
|
||
|
|
pb_rule5 = _GcRulePB(
|
||
|
|
intersection=_GcRuleIntersectionPB(
|
||
|
|
rules=[pb_rule3, pb_rule4]))
|
||
|
|
|
||
|
|
gc_rule_pb = rule5.to_pb()
|
||
|
|
self.assertEqual(gc_rule_pb, pb_rule5)
|
||
|
|
|
||
|
|
|
||
|
|
class TestColumnFamily(unittest2.TestCase):
|
||
|
|
|
||
|
|
def _getTargetClass(self):
|
||
|
|
from gcloud.bigtable.column_family import ColumnFamily
|
||
|
|
return ColumnFamily
|
||
|
|
|
||
|
|
def _makeOne(self, *args, **kwargs):
|
||
|
|
return self._getTargetClass()(*args, **kwargs)
|
||
|
|
|
||
|
|
def test_constructor(self):
|
||
|
|
column_family_id = u'column-family-id'
|
||
|
|
table = object()
|
||
|
|
gc_rule = object()
|
||
|
|
column_family = self._makeOne(
|
||
|
|
column_family_id, table, gc_rule=gc_rule)
|
||
|
|
|
||
|
|
self.assertEqual(column_family.column_family_id, column_family_id)
|
||
|
|
self.assertTrue(column_family._table is table)
|
||
|
|
self.assertTrue(column_family.gc_rule is gc_rule)
|
||
|
|
|
||
|
|
def test_name_property(self):
|
||
|
|
column_family_id = u'column-family-id'
|
||
|
|
table_name = 'table_name'
|
||
|
|
table = _Table(table_name)
|
||
|
|
column_family = self._makeOne(column_family_id, table)
|
||
|
|
|
||
|
|
expected_name = table_name + '/columnFamilies/' + column_family_id
|
||
|
|
self.assertEqual(column_family.name, expected_name)
|
||
|
|
|
||
|
|
def test___eq__(self):
|
||
|
|
column_family_id = 'column_family_id'
|
||
|
|
table = object()
|
||
|
|
gc_rule = object()
|
||
|
|
column_family1 = self._makeOne(column_family_id, table,
|
||
|
|
gc_rule=gc_rule)
|
||
|
|
column_family2 = self._makeOne(column_family_id, table,
|
||
|
|
gc_rule=gc_rule)
|
||
|
|
self.assertEqual(column_family1, column_family2)
|
||
|
|
|
||
|
|
def test___eq__type_differ(self):
|
||
|
|
column_family1 = self._makeOne('column_family_id', None)
|
||
|
|
column_family2 = object()
|
||
|
|
self.assertNotEqual(column_family1, column_family2)
|
||
|
|
|
||
|
|
def test___ne__same_value(self):
|
||
|
|
column_family_id = 'column_family_id'
|
||
|
|
table = object()
|
||
|
|
gc_rule = object()
|
||
|
|
column_family1 = self._makeOne(column_family_id, table,
|
||
|
|
gc_rule=gc_rule)
|
||
|
|
column_family2 = self._makeOne(column_family_id, table,
|
||
|
|
gc_rule=gc_rule)
|
||
|
|
comparison_val = (column_family1 != column_family2)
|
||
|
|
self.assertFalse(comparison_val)
|
||
|
|
|
||
|
|
def test___ne__(self):
|
||
|
|
column_family1 = self._makeOne('column_family_id1', None)
|
||
|
|
column_family2 = self._makeOne('column_family_id2', None)
|
||
|
|
self.assertNotEqual(column_family1, column_family2)
|
||
|
|
|
||
|
|
def _create_test_helper(self, gc_rule=None):
|
||
|
|
from gcloud.bigtable._generated_v2 import (
|
||
|
|
bigtable_table_admin_pb2 as table_admin_v2_pb2)
|
||
|
|
from gcloud.bigtable._testing import _FakeStub
|
||
|
|
|
||
|
|
project_id = 'project-id'
|
||
|
|
zone = 'zone'
|
||
|
|
cluster_id = 'cluster-id'
|
||
|
|
table_id = 'table-id'
|
||
|
|
column_family_id = 'column-family-id'
|
||
|
|
timeout_seconds = 4
|
||
|
|
table_name = ('projects/' + project_id + '/zones/' + zone +
|
||
|
|
'/clusters/' + cluster_id + '/tables/' + table_id)
|
||
|
|
|
||
|
|
client = _Client(timeout_seconds=timeout_seconds)
|
||
|
|
table = _Table(table_name, client=client)
|
||
|
|
column_family = self._makeOne(
|
||
|
|
column_family_id, table, gc_rule=gc_rule)
|
||
|
|
|
||
|
|
# Create request_pb
|
||
|
|
if gc_rule is None:
|
||
|
|
column_family_pb = _ColumnFamilyPB()
|
||
|
|
else:
|
||
|
|
column_family_pb = _ColumnFamilyPB(gc_rule=gc_rule.to_pb())
|
||
|
|
request_pb = table_admin_v2_pb2.ModifyColumnFamiliesRequest(
|
||
|
|
name=table_name)
|
||
|
|
request_pb.modifications.add(
|
||
|
|
id=column_family_id,
|
||
|
|
create=column_family_pb,
|
||
|
|
)
|
||
|
|
|
||
|
|
# Create response_pb
|
||
|
|
response_pb = _ColumnFamilyPB()
|
||
|
|
|
||
|
|
# Patch the stub used by the API method.
|
||
|
|
client._table_stub = stub = _FakeStub(response_pb)
|
||
|
|
|
||
|
|
# Create expected_result.
|
||
|
|
expected_result = None # create() has no return value.
|
||
|
|
|
||
|
|
# Perform the method and check the result.
|
||
|
|
self.assertEqual(stub.results, (response_pb,))
|
||
|
|
result = column_family.create()
|
||
|
|
self.assertEqual(stub.results, ())
|
||
|
|
self.assertEqual(result, expected_result)
|
||
|
|
self.assertEqual(stub.method_calls, [(
|
||
|
|
'ModifyColumnFamilies',
|
||
|
|
(request_pb, timeout_seconds),
|
||
|
|
{},
|
||
|
|
)])
|
||
|
|
|
||
|
|
def test_create(self):
|
||
|
|
self._create_test_helper(gc_rule=None)
|
||
|
|
|
||
|
|
def test_create_with_gc_rule(self):
|
||
|
|
from gcloud.bigtable.column_family import MaxVersionsGCRule
|
||
|
|
gc_rule = MaxVersionsGCRule(1337)
|
||
|
|
self._create_test_helper(gc_rule=gc_rule)
|
||
|
|
|
||
|
|
def _update_test_helper(self, gc_rule=None):
|
||
|
|
from gcloud.bigtable._testing import _FakeStub
|
||
|
|
from gcloud.bigtable._generated_v2 import (
|
||
|
|
bigtable_table_admin_pb2 as table_admin_v2_pb2)
|
||
|
|
|
||
|
|
project_id = 'project-id'
|
||
|
|
zone = 'zone'
|
||
|
|
cluster_id = 'cluster-id'
|
||
|
|
table_id = 'table-id'
|
||
|
|
column_family_id = 'column-family-id'
|
||
|
|
timeout_seconds = 28
|
||
|
|
table_name = ('projects/' + project_id + '/zones/' + zone +
|
||
|
|
'/clusters/' + cluster_id + '/tables/' + table_id)
|
||
|
|
|
||
|
|
client = _Client(timeout_seconds=timeout_seconds)
|
||
|
|
table = _Table(table_name, client=client)
|
||
|
|
column_family = self._makeOne(
|
||
|
|
column_family_id, table, gc_rule=gc_rule)
|
||
|
|
|
||
|
|
# Create request_pb
|
||
|
|
if gc_rule is None:
|
||
|
|
column_family_pb = _ColumnFamilyPB()
|
||
|
|
else:
|
||
|
|
column_family_pb = _ColumnFamilyPB(gc_rule=gc_rule.to_pb())
|
||
|
|
request_pb = table_admin_v2_pb2.ModifyColumnFamiliesRequest(
|
||
|
|
name=table_name)
|
||
|
|
request_pb.modifications.add(
|
||
|
|
id=column_family_id,
|
||
|
|
update=column_family_pb,
|
||
|
|
)
|
||
|
|
|
||
|
|
# Create response_pb
|
||
|
|
response_pb = _ColumnFamilyPB()
|
||
|
|
|
||
|
|
# Patch the stub used by the API method.
|
||
|
|
client._table_stub = stub = _FakeStub(response_pb)
|
||
|
|
|
||
|
|
# Create expected_result.
|
||
|
|
expected_result = None # update() has no return value.
|
||
|
|
|
||
|
|
# Perform the method and check the result.
|
||
|
|
self.assertEqual(stub.results, (response_pb,))
|
||
|
|
result = column_family.update()
|
||
|
|
self.assertEqual(stub.results, ())
|
||
|
|
self.assertEqual(result, expected_result)
|
||
|
|
self.assertEqual(stub.method_calls, [(
|
||
|
|
'ModifyColumnFamilies',
|
||
|
|
(request_pb, timeout_seconds),
|
||
|
|
{},
|
||
|
|
)])
|
||
|
|
|
||
|
|
def test_update(self):
|
||
|
|
self._update_test_helper(gc_rule=None)
|
||
|
|
|
||
|
|
def test_update_with_gc_rule(self):
|
||
|
|
from gcloud.bigtable.column_family import MaxVersionsGCRule
|
||
|
|
gc_rule = MaxVersionsGCRule(1337)
|
||
|
|
self._update_test_helper(gc_rule=gc_rule)
|
||
|
|
|
||
|
|
def test_delete(self):
|
||
|
|
from google.protobuf import empty_pb2
|
||
|
|
from gcloud.bigtable._generated_v2 import (
|
||
|
|
bigtable_table_admin_pb2 as table_admin_v2_pb2)
|
||
|
|
from gcloud.bigtable._testing import _FakeStub
|
||
|
|
|
||
|
|
project_id = 'project-id'
|
||
|
|
zone = 'zone'
|
||
|
|
cluster_id = 'cluster-id'
|
||
|
|
table_id = 'table-id'
|
||
|
|
column_family_id = 'column-family-id'
|
||
|
|
timeout_seconds = 7
|
||
|
|
table_name = ('projects/' + project_id + '/zones/' + zone +
|
||
|
|
'/clusters/' + cluster_id + '/tables/' + table_id)
|
||
|
|
|
||
|
|
client = _Client(timeout_seconds=timeout_seconds)
|
||
|
|
table = _Table(table_name, client=client)
|
||
|
|
column_family = self._makeOne(column_family_id, table)
|
||
|
|
|
||
|
|
# Create request_pb
|
||
|
|
request_pb = table_admin_v2_pb2.ModifyColumnFamiliesRequest(
|
||
|
|
name=table_name)
|
||
|
|
request_pb.modifications.add(
|
||
|
|
id=column_family_id,
|
||
|
|
drop=True)
|
||
|
|
|
||
|
|
# Create response_pb
|
||
|
|
response_pb = empty_pb2.Empty()
|
||
|
|
|
||
|
|
# Patch the stub used by the API method.
|
||
|
|
client._table_stub = stub = _FakeStub(response_pb)
|
||
|
|
|
||
|
|
# Create expected_result.
|
||
|
|
expected_result = None # delete() has no return value.
|
||
|
|
|
||
|
|
# Perform the method and check the result.
|
||
|
|
self.assertEqual(stub.results, (response_pb,))
|
||
|
|
result = column_family.delete()
|
||
|
|
self.assertEqual(stub.results, ())
|
||
|
|
self.assertEqual(result, expected_result)
|
||
|
|
self.assertEqual(stub.method_calls, [(
|
||
|
|
'ModifyColumnFamilies',
|
||
|
|
(request_pb, timeout_seconds),
|
||
|
|
{},
|
||
|
|
)])
|
||
|
|
|
||
|
|
|
||
|
|
class Test__gc_rule_from_pb(unittest2.TestCase):
|
||
|
|
|
||
|
|
def _callFUT(self, *args, **kwargs):
|
||
|
|
from gcloud.bigtable.column_family import _gc_rule_from_pb
|
||
|
|
return _gc_rule_from_pb(*args, **kwargs)
|
||
|
|
|
||
|
|
def test_empty(self):
|
||
|
|
|
||
|
|
gc_rule_pb = _GcRulePB()
|
||
|
|
self.assertEqual(self._callFUT(gc_rule_pb), None)
|
||
|
|
|
||
|
|
def test_max_num_versions(self):
|
||
|
|
from gcloud.bigtable.column_family import MaxVersionsGCRule
|
||
|
|
|
||
|
|
orig_rule = MaxVersionsGCRule(1)
|
||
|
|
gc_rule_pb = orig_rule.to_pb()
|
||
|
|
result = self._callFUT(gc_rule_pb)
|
||
|
|
self.assertTrue(isinstance(result, MaxVersionsGCRule))
|
||
|
|
self.assertEqual(result, orig_rule)
|
||
|
|
|
||
|
|
def test_max_age(self):
|
||
|
|
import datetime
|
||
|
|
from gcloud.bigtable.column_family import MaxAgeGCRule
|
||
|
|
|
||
|
|
orig_rule = MaxAgeGCRule(datetime.timedelta(seconds=1))
|
||
|
|
gc_rule_pb = orig_rule.to_pb()
|
||
|
|
result = self._callFUT(gc_rule_pb)
|
||
|
|
self.assertTrue(isinstance(result, MaxAgeGCRule))
|
||
|
|
self.assertEqual(result, orig_rule)
|
||
|
|
|
||
|
|
def test_union(self):
|
||
|
|
import datetime
|
||
|
|
from gcloud.bigtable.column_family import GCRuleUnion
|
||
|
|
from gcloud.bigtable.column_family import MaxAgeGCRule
|
||
|
|
from gcloud.bigtable.column_family import MaxVersionsGCRule
|
||
|
|
|
||
|
|
rule1 = MaxVersionsGCRule(1)
|
||
|
|
rule2 = MaxAgeGCRule(datetime.timedelta(seconds=1))
|
||
|
|
orig_rule = GCRuleUnion([rule1, rule2])
|
||
|
|
gc_rule_pb = orig_rule.to_pb()
|
||
|
|
result = self._callFUT(gc_rule_pb)
|
||
|
|
self.assertTrue(isinstance(result, GCRuleUnion))
|
||
|
|
self.assertEqual(result, orig_rule)
|
||
|
|
|
||
|
|
def test_intersection(self):
|
||
|
|
import datetime
|
||
|
|
from gcloud.bigtable.column_family import GCRuleIntersection
|
||
|
|
from gcloud.bigtable.column_family import MaxAgeGCRule
|
||
|
|
from gcloud.bigtable.column_family import MaxVersionsGCRule
|
||
|
|
|
||
|
|
rule1 = MaxVersionsGCRule(1)
|
||
|
|
rule2 = MaxAgeGCRule(datetime.timedelta(seconds=1))
|
||
|
|
orig_rule = GCRuleIntersection([rule1, rule2])
|
||
|
|
gc_rule_pb = orig_rule.to_pb()
|
||
|
|
result = self._callFUT(gc_rule_pb)
|
||
|
|
self.assertTrue(isinstance(result, GCRuleIntersection))
|
||
|
|
self.assertEqual(result, orig_rule)
|
||
|
|
|
||
|
|
def test_unknown_field_name(self):
|
||
|
|
class MockProto(object):
|
||
|
|
|
||
|
|
names = []
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def WhichOneof(cls, name):
|
||
|
|
cls.names.append(name)
|
||
|
|
return 'unknown'
|
||
|
|
|
||
|
|
self.assertEqual(MockProto.names, [])
|
||
|
|
self.assertRaises(ValueError, self._callFUT, MockProto)
|
||
|
|
self.assertEqual(MockProto.names, ['rule'])
|
||
|
|
|
||
|
|
|
||
|
|
def _GcRulePB(*args, **kw):
|
||
|
|
from gcloud.bigtable._generated_v2 import (
|
||
|
|
table_pb2 as table_v2_pb2)
|
||
|
|
return table_v2_pb2.GcRule(*args, **kw)
|
||
|
|
|
||
|
|
|
||
|
|
def _GcRuleIntersectionPB(*args, **kw):
|
||
|
|
from gcloud.bigtable._generated_v2 import (
|
||
|
|
table_pb2 as table_v2_pb2)
|
||
|
|
return table_v2_pb2.GcRule.Intersection(*args, **kw)
|
||
|
|
|
||
|
|
|
||
|
|
def _GcRuleUnionPB(*args, **kw):
|
||
|
|
from gcloud.bigtable._generated_v2 import (
|
||
|
|
table_pb2 as table_v2_pb2)
|
||
|
|
return table_v2_pb2.GcRule.Union(*args, **kw)
|
||
|
|
|
||
|
|
|
||
|
|
def _ColumnFamilyPB(*args, **kw):
|
||
|
|
from gcloud.bigtable._generated_v2 import (
|
||
|
|
table_pb2 as table_v2_pb2)
|
||
|
|
return table_v2_pb2.ColumnFamily(*args, **kw)
|
||
|
|
|
||
|
|
|
||
|
|
class _Instance(object):
|
||
|
|
|
||
|
|
def __init__(self, client=None):
|
||
|
|
self._client = client
|
||
|
|
|
||
|
|
|
||
|
|
class _Client(object):
|
||
|
|
|
||
|
|
def __init__(self, timeout_seconds=None):
|
||
|
|
self.timeout_seconds = timeout_seconds
|
||
|
|
|
||
|
|
|
||
|
|
class _Table(object):
|
||
|
|
|
||
|
|
def __init__(self, name, client=None):
|
||
|
|
self.name = name
|
||
|
|
self._instance = _Instance(client)
|