[RIC-475] Enhance LostConnection flow + UTs