[2.0] added Collection object creation

tests/Doctrine/Tests/Models/ECommerce/ECommerceCart.php

@@ -20,7 +20,7 @@ class ECommerceCart
     private $id;
 
     /**
-     * @Column(type="string", length=50)
+     * @Column(type="string", length=50, nullable=true)
      */
     private $payment;
 
@@ -38,6 +38,11 @@ class ECommerceCart
      */
     private $products;
 
+    public function __construct()
+    {
+        $this->products = new \Doctrine\Common\Collections\Collection;
+    }
+    
     public function getId() {
         return $this->id;
     }
@@ -69,4 +74,12 @@ class ECommerceCart
         return $this->customer;
     }
 
+    public function getProducts()
+    {
+        return $this->products;
+    }
+
+    public function addProduct(ECommerceProduct $product) {
+        $this->products[] = $product;
+    }
 }

tests/Doctrine/Tests/Models/ECommerce/ECommerceCustomer.php

@@ -41,11 +41,6 @@ class ECommerceCustomer
         $this->name = $name;
     }
 
-    /**
-     * @OneToMany(targetEntity="ECommerceProduct")
-    public $watched;
-     */
-
     public function setCart(ECommerceCart $cart)
     {
         if ($this->cart !== $cart) {

tests/Doctrine/Tests/Models/ECommerce/ECommerceProduct.php

@@ -43,6 +43,11 @@ class ECommerceProduct
     private $categories;
      */
 
+    public function __construct()
+    {
+        $this->features = new \Doctrine\Common\Collections\Collection;
+    }
+
     public function getId()
     {
         return $this->id;
@@ -89,11 +94,13 @@ class ECommerceProduct
     }
 
     public function removeFeature(ECommerceFeature $feature) {
+        if ($this->features->contains($feature)) {
             $removed = $this->features->removeElement($feature);
-        if ($removed !== null) {
-            $removed->removeProduct();
+            if ($removed) {
+                $feature->removeProduct();
                 return true;
             }
+        }
         return false;
     }
 }