Add test for source parameter