Add test for before_save callback