Add ability to test the location_down behavior.

Assert that we properly handle retries of failed partitions, when the remote location happens to be down.