John Cantrell

> Thanks! I think we also need to move the `peer_lock.their_features` call up - we only call `peer_disconnected` if that line has been hit (`Peer::handshake_complete` checks for it) so we...

> Would be nice to get a test (which should be pretty easy), but either way LGTM. Doesn't look like there's easy way to handle testing it with the existing...

> Yea, I was figuring you'd just create a trivial `CustomMessageHandler` that asserts that connected/disconnecteds all come in order and then errors on connection. Hm, using a CustomMessageHandler doesn't _really_...

@TheBlueMatt Added a test that passes but it duplicates a ton of code to handle all of the setup but with the new message handlers :| not sure if this...