<?php

class BBP_UnitTest_Factory extends WP_UnitTest_Factory {

	public $forum = null;
	public $topic = null;
	public $reply = null;

	public function __construct() {
		parent::__construct();

		$this->forum = new BBP_UnitTest_Factory_For_Forum( $this );
		$this->topic = new BBP_UnitTest_Factory_For_Topic( $this );
		$this->reply = new BBP_UnitTest_Factory_For_Reply( $this );
	}
}

class BBP_UnitTest_Factory_For_Forum extends WP_UnitTest_Factory_For_Thing {

	public function __construct( $factory = null ) {
		parent::__construct( $factory );

		$this->default_generation_definitions = array(
			'post_title'   => new WP_UnitTest_Generator_Sequence( 'Forum %s' ),
			'post_content' => new WP_UnitTest_Generator_Sequence( 'Content of Forum %s' ),
		);
	}

	public function create_object( $args ) {
		$forum_meta = isset( $args['forum_meta'] ) ? $args['forum_meta'] : array();
		unset( $args['forum_meta'] );
		$forum_data = bbp_insert_forum( $args, $forum_meta );
		return $forum_data;
	}

	public function update_object( $forum_id, $fields ) {
		$fields['forum_id'] = $forum_id;
		return bbp_update_forum( $fields );
	}

	public function get_object_by_id( $forum_id ) {
		return bbp_get_forum( $forum_id );
	}
}

class BBP_UnitTest_Factory_For_Topic extends WP_UnitTest_Factory_For_Thing {

	public function __construct( $factory = null ) {
		parent::__construct( $factory );

		$this->default_generation_definitions = array(
			'post_title'   => new WP_UnitTest_Generator_Sequence( 'Topic %s' ),
			'post_content' => new WP_UnitTest_Generator_Sequence( 'Content of Topic %s' ),
		);
	}

	public function create_object( $args ) {
		$topic_meta = isset( $args['topic_meta'] ) ? $args['topic_meta'] : array();
		unset( $args['topic_meta'] );
		$topic_data = bbp_insert_topic( $args, $topic_meta );
		return $topic_data;
	}

	public function update_object( $topic_id, $fields ) {
		$fields['topic_id'] = $topic_id;
		return bbp_update_topic( $fields );
	}

	public function get_object_by_id( $topic_id ) {
		return bbp_get_topic( $topic_id );
	}
}

class BBP_UnitTest_Factory_For_Reply extends WP_UnitTest_Factory_For_Thing {

	public function __construct( $factory = null ) {
		parent::__construct( $factory );

		$this->default_generation_definitions = array(
			'post_content' => new WP_UnitTest_Generator_Sequence( 'Content of Reply %s' ),
		);
	}

	public function create_object( $args ) {
		$reply_meta = isset( $args['reply_meta'] ) ? $args['reply_meta'] : array();
		unset( $args['reply_meta'] );
		$reply_data = bbp_insert_reply( $args, $reply_meta );
		return $reply_data;
	}

	public function update_object( $reply_id, $fields ) {
		$fields['reply_id'] = $reply_id;
		return bbp_update_reply( $fields );
	}

	public function get_object_by_id( $reply_id ) {
		return bbp_get_reply( $reply_id );
	}
}