grpc-swift/Tests/GRPCTests/LengthPrefixedMessageWriterTests.swift

/*
 * Copyright 2020, gRPC Authors 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.
 */
@testable import GRPC
import NIO
import XCTest

class LengthPrefixedMessageWriterTests: GRPCTestCase {
  func testWriteBytesWithNoLeadingSpaceOrCompression() throws {
    let writer = LengthPrefixedMessageWriter()
    let allocator = ByteBufferAllocator()
    let buffer = allocator.buffer(bytes: [1, 2, 3])

    var prefixed = try writer.write(buffer: buffer, allocator: allocator)
    XCTAssertEqual(prefixed.readInteger(as: UInt8.self), 0)
    XCTAssertEqual(prefixed.readInteger(as: UInt32.self), 3)
    XCTAssertEqual(prefixed.readBytes(length: 3), [1, 2, 3])
    XCTAssertEqual(prefixed.readableBytes, 0)
  }

  func testWriteBytesWithLeadingSpaceAndNoCompression() throws {
    let writer = LengthPrefixedMessageWriter()
    let allocator = ByteBufferAllocator()

    var buffer = allocator.buffer(bytes: Array(repeating: 0, count: 5) + [1, 2, 3])
    buffer.moveReaderIndex(forwardBy: 5)

    var prefixed = try writer.write(buffer: buffer, allocator: allocator)
    XCTAssertEqual(prefixed.readInteger(as: UInt8.self), 0)
    XCTAssertEqual(prefixed.readInteger(as: UInt32.self), 3)
    XCTAssertEqual(prefixed.readBytes(length: 3), [1, 2, 3])
    XCTAssertEqual(prefixed.readableBytes, 0)
  }

  func testWriteBytesWithNoLeadingSpaceAndCompression() throws {
    let writer = LengthPrefixedMessageWriter(compression: .gzip)
    let allocator = ByteBufferAllocator()

    let buffer = allocator.buffer(bytes: [1, 2, 3])
    var prefixed = try writer.write(buffer: buffer, allocator: allocator)

    XCTAssertEqual(prefixed.readInteger(as: UInt8.self), 1)
    let size = prefixed.readInteger(as: UInt32.self)!
    XCTAssertGreaterThanOrEqual(size, 0)
    XCTAssertNotNil(prefixed.readBytes(length: Int(size)))
    XCTAssertEqual(prefixed.readableBytes, 0)
  }

  func testWriteBytesWithLeadingSpaceAndCompression() throws {
    let writer = LengthPrefixedMessageWriter(compression: .gzip)
    let allocator = ByteBufferAllocator()

    var buffer = allocator.buffer(bytes: Array(repeating: 0, count: 5) + [1, 2, 3])
    buffer.moveReaderIndex(forwardBy: 5)
    var prefixed = try writer.write(buffer: buffer, allocator: allocator)

    XCTAssertEqual(prefixed.readInteger(as: UInt8.self), 1)
    let size = prefixed.readInteger(as: UInt32.self)!
    XCTAssertGreaterThanOrEqual(size, 0)
    XCTAssertNotNil(prefixed.readBytes(length: Int(size)))
    XCTAssertEqual(prefixed.readableBytes, 0)
  }
}