Restore behaviour of ReferenceList::removeReferenceHash

Author: Benestar

src/ReferenceList.php

@@ -117,7 +117,7 @@ public function hasReference( Reference $reference ) {
 	}
 
 	/**
-	 * Returns the index of a reference or false if the reference could not be found.
+	 * Returns the index of the Reference object or false if the Reference could not be found.
 	 *
 	 * @since 0.5
 	 *
@@ -160,29 +160,31 @@ public function hasReferenceHash( $referenceHash ) {
 	}
 
 	/**
-	 * Removes the reference with the provided hash if it exists in the list.
+	 * Looks for the first Reference object in this list with the provided hash.
+	 * Removes all occurences of that object.
 	 *
 	 * @since 0.3
 	 *
 	 * @param string $referenceHash	`
 	 */
 	public function removeReferenceHash( $referenceHash ) {
-		foreach ( $this->references as $index => $reference ) {
-			if ( $reference->getHash() === $referenceHash ) {
-				$this->removeReferenceAtIndex( $index );
+		$reference = $this->getReference( $referenceHash );
+
+		if ( $reference === null ) {
+			return;
+		}
+
+		foreach ( $this->references as $index => $ref ) {
+			if ( $ref === $reference ) {
+				unset( $this->references[$index] );
 			}
 		}
-	}
 
-	/**
-	 * @param int $index
-	 */
-	private function removeReferenceAtIndex( $index ) {
-		array_splice( $this->references, $index, 1 );
+		$this->references = array_values( $this->references );
 	}
 
 	/**
-	 * Returns the reference with the provided hash, or
+	 * Returns the first Reference object with the provided hash, or
 	 * null if there is no such reference in the list.
 	 *
 	 * @since 0.3

tests/unit/ReferenceListTest.php

@@ -330,6 +330,21 @@ public function testRemoveReferenceHash( ReferenceList $references ) {
 		$this->assertEquals( 0, count( $references ) );
 	}
 
+	public function testRemoveReferenceHash_occurrencesOfSameObjectGetRemoved() {
+		$reference = new Reference( array( new PropertyNoValueSnak( 42 ) ) );
+		$references = new ReferenceList( array(
+			$reference,
+			clone $reference,
+			$reference
+		) );
+
+		$references->removeReferenceHash( $reference->getHash() );
+
+		$this->assertTrue( $references->hasReferenceHash( $reference->getHash() ) );
+		$this->assertCount( 1, $references );
+		$this->assertNotSame( $reference, $references->getReference( $reference->getHash() ) );
+	}
+
 	public function testGivenOneSnak_addNewReferenceAddsSnak() {
 		$references = new ReferenceList();
 		$snak = new PropertyNoValueSnak( 1 );