106 lines
3.5 KiB
Swift
106 lines
3.5 KiB
Swift
//
|
|
// Copyright 2022 New Vector Ltd
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
//
|
|
|
|
import XCTest
|
|
@testable import Element
|
|
|
|
@MainActor
|
|
class RendezvousServiceTests: XCTestCase {
|
|
func testEnd2EndV1() async {
|
|
let mockTransport = MockRendezvousTransport()
|
|
|
|
let aliceService = RendezvousService(transport: mockTransport, algorithm: .ECDH_V1)
|
|
|
|
guard case .success(let rendezvousDetails) = await aliceService.createRendezvous(),
|
|
let alicePublicKey = rendezvousDetails.key else {
|
|
XCTFail("Rendezvous creation failed")
|
|
return
|
|
}
|
|
|
|
XCTAssertNotNil(mockTransport.rendezvousURL)
|
|
|
|
let bobService = RendezvousService(transport: mockTransport, algorithm: .ECDH_V1)
|
|
|
|
guard case .success = await bobService.joinRendezvous(withPublicKey: alicePublicKey) else {
|
|
XCTFail("Bob failed to join")
|
|
return
|
|
}
|
|
|
|
guard case .success = await aliceService.waitForInterlocutor() else {
|
|
XCTFail("Alice failed to establish connection")
|
|
return
|
|
}
|
|
|
|
guard let messageData = "Hello from alice".data(using: .utf8) else {
|
|
fatalError()
|
|
}
|
|
|
|
guard case .success = await aliceService.send(data: messageData) else {
|
|
XCTFail("Alice failed to send message")
|
|
return
|
|
}
|
|
|
|
guard case .success(let data) = await bobService.receive() else {
|
|
XCTFail("Bob failed to receive message")
|
|
return
|
|
}
|
|
|
|
XCTAssertEqual(messageData, data)
|
|
}
|
|
|
|
func testEnd2EndV2() async {
|
|
let mockTransport = MockRendezvousTransport()
|
|
|
|
let aliceService = RendezvousService(transport: mockTransport, algorithm: .ECDH_V2)
|
|
|
|
guard case .success(let rendezvousDetails) = await aliceService.createRendezvous(),
|
|
let alicePublicKey = rendezvousDetails.key else {
|
|
XCTFail("Rendezvous creation failed")
|
|
return
|
|
}
|
|
|
|
XCTAssertNotNil(mockTransport.rendezvousURL)
|
|
|
|
let bobService = RendezvousService(transport: mockTransport, algorithm: .ECDH_V2)
|
|
|
|
guard case .success = await bobService.joinRendezvous(withPublicKey: alicePublicKey) else {
|
|
XCTFail("Bob failed to join")
|
|
return
|
|
}
|
|
|
|
guard case .success = await aliceService.waitForInterlocutor() else {
|
|
XCTFail("Alice failed to establish connection")
|
|
return
|
|
}
|
|
|
|
guard let messageData = "Hello from alice".data(using: .utf8) else {
|
|
fatalError()
|
|
}
|
|
|
|
guard case .success = await aliceService.send(data: messageData) else {
|
|
XCTFail("Alice failed to send message")
|
|
return
|
|
}
|
|
|
|
guard case .success(let data) = await bobService.receive() else {
|
|
XCTFail("Bob failed to receive message")
|
|
return
|
|
}
|
|
|
|
XCTAssertEqual(messageData, data)
|
|
}
|
|
}
|