Cleanup unit tests by introducing "assert_called" assertion into base classes.