Skip to content

Commit 0579462

Browse files
committed
Create test case for testing NetworkGraph anchoring when using ProbabilisticScorer.read, which is not a constructor.
1 parent 6f2617b commit 0579462

File tree

2 files changed

+14
-5
lines changed

2 files changed

+14
-5
lines changed

ci/LDKSwift/Tests/LDKSwiftTests/HumanObjectPeerTestInstance.swift

Lines changed: 11 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -17,10 +17,11 @@ public class HumanObjectPeerTestInstance {
1717

1818
public class Configuration {
1919
//
20-
public var useFilter: Bool = false;
21-
public var useRouter: Bool = false;
22-
public var shouldRecipientRejectPayment: Bool = false;
20+
public var useFilter: Bool = false
21+
public var useRouter: Bool = false
22+
public var shouldRecipientRejectPayment: Bool = false
2323
public var ephemeralNetworkGraphForScorer: Bool = false
24+
public var reserializedProbabilisticScorer: Bool = false
2425

2526
// public var nice_close: Bool = false;
2627
// public var use_km_wrapper: Bool = false;
@@ -30,7 +31,7 @@ public class HumanObjectPeerTestInstance {
3031
// public var use_nio_peer_handler: Bool = false;
3132

3233
private class func listCustomizeableProperties() -> [String] {
33-
return ["useFilter", "useRouter", "shouldRecipientRejectPayment", "ephemeralNetworkGraphForScorer"]
34+
return ["useFilter", "useRouter", "shouldRecipientRejectPayment", "ephemeralNetworkGraphForScorer", "reserializedProbabilisticScorer"]
3435
}
3536

3637
public class func combinationCount() -> UInt {
@@ -308,7 +309,12 @@ public class HumanObjectPeerTestInstance {
308309
}
309310

310311
let scoringParams = ProbabilisticScoringParameters.initWithDefault()
311-
let probabalisticScorer = ProbabilisticScorer(params: scoringParams, networkGraph: scorerGraph, logger: self.logger)
312+
var probabalisticScorer = ProbabilisticScorer(params: scoringParams, networkGraph: scorerGraph, logger: self.logger)
313+
if master.configuration.reserializedProbabilisticScorer {
314+
let serializedScorer = probabalisticScorer.write()
315+
let probabalisticScorerResult = ProbabilisticScorer.read(ser: serializedScorer, argA: scoringParams, argB: scorerGraph, argC: self.logger)
316+
probabalisticScorer = probabalisticScorerResult.getValue()!
317+
}
312318
let score = probabalisticScorer.asScore()
313319
let multiThreadedScorer = MultiThreadedLockableScore(score: score)
314320

ci/LDKSwift/Tests/LDKSwiftTests/LDKSwiftTests.swift

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -472,6 +472,9 @@ class LDKSwiftTests: XCTestCase {
472472

473473
config.ephemeralNetworkGraphForScorer = (i & (1 << 3)) != 0
474474
print("ephemeralNetworkGraphForScorer: \(config.ephemeralNetworkGraphForScorer)")
475+
476+
config.reserializedProbabilisticScorer = (i & (1 << 4)) != 0
477+
print("reserializedProbabilisticScorer: \(config.reserializedProbabilisticScorer)")
475478

476479
let instance = HumanObjectPeerTestInstance(configuration: config)
477480
await instance.testMessageHandling()

0 commit comments

Comments
 (0)