Add Unit Test for @Source