1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
|
//===----------------------------------------------------------------------===//
//
// This source file is part of the Swift open source project
//
// Copyright (c) 2022-2023 Apple Inc. and the Swift project authors
// Licensed under Apache License v2.0 with Runtime Library Exception
//
// See https://swift.org/LICENSE.txt for license information
// See https://swift.org/CONTRIBUTORS.txt for the list of Swift project authors
//
//===----------------------------------------------------------------------===//
import AsyncProcess
import Foundation
import NIO
import XCTest
final class AsyncByteBufferLineSequenceTests: XCTestCase {
func testJustManyNewlines() async throws {
for n in 0..<100 {
let inputs: [ByteBuffer] = [ByteBuffer(repeating: UInt8(ascii: "\n"), count: n)]
let lines = try await Array(inputs.async.splitIntoLines().strings)
XCTAssertEqual(Array(repeating: "", count: n), lines)
}
}
func testJustOneNewlineAtATime() async throws {
for n in 0..<100 {
let inputs: [ByteBuffer] = Array(repeating: ByteBuffer(integer: UInt8(ascii: "\n")), count: n)
let lines = try await Array(inputs.async.splitIntoLines().strings)
XCTAssertEqual(Array(repeating: "", count: n), lines)
}
}
func testManyChunksNoNewlineDeliveringLastChunk() async throws {
for n in 1..<100 {
let inputs: [ByteBuffer] = [ByteBuffer(repeating: 0, count: n)]
let lines = try await Array(inputs.async.splitIntoLines().strings)
XCTAssertEqual([String(repeating: "\0", count: n)], lines)
}
}
func testManyChunksNoNewlineNotDeliveringLastChunk() async throws {
for n in 0..<100 {
let inputs: [ByteBuffer] = [ByteBuffer(repeating: 0, count: n)]
let lines = try await Array(inputs.async.splitIntoLines(dropLastChunkIfNoNewline: true).strings)
XCTAssertEqual([], lines)
}
}
func testOverlyLongLineIsSplit() async throws {
var inputs = Array(repeating: ByteBuffer(integer: UInt8(0)), count: 10)
inputs.append(ByteBuffer(integer: UInt8(ascii: "\n")))
let lines = try await Array(
inputs.async.splitIntoLines(
maximumAllowableBufferSize: 3,
dropLastChunkIfNoNewline: true
).strings
)
XCTAssertEqual(["\0\0\0\0", "\0\0\0\0", "\0\0"], lines)
}
func testOverlyLongLineIsSplitByDefault() async throws {
var inputs = [ByteBuffer(repeating: UInt8(0), count: 1024 * 1024 - 2)] // almost at the limit
inputs.append(ByteBuffer(integer: UInt8(ascii: "\0")))
inputs.append(ByteBuffer(integer: UInt8(ascii: "\0"))) // hitting the limit
inputs.append(ByteBuffer(integer: UInt8(ascii: "\0"))) // over the limit
inputs.append(ByteBuffer(integer: UInt8(ascii: "\n"))) // too late
let lines = try await Array(
inputs.async.splitIntoLines(
dropTerminator: false,
dropLastChunkIfNoNewline: true
).strings
)
XCTAssertEqual([String(repeating: "\0", count: 1024 * 1024 + 1), "\n"], lines)
}
}
|
