Fix bad close handshake logic

This doesn't affect real world applications due to buffering but
the testss would occasionally fail on CI due to the code not handling
an immediate close after writing the close frame while resetting
the write timeout.