Updated unit tests to work with new preRun / postRun functionality