Use mock connections instead of real ones in tests