Ticket #6188: 6188.2.tests.diff

tests/phpunit/testcases/members/types.php

@@ -235 +235 @@
 
                 $this->assertSame( 'foo', bp_get_member_type( $new_user_id, true ) );
         }
+
+        /**
+         * @group bp_remove_member_type
+         */
+        public function test_bp_remove_member_type_remove_signle_type() {
+                $u = $this->factory->user->create();
+                bp_register_member_type( 'foo' );
+                bp_register_member_type( 'bar' );
+                bp_set_member_type( $u, 'foo' );
+                bp_set_member_type( $u, 'bar', true );
+
+                // Remove type
+                bp_remove_member_type( $u, 'foo' );
+
+                $this->assertSame( 'bar', bp_get_member_type( $u, true ) );
+        }
+
+        /**
+         * @group bp_remove_member_type
+         */
+        public function test_bp_remove_member_type_non_existing_type_for_user() {
+                $u = $this->factory->user->create();
+                bp_register_member_type( 'foo' );
+                bp_register_member_type( 'bar' );
+                bp_set_member_type( $u, 'foo' );
+
+                $this->assertFalse( bp_remove_member_type( $u, 'bar' ) );
+        }
 }