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
|
//===----------------------------------------------------------------------===//
//
// This source file is part of the SwiftNIO open source project
//
// Copyright (c) 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 XCTest
import NIOWebSocket
/// a mock random number generator which will return the given `numbers` in order
fileprivate struct TestRandomNumberGenerator: RandomNumberGenerator {
var numbers: [UInt64]
var nextRandomNumberIndex: Int
init(numbers: [UInt64], nextRandomNumberIndex: Int = 0) {
self.numbers = numbers
self.nextRandomNumberIndex = nextRandomNumberIndex
}
mutating func next() -> UInt64 {
defer { nextRandomNumberIndex += 1 }
return numbers[nextRandomNumberIndex % numbers.count]
}
}
final class NIOWebSocketClientUpgraderTests: XCTestCase {
func testRandomRequestKey() {
var generator = TestRandomNumberGenerator(numbers: [10, 11])
let requestKey = NIOWebSocketClientUpgrader.randomRequestKey(using: &generator)
XCTAssertEqual(requestKey, "AAAAAAAAAAoAAAAAAAAACw==")
}
func testRandomRequestKeyWithSystemRandomNumberGenerator() {
XCTAssertEqual(NIOWebSocketClientUpgrader.randomRequestKey().count, 24, "request key must be exactly 16 bytes long and this corresponds to 24 characters in base64")
}
}
|
