Changeset 11063

Timestamp:

09/06/2016 03:31:21 AM (10 years ago)

Author:

boonebgorges

Message:

Tests: Add Friendship factory.

Props jdgrimes.
Fixes #7243.

Location:

trunk/tests/phpunit

Files:

• includes/factory.php (modified) (2 diffs)
• testcases/testsuite/factory.php (modified) (1 diff)

trunk/tests/phpunit/includes/factory.php

@@ -14 +14 @@
                 $this->notification = new BP_UnitTest_Factory_For_Notification( $this );
                 $this->signup = new BP_UnitTest_Factory_For_Signup( $this );
+                $this->friendship = new BP_UnitTest_Factory_For_Friendship( $this );
         }
 }
@@ -269 +270 @@
         }
 }
+
+/**
+ * Factory for friendships.
+ *
+ * @since 2.7.0
+ */
+class BP_UnitTest_Factory_For_Friendship extends WP_UnitTest_Factory_For_Thing {
+        /**
+         * Constructor.
+         *
+         * @since 2.7.0
+         *
+         * @param $factory WP_UnitTest_Factory
+         */
+        public function __construct( $factory = null ) {
+                parent::__construct( $factory );
+        }
+
+        /**
+         * Create friendship object.
+         *
+         * @since 2.7.0
+         *
+         * @param array $args Array of arguments.
+         * @return int Friendship ID.
+         */
+        public function create_object( $args ) {
+                $friendship = new BP_Friends_Friendship();
+
+                foreach ( array( 'initiator_user_id', 'friend_user_id' ) as $arg ) {
+                        if ( isset( $args[ $arg ] ) ) {
+                                $friendship->$arg = $args[ $arg ];
+                        } else {
+                                $friendship->$arg = $this->factory->user->create();
+                        }
+                }
+
+                foreach ( array( 'is_confirmed', 'is_limited', 'date_created' ) as $arg ) {
+                        if ( isset( $args[ $arg ] ) ) {
+                                $friendship->$arg = $args[ $arg ];
+                        }
+                }
+
+                $friendship->save();
+
+                return $friendship->id;
+        }
+
+        /**
+         * Update a friendship object.
+         *
+         * @since 2.7.0
+         *
+         * @todo Implement.
+         *
+         * @param int   $id     ID of the friendship.
+         * @param array $fields Fields to update.
+         */
+        public function update_object( $id, $fields ) {}
+
+        /**
+         * Get a friendship object by its ID.
+         *
+         * @since 2.7.0
+         *
+         * @param int $id
+         * @return BP_Friends_Friendship
+         */
+        public function get_object_by_id( $id ) {
+                return new BP_Friends_Friendship( $id );
+        }
+}

trunk/tests/phpunit/testcases/testsuite/factory.php

@@ -23 +23 @@
                 $this->assertNotEmpty( $m->get_recipients() );
         }
+
+        /**
+         * @ticket BP7243
+         */
+        public function test_friendship_should_create_default_initiator_and_friend() {
+                $f = $this->factory->friendship->create_and_get();
+
+                $u1 = new WP_User( $f->initiator_user_id );
+                $u2 = new WP_User( $f->friend_user_id );
+
+                $this->assertTrue( $u1->exists() );
+                $this->assertTrue( $u2->exists() );
+        }
+
+        /**
+         * @ticket BP7243
+         */
+        public function test_friendship_should_respect_passed_initiator_and_friend() {
+                $u1 = $this->factory->user->create();
+                $u2 = $this->factory->user->create();
+
+                $f = $this->factory->friendship->create_and_get( array(
+                        'initiator_user_id' => $u1,
+                        'friend_user_id' => $u2,
+                ) );
+
+                $this->assertSame( $u1, $f->initiator_user_id );
+                $this->assertSame( $u2, $f->friend_user_id );
+        }
 }