Add test for customer amount method