working test using WP_Mock