Update `ensure_connected` for websockets #1209

Job	Run time
find-tests	2s
run (nightly-2024-03-05, x86_64-unknown-linux-gnu, ubuntu-latest, tests/e2e_tests/test_incentive.py)	12m 21s
run (nightly-2024-03-05, x86_64-unknown-linux-gnu, ubuntu-latest, tests/e2e_tests/test_subtensor_...	8m 35s
run (nightly-2024-03-05, x86_64-unknown-linux-gnu, ubuntu-latest, tests/e2e_tests/test_liquid_alp...	9m 3s
run (nightly-2024-03-05, x86_64-unknown-linux-gnu, ubuntu-latest, tests/e2e_tests/test_root_set_w...	9m 54s
run (nightly-2024-03-05, x86_64-unknown-linux-gnu, ubuntu-latest, tests/e2e_tests/test_metagraph.py)	9m 8s
run (nightly-2024-03-05, x86_64-unknown-linux-gnu, ubuntu-latest, tests/e2e_tests/test_neuron_cer...	9m 11s
run (nightly-2024-03-05, x86_64-unknown-linux-gnu, ubuntu-latest, tests/e2e_tests/test_transfer.py)	8m 47s
run (nightly-2024-03-05, x86_64-unknown-linux-gnu, ubuntu-latest, tests/e2e_tests/test_commit_wei...	10m 12s
run (nightly-2024-03-05, x86_64-unknown-linux-gnu, ubuntu-latest, tests/e2e_tests/test_axon.py)	8m 50s
run (nightly-2024-03-05, x86_64-unknown-linux-gnu, ubuntu-latest, tests/e2e_tests/test_dendrite.py)	10m 48s
	1h 36m 51s

Provide feedback