Adds stage controller unittests.