Refactor and move the publish specific tests to match the new test structure