41 lines
1.5 KiB
Swift
41 lines
1.5 KiB
Swift
//===----------------------------------------------------------------------===//
|
|
//
|
|
// 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 NIOCore
|
|
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")
|
|
}
|
|
|
|
func testRandomMaskingKeyIsNotAlwaysZeroWithDefaultGenerator() {
|
|
XCTAssertTrue((0..<1000).contains { _ in
|
|
WebSocketMaskingKey.random() != [0, 0, 0, 0]
|
|
}, "at least 1 of 1000 random masking keys with default generator should not be all zeros")
|
|
}
|
|
}
|