Ticket #11611: 11611-unit.diff

wp-testcase/test_includes_theme.php

@@ -439 +439 @@
                 $this->assertFalse( current_theme_supports( 'post-thumbnails' ) );
 
         }
+
+        function supports_foobar( $yesno, $args, $feature ) {
+                if ( $args[0] == $feature[0] )
+                        return true;
+                return false;
+        }
+
+        function test_plugin_hook() {
+                $this->assertFalse( current_theme_supports( 'foobar' ) );
+                add_theme_support( 'foobar' );
+                $this->assertTrue( current_theme_supports( 'foobar' ) );
+
+                add_filter( 'current_theme_supports-foobar', array( $this, 'supports_foobar'), 10, 3 );
+
+                add_theme_support( 'foobar', 'bar' );
+                $this->assertFalse( current_theme_supports( 'foobar', 'foo' ) );
+                $this->assertTrue( current_theme_supports( 'foobar', 'bar' ) );
+
+                remove_theme_support( 'foobar' );
+                $this->assertFalse( current_theme_supports( 'foobar', 'bar' ) );
+        }
 }
 ?>