unit test only working with a traffic light mock