swift-nio/Tests/NIOHTTP1Tests/ByteBufferUtilsTest.swift

//===----------------------------------------------------------------------===//
//
// This source file is part of the SwiftNIO open source project
//
// Copyright (c) 2017-2021 Apple Inc. and the SwiftNIO project authors
// Licensed under Apache License v2.0
//
// See LICENSE.txt for license information
// See CONTRIBUTORS.txt for the list of SwiftNIO project authors
//
// SPDX-License-Identifier: Apache-2.0
//
//===----------------------------------------------------------------------===//

import NIOCore
import XCTest

@testable import NIOHTTP1

private enum DummyError: Error {
    case err
}

class ByteBufferUtilsTest: XCTestCase {

    func testComparators() {
        let someByteBuffer: ByteBuffer = ByteBuffer(string: "fiRSt")
        XCTAssert(
            someByteBuffer.readableBytesView.compareCaseInsensitiveASCIIBytes(
                to: "first".utf8
            )
        )
        XCTAssert(
            someByteBuffer.readableBytesView.compareCaseInsensitiveASCIIBytes(
                to: "fiRSt".utf8
            )
        )
        XCTAssert(
            someByteBuffer.readableBytesView.compareCaseInsensitiveASCIIBytes(
                to: "fIrst".utf8
            )
        )
        XCTAssertFalse(
            someByteBuffer.readableBytesView.compareCaseInsensitiveASCIIBytes(
                to: "fIrt".utf8
            )
        )
        XCTAssertFalse(
            someByteBuffer.readableBytesView.compareCaseInsensitiveASCIIBytes(
                to: "firsta".utf8
            )
        )
        XCTAssertFalse(
            someByteBuffer.readableBytesView.compareCaseInsensitiveASCIIBytes(
                to: "afirst".utf8
            )
        )
        XCTAssertFalse(
            someByteBuffer.readableBytesView.compareCaseInsensitiveASCIIBytes(
                to: "eiRSt".utf8
            )
        )
        XCTAssertFalse(
            someByteBuffer.readableBytesView.compareCaseInsensitiveASCIIBytes(
                to: "fIrso".utf8
            )
        )
        XCTAssertFalse(
            someByteBuffer.readableBytesView.compareCaseInsensitiveASCIIBytes(
                to: "firot".utf8
            )
        )
    }

    private func byteBufferView(string: String) -> ByteBufferView {
        let byteBufferAllocator = ByteBufferAllocator()
        var buffer = byteBufferAllocator.buffer(capacity: string.lengthOfBytes(using: .utf8))
        buffer.writeString(string)
        return buffer.readableBytesView
    }

    func testTrimming() {
        XCTAssertEqual(
            byteBufferView(string: "   first").trimSpaces().map({ CChar($0) }),
            byteBufferView(string: "first").map({ CChar($0) })
        )
        XCTAssertEqual(
            byteBufferView(string: "   first  ").trimSpaces().map({ CChar($0) }),
            byteBufferView(string: "first").map({ CChar($0) })
        )
        XCTAssertEqual(
            byteBufferView(string: "first  ").trimSpaces().map({ CChar($0) }),
            byteBufferView(string: "first").map({ CChar($0) })
        )
        XCTAssertEqual(
            byteBufferView(string: "first").trimSpaces().map({ CChar($0) }),
            byteBufferView(string: "first").map({ CChar($0) })
        )
        XCTAssertEqual(
            byteBufferView(string: " \t\t  fi  rst").trimSpaces().map({ CChar($0) }),
            byteBufferView(string: "fi  rst").map({ CChar($0) })
        )
        XCTAssertEqual(
            byteBufferView(string: "   firs  t \t ").trimSpaces().map({ CChar($0) }),
            byteBufferView(string: "firs  t").map({ CChar($0) })
        )
        XCTAssertEqual(
            byteBufferView(string: "f\t  irst  ").trimSpaces().map({ CChar($0) }),
            byteBufferView(string: "f\t  irst").map({ CChar($0) })
        )
        XCTAssertEqual(
            byteBufferView(string: "f i  rs  t").trimSpaces().map({ CChar($0) }),
            byteBufferView(string: "f i  rs  t").map({ CChar($0) })
        )
        XCTAssertEqual(
            byteBufferView(string: "   \t \t ").trimSpaces().map({ CChar($0) }),
            byteBufferView(string: "").map({ CChar($0) })
        )
    }

}