ceph/src/arrow/r/tests/testthat/test-message-reader.R

   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 })