Add unit tests for VTGate shard_conn reconnect