@@ -247,18 +247,29 @@ class ReconnectionTaskSpec extends TestKitBaseClass with FixtureAnyFunSuiteLike
247247 nodeParams.socksProxy_opt returns Some (socksParams)
248248 assert(ReconnectionTask .selectNodeAddress(nodeParams, List (clearnet)).contains(clearnet))
249249 assert(ReconnectionTask .selectNodeAddress(nodeParams, List (tor)).contains(tor))
250- assert(ReconnectionTask .selectNodeAddress(nodeParams, List (clearnet, tor)).contains( clearnet))
250+ assert(ReconnectionTask .selectNodeAddress(nodeParams, List (clearnet, tor)).exists( Set ( clearnet, tor)(_) ))
251251 }
252252 {
253- // tor supported and enabled for clearnet addresses: return tor addresses when available
253+ // tor supported and enabled for clearnet addresses: return both tor and clearnet addresses when available
254254 val socksParams = mock[Socks5ProxyParams ]
255255 socksParams.useForTor returns true
256256 socksParams.useForIPv4 returns true
257257 socksParams.useForIPv6 returns true
258258 nodeParams.socksProxy_opt returns Some (socksParams)
259259 assert(ReconnectionTask .selectNodeAddress(nodeParams, List (clearnet)).contains(clearnet))
260260 assert(ReconnectionTask .selectNodeAddress(nodeParams, List (tor)).contains(tor))
261- assert(ReconnectionTask .selectNodeAddress(nodeParams, List (clearnet, tor)).contains(tor))
261+ assert(ReconnectionTask .selectNodeAddress(nodeParams, List (clearnet, tor)).exists(Set (clearnet, tor)(_)))
262+ }
263+ {
264+ // tor supported and enabled for clearnet addresses, but disabled for tor: return clearnet addresses when available
265+ val socksParams = mock[Socks5ProxyParams ]
266+ socksParams.useForTor returns false
267+ socksParams.useForIPv4 returns true
268+ socksParams.useForIPv6 returns true
269+ nodeParams.socksProxy_opt returns Some (socksParams)
270+ assert(ReconnectionTask .selectNodeAddress(nodeParams, List (clearnet)).contains(clearnet))
271+ assert(ReconnectionTask .selectNodeAddress(nodeParams, List (tor)).isEmpty)
272+ assert(ReconnectionTask .selectNodeAddress(nodeParams, List (clearnet, tor)).contains(clearnet))
262273 }
263274 }
264275
0 commit comments