Rethinking customized classes while unit testing