Esempio n. 1
0
    def test_to_pb_nested(self):
        import datetime
        from gcloud.bigtable._generated import (bigtable_table_data_pb2 as
                                                data_pb2)
        from gcloud.bigtable._generated 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 = data_pb2.GcRule(max_num_versions=max_num_versions1)

        max_age = datetime.timedelta(seconds=1)
        rule2 = MaxAgeGCRule(max_age)
        pb_rule2 = data_pb2.GcRule(max_age=duration_pb2.Duration(seconds=1))

        rule3 = self._makeOne(rules=[rule1, rule2])
        pb_rule3 = data_pb2.GcRule(intersection=data_pb2.GcRule.Intersection(
            rules=[pb_rule1, pb_rule2]))

        max_num_versions2 = 1337
        rule4 = MaxVersionsGCRule(max_num_versions2)
        pb_rule4 = data_pb2.GcRule(max_num_versions=max_num_versions2)

        rule5 = self._makeOne(rules=[rule3, rule4])
        pb_rule5 = data_pb2.GcRule(intersection=data_pb2.GcRule.Intersection(
            rules=[pb_rule3, pb_rule4]))

        gc_rule_pb = rule5.to_pb()
        self.assertEqual(gc_rule_pb, pb_rule5)
Esempio n. 2
0
    def to_pb(self):
        """Converts the garbage collection rule to a protobuf.

        :rtype: :class:`.data_pb2.GcRule`
        :returns: The converted current object.
        """
        return data_pb2.GcRule(max_num_versions=self.max_num_versions)
Esempio n. 3
0
 def test_to_pb(self):
     from gcloud.bigtable._generated import (bigtable_table_data_pb2 as
                                             data_pb2)
     max_num_versions = 1337
     gc_rule = self._makeOne(max_num_versions=max_num_versions)
     pb_val = gc_rule.to_pb()
     self.assertEqual(pb_val,
                      data_pb2.GcRule(max_num_versions=max_num_versions))
Esempio n. 4
0
    def to_pb(self):
        """Converts the garbage collection rule to a protobuf.

        :rtype: :class:`.data_pb2.GcRule`
        :returns: The converted current object.
        """
        max_age = _timedelta_to_duration_pb(self.max_age)
        return data_pb2.GcRule(max_age=max_age)
Esempio n. 5
0
    def to_pb(self):
        """Converts the intersection into a single GC rule as a protobuf.

        :rtype: :class:`.data_pb2.GcRule`
        :returns: The converted current object.
        """
        intersection = data_pb2.GcRule.Intersection(
            rules=[rule.to_pb() for rule in self.rules])
        return data_pb2.GcRule(intersection=intersection)
Esempio n. 6
0
    def to_pb(self):
        """Converts the union into a single gc rule as a protobuf.

        :rtype: :class:`.data_pb2.GcRule`
        :returns: The converted current object.
        """
        union = data_pb2.GcRule.Union(
            rules=[rule.to_pb() for rule in self.rules])
        return data_pb2.GcRule(union=union)
Esempio n. 7
0
    def test_to_pb(self):
        import datetime
        from gcloud.bigtable._generated import (bigtable_table_data_pb2 as
                                                data_pb2)
        from gcloud.bigtable._generated 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, data_pb2.GcRule(max_age=duration))
Esempio n. 8
0
    def test_to_pb(self):
        import datetime
        from google.protobuf import duration_pb2
        from gcloud.bigtable._generated import (bigtable_table_data_pb2 as
                                                data_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 = data_pb2.GcRule(max_num_versions=max_num_versions)

        max_age = datetime.timedelta(seconds=1)
        rule2 = MaxAgeGCRule(max_age)
        pb_rule2 = data_pb2.GcRule(max_age=duration_pb2.Duration(seconds=1))

        rule3 = self._makeOne(rules=[rule1, rule2])
        pb_rule3 = data_pb2.GcRule(union=data_pb2.GcRule.Union(
            rules=[pb_rule1, pb_rule2]))

        gc_rule_pb = rule3.to_pb()
        self.assertEqual(gc_rule_pb, pb_rule3)
Esempio n. 9
0
    def test_empty(self):
        from gcloud.bigtable._generated import (bigtable_table_data_pb2 as
                                                data_pb2)

        gc_rule_pb = data_pb2.GcRule()
        self.assertEqual(self._callFUT(gc_rule_pb), None)