Properly mocks more of the wp-core methods used