PDI-11072 - Adding register logic to MockDriver for unit tests