106 lines
3.6 KiB
PHP
106 lines
3.6 KiB
PHP
|
<?php
|
||
|
|
|
||
|
|
abstract class WP_UnitTest_Factory_For_Thing {
|
||
|
|
|
||
|
|
var $default_generation_definitions;
|
||
|
|
var $factory;
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Creates a new factory, which will create objects of a specific Thing
|
||
|
|
*
|
||
|
|
* @param object $factory Global factory that can be used to create other objects on the system
|
||
|
|
* @param array $default_generation_definitions Defines what default values should the properties of the object have. The default values
|
||
|
|
* can be generators -- an object with next() method. There are some default generators: {@link WP_UnitTest_Generator_Sequence},
|
||
|
|
* {@link WP_UnitTest_Generator_Locale_Name}, {@link WP_UnitTest_Factory_Callback_After_Create}.
|
||
|
|
*/
|
||
|
|
function __construct( $factory, $default_generation_definitions = array() ) {
|
||
|
|
$this->factory = $factory;
|
||
|
|
$this->default_generation_definitions = $default_generation_definitions;
|
||
|
|
}
|
||
|
|
|
||
|
|
abstract function create_object( $args );
|
||
|
|
abstract function update_object( $object, $fields );
|
||
|
|
|
||
|
|
function create( $args = array(), $generation_definitions = null ) {
|
||
|
|
if ( is_null( $generation_definitions ) )
|
||
|
|
$generation_definitions = $this->default_generation_definitions;
|
||
|
|
|
||
|
|
$generated_args = $this->generate_args( $args, $generation_definitions, $callbacks );
|
||
|
|
$created = $this->create_object( $generated_args );
|
||
|
|
if ( !$created || is_wp_error( $created ) )
|
||
|
|
return $created;
|
||
|
|
|
||
|
|
if ( $callbacks ) {
|
||
|
|
$updated_fields = $this->apply_callbacks( $callbacks, $created );
|
||
|
|
$save_result = $this->update_object( $created, $updated_fields );
|
||
|
|
if ( !$save_result || is_wp_error( $save_result ) )
|
||
|
|
return $save_result;
|
||
|
|
}
|
||
|
|
return $created;
|
||
|
|
}
|
||
|
|
|
||
|
|
function create_and_get( $args = array(), $generation_definitions = null ) {
|
||
|
|
$object_id = $this->create( $args, $generation_definitions );
|
||
|
|
return $this->get_object_by_id( $object_id );
|
||
|
|
}
|
||
|
|
|
||
|
|
abstract function get_object_by_id( $object_id );
|
||
|
|
|
||
|
|
function create_many( $count, $args = array(), $generation_definitions = null ) {
|
||
|
|
$results = array();
|
||
|
|
for ( $i = 0; $i < $count; $i++ ) {
|
||
|
|
$results[] = $this->create( $args, $generation_definitions );
|
||
|
|
}
|
||
|
|
return $results;
|
||
|
|
}
|
||
|
|
|
||
|
|
function generate_args( $args = array(), $generation_definitions = null, &$callbacks = null ) {
|
||
|
|
$callbacks = array();
|
||
|
|
if ( is_null( $generation_definitions ) )
|
||
|
|
$generation_definitions = $this->default_generation_definitions;
|
||
|
|
|
||
|
|
foreach( array_keys( $generation_definitions ) as $field_name ) {
|
||
|
|
if ( !isset( $args[$field_name] ) ) {
|
||
|
|
$generator = $generation_definitions[$field_name];
|
||
|
|
if ( is_scalar( $generator ) )
|
||
|
|
$args[$field_name] = $generator;
|
||
|
|
elseif ( is_object( $generator ) && method_exists( $generator, 'call' ) ) {
|
||
|
|
$callbacks[$field_name] = $generator;
|
||
|
|
} elseif ( is_object( $generator ) )
|
||
|
|
$args[$field_name] = $generator->next();
|
||
|
|
else
|
||
|
|
return new WP_Error( 'invalid_argument', 'Factory default value should be either a scalar or an generator object.' );
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return $args;
|
||
|
|
}
|
||
|
|
|
||
|
|
function apply_callbacks( $callbacks, $created ) {
|
||
|
|
$updated_fields = array();
|
||
|
|
foreach( $callbacks as $field_name => $generator ) {
|
||
|
|
$updated_fields[$field_name] = $generator->call( $created );
|
||
|
|
}
|
||
|
|
return $updated_fields;
|
||
|
|
}
|
||
|
|
|
||
|
|
function callback( $function ) {
|
||
|
|
return new WP_UnitTest_Factory_Callback_After_Create( $function );
|
||
|
|
}
|
||
|
|
|
||
|
|
function addslashes_deep($value) {
|
||
|
|
if ( is_array( $value ) ) {
|
||
|
|
$value = array_map( array( $this, 'addslashes_deep' ), $value );
|
||
|
|
} elseif ( is_object( $value ) ) {
|
||
|
|
$vars = get_object_vars( $value );
|
||
|
|
foreach ($vars as $key=>$data) {
|
||
|
|
$value->{$key} = $this->addslashes_deep( $data );
|
||
|
|
}
|
||
|
|
} elseif ( is_string( $value ) ) {
|
||
|
|
$value = addslashes( $value );
|
||
|
|
}
|
||
|
|
|
||
|
|
return $value;
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|