Add tests for the payment process