Add more tests for source feature