190 lines
6.8 KiB
Swift
190 lines
6.8 KiB
Swift
//
|
|
// BlockStreamingTest.swift
|
|
// ZcashLightClientKit-Unit-Tests
|
|
//
|
|
// Created by Francisco Gindre on 5/25/21.
|
|
//
|
|
|
|
import XCTest
|
|
@testable import TestUtils
|
|
@testable import ZcashLightClientKit
|
|
|
|
class BlockStreamingTest: XCTestCase {
|
|
let testFileManager = FileManager()
|
|
var rustBackend: ZcashRustBackendWelding!
|
|
var testTempDirectory: URL!
|
|
|
|
override func setUp() async throws {
|
|
try await super.setUp()
|
|
logger = OSLogger(logLevel: .debug)
|
|
testTempDirectory = Environment.uniqueTestTempDirectory
|
|
|
|
try self.testFileManager.createDirectory(at: testTempDirectory, withIntermediateDirectories: false)
|
|
rustBackend = ZcashRustBackend.makeForTests(fsBlockDbRoot: testTempDirectory, networkType: .testnet)
|
|
}
|
|
|
|
override func tearDownWithError() throws {
|
|
try super.tearDownWithError()
|
|
rustBackend = nil
|
|
try? FileManager.default.removeItem(at: __dataDbURL())
|
|
try? testFileManager.removeItem(at: testTempDirectory)
|
|
testTempDirectory = nil
|
|
}
|
|
|
|
func testStream() async throws {
|
|
let endpoint = LightWalletEndpoint(
|
|
address: LightWalletEndpointBuilder.eccTestnet.host,
|
|
port: 9067,
|
|
secure: true,
|
|
singleCallTimeoutInMillis: 1000,
|
|
streamingCallTimeoutInMillis: 100000
|
|
)
|
|
let service = LightWalletServiceFactory(endpoint: endpoint).make()
|
|
|
|
let latestHeight = try await service.latestBlockHeight()
|
|
|
|
let startHeight = latestHeight - 100_000
|
|
var blocks: [ZcashCompactBlock] = []
|
|
let stream = service.blockStream(startHeight: startHeight, endHeight: latestHeight)
|
|
|
|
do {
|
|
for try await compactBlock in stream {
|
|
print("received block \(compactBlock.height)")
|
|
blocks.append(compactBlock)
|
|
print("progressHeight: \(compactBlock.height)")
|
|
print("startHeight: \(startHeight)")
|
|
print("targetHeight: \(latestHeight)")
|
|
}
|
|
} catch {
|
|
XCTFail("failed with error: \(error)")
|
|
}
|
|
}
|
|
|
|
func testStreamCancellation() async throws {
|
|
let endpoint = LightWalletEndpoint(
|
|
address: LightWalletEndpointBuilder.eccTestnet.host,
|
|
port: 9067,
|
|
secure: true,
|
|
singleCallTimeoutInMillis: 10000,
|
|
streamingCallTimeoutInMillis: 10000
|
|
)
|
|
let service = LightWalletServiceFactory(endpoint: endpoint).make()
|
|
|
|
let storage = FSCompactBlockRepository(
|
|
fsBlockDbRoot: testTempDirectory,
|
|
metadataStore: FSMetadataStore.live(
|
|
fsBlockDbRoot: testTempDirectory,
|
|
rustBackend: rustBackend,
|
|
logger: logger
|
|
),
|
|
blockDescriptor: .live,
|
|
contentProvider: DirectoryListingProviders.defaultSorted,
|
|
logger: logger
|
|
)
|
|
|
|
try await storage.create()
|
|
|
|
let latestBlockHeight = try await service.latestBlockHeight()
|
|
let startHeight = latestBlockHeight - 100_000
|
|
let processorConfig = CompactBlockProcessor.Configuration.standard(
|
|
for: ZcashNetworkBuilder.network(for: .testnet),
|
|
walletBirthday: ZcashNetworkBuilder.network(for: .testnet).constants.saplingActivationHeight
|
|
)
|
|
|
|
let compactBlockProcessor = CompactBlockProcessor(
|
|
service: service,
|
|
storage: storage,
|
|
rustBackend: rustBackend,
|
|
config: processorConfig,
|
|
metrics: SDKMetrics(),
|
|
logger: logger,
|
|
latestBlocksDataProvider: LatestBlocksDataProviderMock()
|
|
)
|
|
|
|
let cancelableTask = Task {
|
|
do {
|
|
let blockDownloader = await compactBlockProcessor.blockDownloader
|
|
await blockDownloader.setDownloadLimit(latestBlockHeight)
|
|
await blockDownloader.startDownload(maxBlockBufferSize: 10, syncRange: startHeight...latestBlockHeight)
|
|
try await blockDownloader.waitUntilRequestedBlocksAreDownloaded(in: startHeight...latestBlockHeight)
|
|
} catch {
|
|
XCTAssertTrue(Task.isCancelled)
|
|
}
|
|
}
|
|
|
|
cancelableTask.cancel()
|
|
await compactBlockProcessor.stop()
|
|
}
|
|
|
|
func testStreamTimeout() async throws {
|
|
let endpoint = LightWalletEndpoint(
|
|
address: LightWalletEndpointBuilder.eccTestnet.host,
|
|
port: 9067,
|
|
secure: true,
|
|
singleCallTimeoutInMillis: 1000,
|
|
streamingCallTimeoutInMillis: 1000
|
|
)
|
|
let service = LightWalletServiceFactory(endpoint: endpoint).make()
|
|
|
|
let storage = FSCompactBlockRepository(
|
|
fsBlockDbRoot: testTempDirectory,
|
|
metadataStore: FSMetadataStore.live(
|
|
fsBlockDbRoot: testTempDirectory,
|
|
rustBackend: rustBackend,
|
|
logger: logger
|
|
),
|
|
blockDescriptor: .live,
|
|
contentProvider: DirectoryListingProviders.defaultSorted,
|
|
logger: logger
|
|
)
|
|
|
|
try await storage.create()
|
|
|
|
let latestBlockHeight = try await service.latestBlockHeight()
|
|
|
|
let startHeight = latestBlockHeight - 100_000
|
|
|
|
let processorConfig = CompactBlockProcessor.Configuration.standard(
|
|
for: ZcashNetworkBuilder.network(for: .testnet),
|
|
walletBirthday: ZcashNetworkBuilder.network(for: .testnet).constants.saplingActivationHeight
|
|
)
|
|
|
|
let compactBlockProcessor = CompactBlockProcessor(
|
|
service: service,
|
|
storage: storage,
|
|
rustBackend: rustBackend,
|
|
config: processorConfig,
|
|
metrics: SDKMetrics(),
|
|
logger: logger,
|
|
latestBlocksDataProvider: LatestBlocksDataProviderMock()
|
|
)
|
|
|
|
let date = Date()
|
|
|
|
do {
|
|
let blockDownloader = await compactBlockProcessor.blockDownloader
|
|
await blockDownloader.setDownloadLimit(latestBlockHeight)
|
|
await blockDownloader.startDownload(maxBlockBufferSize: 10, syncRange: startHeight...latestBlockHeight)
|
|
try await blockDownloader.waitUntilRequestedBlocksAreDownloaded(in: startHeight...latestBlockHeight)
|
|
} catch {
|
|
if let lwdError = error as? ZcashError {
|
|
switch lwdError {
|
|
case .serviceBlockStreamFailed:
|
|
XCTAssert(true)
|
|
default:
|
|
XCTFail("LWD Service error found, but should have been a timeLimit reached \(lwdError)")
|
|
}
|
|
} else {
|
|
XCTFail("Error should have been a timeLimit reached Error")
|
|
}
|
|
}
|
|
|
|
let now = Date()
|
|
|
|
let elapsed = now.timeIntervalSince(date)
|
|
print("took \(elapsed) seconds")
|
|
|
|
await compactBlockProcessor.stop()
|
|
}
|
|
}
|