Add test for upload functionality