This still looks like more test code than is really necessary to test the changes. You simply need to verify that 1) the subclass version of the method is called, and 2) the return value is used. Could your tests just test that?