swarm/network: Rename check back to assert now that we have helper magic

This commit is contained in:
lash 2018-12-19 12:02:45 +01:00
parent 78905f5514
commit a5953f6b75

View file

@ -168,65 +168,65 @@ func TestHealthStrict(t *testing.T) {
// no peers // no peers
// unhealthy (and lonely) // unhealthy (and lonely)
k := newTestKademlia("11111111") k := newTestKademlia("11111111")
checkHealth(t, k, false, false) assertHealth(t, k, false, false)
// know one peer but not connected // know one peer but not connected
// unhealthy // unhealthy
Register(k, "11100000") Register(k, "11100000")
log.Trace(k.String()) log.Trace(k.String())
checkHealth(t, k, false, false) assertHealth(t, k, false, false)
// know one peer and connected // know one peer and connected
// healthy // healthy
On(k, "11100000") On(k, "11100000")
checkHealth(t, k, true, false) assertHealth(t, k, true, false)
// know two peers, only one connected // know two peers, only one connected
// unhealthy // unhealthy
Register(k, "11111100") Register(k, "11111100")
log.Trace(k.String()) log.Trace(k.String())
checkHealth(t, k, false, false) assertHealth(t, k, false, false)
// know two peers and connected to both // know two peers and connected to both
// healthy // healthy
On(k, "11111100") On(k, "11111100")
checkHealth(t, k, true, false) assertHealth(t, k, true, false)
// know three peers, connected to the two deepest // know three peers, connected to the two deepest
// healthy // healthy
Register(k, "00000000") Register(k, "00000000")
log.Trace(k.String()) log.Trace(k.String())
checkHealth(t, k, true, false) assertHealth(t, k, true, false)
// know three peers, connected to all three // know three peers, connected to all three
// healthy // healthy
On(k, "00000000") On(k, "00000000")
checkHealth(t, k, true, false) assertHealth(t, k, true, false)
// add fourth peer deeper than current depth // add fourth peer deeper than current depth
// unhealthy // unhealthy
Register(k, "11110000") Register(k, "11110000")
log.Trace(k.String()) log.Trace(k.String())
checkHealth(t, k, false, false) assertHealth(t, k, false, false)
// connected to three deepest peers // connected to three deepest peers
// healthy // healthy
On(k, "11110000") On(k, "11110000")
checkHealth(t, k, true, false) assertHealth(t, k, true, false)
// add additional peer in same bin as deepest peer // add additional peer in same bin as deepest peer
// unhealthy // unhealthy
Register(k, "11111101") Register(k, "11111101")
log.Trace(k.String()) log.Trace(k.String())
checkHealth(t, k, false, false) assertHealth(t, k, false, false)
// four deepest of five peers connected // four deepest of five peers connected
// healthy // healthy
On(k, "11111101") On(k, "11111101")
checkHealth(t, k, true, false) assertHealth(t, k, true, false)
} }
func checkHealth(t *testing.T, k *Kademlia, expectHealthy bool, expectSaturation bool) { func assertHealth(t *testing.T, k *Kademlia, expectHealthy bool, expectSaturation bool) {
t.Helper() t.Helper()
kid := common.Bytes2Hex(k.BaseAddr()) kid := common.Bytes2Hex(k.BaseAddr())
addrs := [][]byte{k.BaseAddr()} addrs := [][]byte{k.BaseAddr()}