Updated tests for new BaseTaskRunner