Mettre le buyer à null lors d'une transaction CB

Si l'utilisateur essaie de payer par compte payutc, que ça échoue, puis paie par CB, on veut effacer le buyerId pour pouvoir distinguer le type de transaction.

src/Payutc/Bom/Transaction.php

@@ -124,7 +124,12 @@ public function setEmail($email){
     }
 
     public function setBuyer($buyer){
-        $this->buyerId = $buyer->getId();
+        if($buyer == null){
+            $this->buyerId = null;
+        }
+        else {
+            $this->buyerId = $buyer->getId();
+        }
 
         Dbal::conn()->update('t_transaction_tra',
             array('usr_id_buyer' => $this->buyerId),

src/Payutc/Service/WEBSALECONFIRM.php

@@ -111,6 +111,7 @@ public function doTransaction($tra_id, $token, $montant_reload) {
             }
         } else {
             $pl = new Payline($this->application()->getId(), $this->service_name);
+            $transaction->setBuyer(null);
             return $pl->doWebPayment(
                 null, 
                 $transaction, 

tests/TransactionRwdbTest.php

@@ -135,5 +135,23 @@ public function testCreateEmail(){
         $transaction2 = Transaction::getById($transaction->getId());
         $this->assertEquals("[email protected]", $transaction2->getEmail());
     }
+
+	/**
+     * @requires PHP 5.4
+	 */    
+    public function testSetToNull(){
+        $items = array(
+            array(4, 1),
+            array(4, 3)
+        );
+        $matthieu = new User("mguffroy");
+        $transaction = Transaction::create($matthieu, $matthieu, 51, 1, $items, null, null);
+
+        $this->assertEquals(600, $transaction->getMontantTotal());
+
+        $transaction->setBuyer(null);
+
+        $this->assertEquals(null, $transaction->getBuyerId());
+    }
 }