Add test for total_quantity method