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
|
//===----------------------------------------------------------------------===//
//
// 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 NIO
import NIOWebSocket
final class WebSocketMaskingKeyTests: XCTestCase {
var generator = SystemRandomNumberGenerator()
func testRandomMaskingKey() {
let key = WebSocketMaskingKey.random(using: &generator)
var buffer = ByteBuffer(bytes: [1, 2, 3, 4, 5, 6, 7, 8])
buffer.webSocketMask(key)
buffer.webSocketUnmask(key)
XCTAssertEqual(buffer, ByteBuffer(bytes: [1, 2, 3, 4, 5, 6, 7, 8]))
}
func testRandomMaskingKeyIsNotAlwaysZero() {
XCTAssertTrue((0..<1000).contains { _ in
WebSocketMaskingKey.random(using: &generator) != [0, 0, 0, 0]
}, "at least 1 of 1000 random masking keys should not be all zeros")
}
}
|
