Updated tests for portfolio controller