This commit is contained in:
John Bintz 2010-02-02 21:29:09 -05:00
parent c24214afff
commit 8e3108f1f9
2 changed files with 57 additions and 0 deletions

View File

@ -22,6 +22,23 @@ class ComicPressTagBuilderFactory {
$a = new ComicPressTagBuilder($post, $this->storyline, $this->dbi); $a = new ComicPressTagBuilder($post, $this->storyline, $this->dbi);
return call_user_func_array(array($a, $method), $arguments); return call_user_func_array(array($a, $method), $arguments);
} }
public function post_in($category, $override_post = null) {
global $post;
$post_to_use = empty($override_post) ? $post : $override_post;
$is_in = false;
$categories = wp_get_post_categories($post_to_use->ID);
if (is_array($categories)) {
$children = $this->storyline->_find_children($category);
if (is_array($children)) {
$is_in = count(array_intersect($children, $categories)) > 0;
}
}
return $is_in;
}
} }
class ComicPressTagBuilder { class ComicPressTagBuilder {

View File

@ -116,6 +116,46 @@ class ComicPressTagBuilderTest extends PHPUnit_Framework_TestCase {
} }
} }
function providerTestPostIn() {
return array(
array(null, false),
array((object)array('ID' => 2), true)
);
}
/**
* @dataProvider providerTestPostIn
*/
function testPostIn($post_to_use = null, $expected_result) {
global $post;
$post = (object)array('ID' => 1);
wp_insert_post($post);
$storyline = new ComicPressStoryline();
$storyline->set_flattened_storyline('0/1,0/2,0/2/3,0/2/4,0/5');
foreach (array(
1 => array('cat_name' => 'Test 1', 'category_nicename' => 'category-1', 'category_parent' => 0),
2 => array('cat_name' => 'Test 2', 'category_nicename' => 'category-2', 'category_parent' => 0),
3 => array('cat_name' => 'Test 3', 'category_nicename' => 'category-3', 'category_parent' => 2),
4 => array('cat_name' => 'Test 4', 'category_nicename' => 'category-4', 'category_parent' => 2),
5 => array('cat_name' => 'Test 5', 'category_nicename' => 'category-5', 'category_parent' => 0),
) as $id => $category) {
add_category($id, (object)$category);
}
wp_set_post_categories(1, array(1));
wp_set_post_categories(2, array(2));
$dbi = $this->getMock('ComicPressDBInterface');
$core = new ComicPressTagBuilderFactory($dbi);
$post_to_use = empty($post_to_use) ? $post : $post_to_use;
$this->assertEquals($expected_result, $core->post_in('category-2', $post_to_use));
}
private function setupStorylineBuilderTest($for_exceptions = false) { private function setupStorylineBuilderTest($for_exceptions = false) {
$target_post = (object)array( $target_post = (object)array(
'ID' => 1, 'ID' => 1,