70 lines
2.3 KiB
Swift
70 lines
2.3 KiB
Swift
//
|
|
// TorrentPeerCommunicatorStub.swift
|
|
// BitTorrentTests
|
|
//
|
|
// Created by Ben Davis on 16/07/2017.
|
|
// Copyright © 2017 Ben Davis. All rights reserved.
|
|
//
|
|
|
|
@testable import BitTorrent
|
|
|
|
class TorrentPeerCommunicatorStub: TorrentPeerCommunicator {
|
|
|
|
var testConnected: Bool = false
|
|
override var connected: Bool {
|
|
return testConnected
|
|
}
|
|
|
|
var connectCalled = false
|
|
override func connect() throws {
|
|
connectCalled = true
|
|
}
|
|
|
|
var sendHandshakeCalled = false
|
|
var sendHandshakeParameters: (clientId: Data, completion: (()->Void)?)?
|
|
override func sendHandshake(for clientId: Data, _ completion: (() -> Void)?) {
|
|
sendHandshakeCalled = true
|
|
sendHandshakeParameters = (clientId, completion)
|
|
}
|
|
|
|
var sendBitFieldCalled = false
|
|
var sendBitFieldParameters: (bitField: BitField, completion: (()->Void)?)?
|
|
override func sendBitField(_ bitField: BitField, _ completion: (() -> Void)?) {
|
|
sendBitFieldCalled = true
|
|
sendBitFieldParameters = (bitField, completion)
|
|
}
|
|
|
|
var sendInterestedCalled = false
|
|
var sendInterestedParameter: ((()->Void)?)?
|
|
override func sendInterested(_ completion: (() -> Void)?) {
|
|
sendInterestedCalled = true
|
|
sendInterestedParameter = completion
|
|
}
|
|
|
|
var sendRequestCalled = false
|
|
var sendRequestParameters: [(index: Int, begin: Int, length: Int, completion:(()->Void)?)] = []
|
|
override func sendRequest(fromPieceAtIndex index: Int, begin: Int, length: Int, _ completion: (() -> Void)?) {
|
|
sendRequestCalled = true
|
|
sendRequestParameters.append((index, begin, length, completion))
|
|
}
|
|
|
|
var sendKeepAliveCalled = false
|
|
var onSendKeepAliveCalled: (()->Void)?
|
|
override func sendKeepAlive(_ completion: (() -> Void)?) {
|
|
sendKeepAliveCalled = true
|
|
onSendKeepAliveCalled?()
|
|
}
|
|
|
|
var sendPieceCallCount = 0
|
|
var sendPieceParameters: (index: Int, begin: Int, block: Data, completion: (()->Void)?)?
|
|
override func sendPiece(fromPieceAtIndex index: Int, begin: Int, block: Data, _ completion: (() -> Void)?) {
|
|
sendPieceCallCount += 1
|
|
sendPieceParameters = (index, begin, block, completion)
|
|
}
|
|
|
|
var sendUnchokeCalled = false
|
|
override func sendUnchoke(_ completion: (() -> Void)?) {
|
|
sendUnchokeCalled = true
|
|
}
|
|
}
|