Media: Introduce wp_get_attachment_image_context filter.

Since WordPress 5.9, a "context" value of "wp_get_attachment_image" has been used in the `wp_get_attachment_image()` function to provide context to underlying functions where that is relevant, e.g. `wp_get_loading_attr_default()`. Since that value used to be not customizable, it required a workaround in `get_the_post_thumbnail()` to avoid calling those functions in `wp_get_attachment_image()`, which resulted in unnecessary complexity and was prone to errors.

This changeset introduces a `wp_get_attachment_image_context` filter and leverages it with private filter callback functions that are leveraged by default when `get_the_post_thumbnail()` is called. This avoids the need for the previous workaround and furthermore provides flexibility for other callers of `wp_get_attachment_image()` to provide their own contexts.

Props flixos90, costdev, thekt12, westonruter, spacedmonkey.
Fixes #58212.


git-svn-id: https://develop.svn.wordpress.org/trunk@55821 602fd350-edb4-49c9-b593-d223f7449a82
This commit is contained in:
Felix Arntz
2023-05-17 18:29:41 +00:00
parent 261ab3fa5d
commit 5db31259ec
5 changed files with 240 additions and 19 deletions

View File

@@ -3963,6 +3963,62 @@ EOF;
);
}
/**
* Tests that `wp_get_attachment_image()` uses the correct default context.
*
* @ticket 58212
*
* @covers ::wp_get_attachment_image()
*/
public function test_wp_get_attachment_image_context_filter_default() {
$last_context = '';
$this->track_last_attachment_image_context( $last_context );
wp_get_attachment_image( self::$large_id );
$this->assertSame( 'wp_get_attachment_image', $last_context );
}
/**
* Tests that `wp_get_attachment_image()` allows overriding the context via filter.
*
* @ticket 58212
*
* @covers ::wp_get_attachment_image()
*/
public function test_wp_get_attachment_image_context_filter_value_is_passed_correctly() {
$last_context = '';
$this->track_last_attachment_image_context( $last_context );
// Add a filter that modifies the context.
add_filter(
'wp_get_attachment_image_context',
static function() {
return 'my_custom_context';
}
);
wp_get_attachment_image( self::$large_id );
$this->assertSame( 'my_custom_context', $last_context );
}
/**
* Helper method to keep track of the last context returned by the 'wp_get_attachment_image_context' filter.
*
* The method parameter is passed by reference and therefore will always contain the last context value.
*
* @param mixed $last_context Variable to track last context. Passed by reference.
*/
private function track_last_attachment_image_context( &$last_context ) {
add_filter(
'wp_get_attachment_image_context',
static function( $context ) use ( &$last_context ) {
$last_context = $context;
return $context;
},
11
);
}
/**
* Add threshold to create a `-scaled` output image for testing.
*/

View File

@@ -450,6 +450,135 @@ class Tests_Post_Thumbnail_Template extends WP_UnitTestCase {
);
}
/**
* Tests that `_wp_post_thumbnail_context_filter()` returns 'the_post_thumbnail'.
*
* @ticket 58212
*
* @covers::_wp_post_thumbnail_context_filter
*/
public function test_wp_post_thumbnail_context_filter_should_return_the_post_thumbnail() {
$this->assertSame( 'the_post_thumbnail', _wp_post_thumbnail_context_filter( 'wp_get_attachment_image' ) );
}
/**
* Tests that `::_wp_post_thumbnail_context_filter_add` adds a filter to override the context
* used in `wp_get_attachment_image()`.
*
* @ticket 58212
*
* @covers ::_wp_post_thumbnail_context_filter_add
*/
public function test_wp_post_thumbnail_context_filter_add_should_add_the_filter() {
$last_context = '';
$this->track_last_attachment_image_context( $last_context );
_wp_post_thumbnail_context_filter_add();
wp_get_attachment_image( self::$attachment_id );
$this->assertSame( 'the_post_thumbnail', $last_context );
}
/**
* Tests that `_wp_post_thumbnail_context_filter_remove()` removes a filter to override the context
* used in `wp_get_attachment_image()`.
*
* @ticket 58212
*
* @covers ::_wp_post_thumbnail_context_filter_remove
*/
public function test_wp_post_thumbnail_context_filter_remove_should_remove_the_filter() {
$last_context = '';
$this->track_last_attachment_image_context( $last_context );
_wp_post_thumbnail_context_filter_add();
wp_get_attachment_image( self::$attachment_id );
// Verify that the filter has been added before testing that it has been removed.
$this->assertSame(
'the_post_thumbnail',
$last_context,
'The filter was not added.'
);
_wp_post_thumbnail_context_filter_remove();
// The context should no longer be modified by the filter.
wp_get_attachment_image( self::$attachment_id );
$this->assertSame(
'wp_get_attachment_image',
$last_context,
'The filter was not removed.'
);
}
/**
* Tests that `get_the_post_thumbnail()` uses the 'the_post_thumbnail' context.
*
* @ticket 58212
*
* @covers ::get_the_post_thumbnail
*/
public function test_get_the_post_thumbnail_should_use_the_post_thumbnail_context() {
$last_context = '';
$this->track_last_attachment_image_context( $last_context );
set_post_thumbnail( self::$post, self::$attachment_id );
get_the_post_thumbnail( self::$post );
$this->assertSame( 'the_post_thumbnail', $last_context );
}
/**
* Tests that `get_the_post_thumbnail()` restores the context afterwards.
*
* @ticket 58212
*
* @covers ::get_the_post_thumbnail
*/
public function test_get_the_post_thumbnail_should_remove_the_post_thumbnail_context_afterwards() {
$last_context = '';
$this->track_last_attachment_image_context( $last_context );
set_post_thumbnail( self::$post, self::$attachment_id );
get_the_post_thumbnail( self::$post );
// Verify that the context was overridden before testing that it has been restored.
$this->assertSame(
'the_post_thumbnail',
$last_context,
'The context was not overridden.'
);
// The context should no longer be overridden.
wp_get_attachment_image( self::$attachment_id );
$this->assertSame(
'wp_get_attachment_image',
$last_context,
'The context was not restored.'
);
}
/**
* Helper method to keep track of the last context returned by the 'wp_get_attachment_image_context' filter.
*
* The method parameter is passed by reference and therefore will always contain the last context value.
*
* @param mixed $last_context Variable to track last context. Passed by reference.
*/
private function track_last_attachment_image_context( &$last_context ) {
add_filter(
'wp_get_attachment_image_context',
static function( $context ) use ( &$last_context ) {
$last_context = $context;
return $context;
},
11
);
}
public function filter_post_thumbnail_size( $size, $post_id ) {
if ( is_array( $this->current_size_filter_data ) && isset( $this->current_size_filter_data[ $post_id ] ) ) {
return $this->current_size_filter_data[ $post_id ];