Changeset 11063

Timestamp:

09/06/2016 03:31:21 AM (5 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 );
+    }
 }