Refactor custom pod unit tests