Adds test case for @return keyword handling