We should probably do this in the component mock itself and not in the rootComponent mock.
However, that's not that easy without a proper mock library for components... 
The options I see are:

1) Keep this PR as it is.
2) Have an optional argument to component() with the desired rootComponent and default to one created just for that purpose.