Allow author and date_gmt to be supplied in Post_Type::save()

westonruter · westonruter · commit 13f8edfd8813 · 2016-07-23T09:53:58.000-07:00
diff --git a/php/class-post-type.php b/php/class-post-type.php
@@ -531,6 +531,12 @@ public function save( array $args ) {
 			}
 			$post_arr['post_status'] = $args['status'];
 		}
+		if ( ! empty( $args['author'] ) ) {
+			$post_arr['post_author'] = $args['author'];
+		}
+		if ( ! empty( $args['date_gmt'] ) ) {
+			$post_arr['post_date_gmt'] = $args['date_gmt'];
+		}
 
 		$this->suspend_kses();
 		if ( $is_update ) {
diff --git a/tests/php/test-class-post-type.php b/tests/php/test-class-post-type.php
@@ -489,6 +489,25 @@ public function test_save() {
 
 		$this->assertEquals( get_stylesheet(), get_post_meta( $r, '_snapshot_theme', true ) );
 		$this->assertEquals( $this->plugin->version, get_post_meta( $r, '_snapshot_version', true ) );
+
+		// Success with author supplied.
+		$user_id = $this->factory()->user->create( array( 'role' => 'administrator' ) );
+		$post_id = $post_type->save( array(
+			'uuid' => self::UUID,
+			'data' => $data,
+			'status' => 'publish',
+			'author' => $user_id,
+		) );
+		$this->assertEquals( $user_id, get_post( $post_id )->post_author );
+
+		// Success with future date.
+		$post_id = $post_type->save( array(
+			'uuid' => self::UUID,
+			'data' => $data,
+			'status' => 'publish',
+			'date_gmt' => gmdate( 'Y-m-d H:i:s', time() + 24 * 3600 ),
+		) );
+		$this->assertEquals( 'future', get_post_status( $post_id ) );
 	}
 
 	/**

Original file line number	Diff line number	Diff line change
`@@ -531,6 +531,12 @@ public function save( array $args ) {`
`531`	`531`	`}`
`532`	`532`	`$post_arr['post_status'] = $args['status'];`
`533`	`533`	`}`
	`534`	`+ if ( ! empty( $args['author'] ) ) {`
	`535`	`+ $post_arr['post_author'] = $args['author'];`
	`536`	`+ }`
	`537`	`+ if ( ! empty( $args['date_gmt'] ) ) {`
	`538`	`+ $post_arr['post_date_gmt'] = $args['date_gmt'];`
	`539`	`+ }`
`534`	`540`
`535`	`541`	`$this->suspend_kses();`
`536`	`542`	`if ( $is_update ) {`