Add Source model tests