tests

Author: vgvoleg

ydb/_grpc/grpcwrapper/ydb_topic_test.py

@@ -1,4 +1,24 @@
+import datetime
+import typing
+
+from google.protobuf.internal import containers
+from google.protobuf import duration_pb2
+
 from ydb._grpc.grpcwrapper.ydb_topic import OffsetsRange
+from .ydb_topic import AlterTopicRequest
+from .ydb_topic_public_types import (
+    AlterTopicRequestParams,
+    PublicAlterConsumer,
+    PublicConsumer,
+    PublicMeteringMode,
+    PublicCodec
+)
+
+# Workaround for good IDE and universal for runtime
+if typing.TYPE_CHECKING:
+    from ..v4.protos import ydb_topic_pb2
+else:
+    from ..common.protos import ydb_topic_pb2
 
 
 def test_offsets_range_intersected():
@@ -17,3 +37,72 @@ def test_offsets_range_intersected():
     ]:
         assert OffsetsRange(test[0], test[1]).is_intersected_with(OffsetsRange(test[2], test[3]))
         assert OffsetsRange(test[2], test[3]).is_intersected_with(OffsetsRange(test[0], test[1]))
+
+
+def test_alter_topic_request_from_public_to_proto():
+    # Specify all fields with all possible input ways
+    params = {
+        "path": "topic_name",
+        "add_consumers": [
+            "new_consumer_1",
+            PublicConsumer("new_consumer_2"),
+        ],
+        "alter_consumers": [
+            "old_consumer_1",
+            PublicAlterConsumer("old_consumer_2"),
+        ],
+        "drop_consumers": ["redundant_consumer"],
+        "set_retention_period": datetime.timedelta(weeks=4),
+        "set_retention_storage_mb": 4,
+        "set_supported_codecs": [
+            1,
+            PublicCodec(2)
+        ],
+        "set_partition_write_burst_bytes": 8,
+        "set_partition_write_speed_bytes_per_second": 15,
+        "alter_attributes": {'key': 'value'},
+        "set_metering_mode": 1,
+        "set_min_active_partitions": 2,
+        "set_partition_count_limit": 4,
+    }
+
+    params_public = AlterTopicRequestParams(**params)
+    request = AlterTopicRequest.from_public(params_public)
+    request_proto = request.to_proto()
+
+    # Check that everything was converted right
+    assert type(request_proto) == ydb_topic_pb2.AlterTopicRequest
+
+    assert request_proto.path == params['path']
+
+    assert type(request_proto.drop_consumers) == containers.RepeatedScalarFieldContainer
+    assert len(request_proto.drop_consumers) == len(params["drop_consumers"])
+    assert request_proto.drop_consumers[0] == params["drop_consumers"][0]
+
+    assert type(request_proto.add_consumers) == containers.RepeatedCompositeFieldContainer
+    assert len(request_proto.add_consumers) == len(params['add_consumers'])
+    for consumer in request_proto.add_consumers:
+        assert type(consumer) == ydb_topic_pb2.Consumer
+
+    assert type(request_proto.alter_consumers) == containers.RepeatedCompositeFieldContainer
+    assert len(request_proto.alter_consumers) == len(params['alter_consumers'])
+    for consumer in request_proto.alter_consumers:
+        assert type(consumer) == ydb_topic_pb2.AlterConsumer
+
+    assert type(request_proto.alter_attributes) == containers.ScalarMap
+    assert request_proto.alter_attributes == params['alter_attributes']
+
+    assert type(request_proto.alter_partitioning_settings) == ydb_topic_pb2.AlterPartitioningSettings
+    assert request_proto.alter_partitioning_settings.set_min_active_partitions == params['set_min_active_partitions']
+    assert request_proto.alter_partitioning_settings.set_partition_count_limit == params['set_partition_count_limit']
+
+    assert request_proto.set_partition_write_burst_bytes == params['set_partition_write_burst_bytes']
+    assert request_proto.set_partition_write_speed_bytes_per_second == params['set_partition_write_speed_bytes_per_second']
+
+    assert type(request_proto.set_retention_period) == duration_pb2.Duration
+    assert request_proto.set_retention_storage_mb == params['set_retention_storage_mb']
+
+    assert int(request_proto.set_metering_mode) == params['set_metering_mode']
+    assert type(request_proto.set_supported_codecs) == ydb_topic_pb2.SupportedCodecs
+    assert len(request_proto.set_supported_codecs.codecs) == len(params['set_supported_codecs'])
+    # TODO: finish