Refactor tests to use the new helper.

Refactor setup of tests to use the provision_replication/0 helper.