Merge pull request #3627 from rchavik/3.0-associations-remove-all

ORM\Associations: Add removeAll() method
cakephp · Jun 3, 2014 · 2273621 · 2273621
2 parents 6f3a1c4 + 74db983
commit 2273621
Show file tree

Hide file tree

Showing 2 changed files with 30 additions and 0 deletions.
diff --git a/src/ORM/Associations.php b/src/ORM/Associations.php
@@ -122,6 +122,19 @@ public function remove($alias) {
 		unset($this->_items[strtolower($alias)]);
 	}
 
+/**
+ * Remove all registered associations.
+ *
+ * Once removed associations will not longer be reachable
+ *
+ * @return void
+ */
+	public function removeAll() {
+		foreach ($this->_items as $alias => $object) {
+			$this->remove($alias);
+		}
+	}
+
 /**
  * Save all the associations that are parents of the given entity.
  *

diff --git a/tests/TestCase/ORM/AssociationsTest.php b/tests/TestCase/ORM/AssociationsTest.php
@@ -63,6 +63,23 @@ public function testAddHasRemoveAndGet() {
 		$this->assertNull($this->associations->get('Users'));
 	}
 
+/**
+ * Test removeAll method
+ *
+ * @return void
+ */
+	public function testRemoveAll() {
+		$this->assertEmpty($this->associations->keys());
+
+		$belongsTo = new BelongsTo([]);
+		$this->assertSame($belongsTo, $this->associations->add('Users', $belongsTo));
+		$belongsToMany = new BelongsToMany([]);
+		$this->assertSame($belongsToMany, $this->associations->add('Cart', $belongsToMany));
+
+		$this->associations->removeAll();
+		$this->assertEmpty($this->associations->keys());
+	}
+
 /**
  * Test getting associations by property.
  *