projects / ceph.git / blame
| Commit | Line | Data |
|---|---|---|
| 1d09f67e TL |
1 | # Licensed to the Apache Software Foundation (ASF) under one |
| 2 | # or more contributor license agreements. See the NOTICE file | |
| 3 | # distributed with this work for additional information | |
| 4 | # regarding copyright ownership. The ASF licenses this file | |
| 5 | # to you under the Apache License, Version 2.0 (the | |
| 6 | # "License"); you may not use this file except in compliance | |
| 7 | # with the License. You may obtain a copy of the License at | |
| 8 | # | |
| 9 | # http://www.apache.org/licenses/LICENSE-2.0 | |
| 10 | # | |
| 11 | # Unless required by applicable law or agreed to in writing, | |
| 12 | # software distributed under the License is distributed on an | |
| 13 | # "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY | |
| 14 | # KIND, either express or implied. See the License for the | |
| 15 | # specific language governing permissions and limitations | |
| 16 | # under the License. | |
| 17 | ||
| 18 | ||
| 19 | test_that("MessageReader can be created from raw vectors", { | |
| 20 | batch <- record_batch(x = 1:10) | |
| 21 | bytes <- batch$serialize() | |
| 22 | ||
| 23 | reader <- MessageReader$create(bytes) | |
| 24 | ||
| 25 | message <- reader$ReadNextMessage() | |
| 26 | expect_r6_class(message, "Message") | |
| 27 | expect_equal(message$type, MessageType$RECORD_BATCH) | |
| 28 | expect_r6_class(message$body, "Buffer") | |
| 29 | expect_r6_class(message$metadata, "Buffer") | |
| 30 | ||
| 31 | message <- reader$ReadNextMessage() | |
| 32 | expect_null(message) | |
| 33 | ||
| 34 | schema <- schema(x = int32()) | |
| 35 | bytes <- schema$serialize() | |
| 36 | ||
| 37 | reader <- MessageReader$create(bytes) | |
| 38 | ||
| 39 | message <- reader$ReadNextMessage() | |
| 40 | expect_r6_class(message, "Message") | |
| 41 | expect_equal(message$type, MessageType$SCHEMA) | |
| 42 | expect_r6_class(message$body, "Buffer") | |
| 43 | expect_r6_class(message$metadata, "Buffer") | |
| 44 | ||
| 45 | message <- reader$ReadNextMessage() | |
| 46 | expect_null(message) | |
| 47 | }) | |
| 48 | ||
| 49 | test_that("MessageReader can be created from input stream", { | |
| 50 | batch <- record_batch(x = 1:10) | |
| 51 | bytes <- batch$serialize() | |
| 52 | ||
| 53 | stream <- BufferReader$create(bytes) | |
| 54 | expect_r6_class(stream, "BufferReader") | |
| 55 | ||
| 56 | reader <- MessageReader$create(stream) | |
| 57 | expect_r6_class(reader, "MessageReader") | |
| 58 | ||
| 59 | message <- reader$ReadNextMessage() | |
| 60 | expect_r6_class(message, "Message") | |
| 61 | expect_equal(message$type, MessageType$RECORD_BATCH) | |
| 62 | expect_r6_class(message$body, "Buffer") | |
| 63 | expect_r6_class(message$metadata, "Buffer") | |
| 64 | ||
| 65 | message <- reader$ReadNextMessage() | |
| 66 | expect_null(message) | |
| 67 | ||
| 68 | schema <- schema(x = int32()) | |
| 69 | bytes <- schema$serialize() | |
| 70 | ||
| 71 | stream <- BufferReader$create(bytes) | |
| 72 | expect_r6_class(stream, "BufferReader") | |
| 73 | ||
| 74 | reader <- MessageReader$create(stream) | |
| 75 | expect_r6_class(reader, "MessageReader") | |
| 76 | ||
| 77 | message <- reader$ReadNextMessage() | |
| 78 | expect_r6_class(message, "Message") | |
| 79 | expect_equal(message$type, MessageType$SCHEMA) | |
| 80 | expect_r6_class(message$body, "Buffer") | |
| 81 | expect_r6_class(message$metadata, "Buffer") | |
| 82 | ||
| 83 | message <- reader$ReadNextMessage() | |
| 84 | expect_null(message) | |
| 85 | }) |