Add tests for modifying the eager-loaded post meta

aj-adl · aj-adl · commit 0ef9f04b46d0 · 2019-05-24T15:37:39.000+09:30
- Test for 'short circuit' where no meta should be eager loaded
- Test for custom list of eager-loaded post meta
diff --git a/tests/test-timber-post.php b/tests/test-timber-post.php
@@ -319,6 +319,47 @@ function testPagedContent(){
 			$this->assertEquals($page2, trim(strip_tags( $post->get_paged_content() )));
 		}
 
+		function testMetaCustomPreFilterDisable(){
+
+			$callable = function(){ return false; };
+
+			add_filter( 'timber_post_get_meta_pre', $callable );
+
+			$post_id = $this->factory->post->create();
+
+			update_post_meta($post_id, 'hidden_value', 'Super secret value');
+
+			$post = new TimberPost($post_id);
+
+			$this->assertCount( 0, $post->custom);
+
+			remove_filter( 'timber_post_get_meta_pre', $callable );
+		}
+
+		function testMetaCustomPreFilterAlter(){
+
+			$callable = function( $customs, $pid, $post ) {
+				$key = 'critical_value';
+
+				return [
+					$key => get_post_meta( $pid, $key ),
+				];
+			};
+
+			add_filter( 'timber_post_get_meta_pre', $callable , 10, 3);
+
+			$post_id = $this->factory->post->create();
+
+			update_post_meta($post_id, 'hidden_value', 'super-big-secret');
+			update_post_meta($post_id, 'critical_value', 'I am needed, all the time');
+
+			$post = new TimberPost($post_id);
+			$this->assertCount( 1, $post->custom );
+			$this->assertEquals( $post->custom, array( 'critical_value' => 'I am needed, all the time' ) );
+
+			remove_filter( 'timber_post_get_meta_pre', $callable );
+		}
+
 		function testMetaCustomArrayFilter(){
 			add_filter('timber_post_get_meta', function($customs){
 				foreach($customs as $key=>$value){
