Playlists tests: add and remove element

src/Muzich/CoreBundle/Tests/Controller/PlaylistControllerTest.php

@@ -314,4 +314,58 @@ class PlaylistControllerTest extends FunctionalTest
     $this->jsonResponseIsSuccess($response);
   }
   
+  public function testAddAndRemoveElement()
+  {
+    $this->init();
+    $this->initAddRemoveContextData();
+    $this->connectUser('bux', 'toor');
+    
+    $this->tests_cases->playlistShow($this->users['bux']->getSlug(), $this->playlists['bux_2_priv']->getId());
+    $this->checkReadPlaylist($this->playlists['bux_2_priv'],
+      array($this->playlists['bux_2_priv']->getId() => array(
+        $this->elements['heretik'],
+        $this->elements['fab']
+      ))
+    );
+    
+    $this->addElementToPlaylist($this->playlists['bux_2_priv'], $this->elements['azyd']);
+    
+    $this->tests_cases->playlistShow($this->users['bux']->getSlug(), $this->playlists['bux_2_priv']->getId());
+    $this->checkReadPlaylist($this->playlists['bux_2_priv'],
+      array($this->playlists['bux_2_priv']->getId() => array(
+        $this->elements['heretik'],
+        $this->elements['fab'],
+        $this->elements['azyd']
+      ))
+    );
+    
+    $this->removeElementFromPlaylist($this->playlists['bux_2_priv'], $this->elements['fab']);
+    
+    $this->tests_cases->playlistShow($this->users['bux']->getSlug(), $this->playlists['bux_2_priv']->getId());
+    $this->checkReadPlaylist($this->playlists['bux_2_priv'],
+      array($this->playlists['bux_2_priv']->getId() => array(
+        $this->elements['heretik'],
+        $this->elements['azyd']
+      ))
+    );
+  }
+  
+  protected function initAddRemoveContextData()
+  {
+    $this->initOrderContextData();
+    $this->elements['heretik'] = $this->findOneBy('Element', 'Heretik System Popof - Resistance');
+    $this->elements['fab'] = $this->findOneBy('Element', 'DJ FAB');
+    $this->elements['azyd'] = $this->findOneBy('Element', 'AZYD AZYLUM Live au Café Provisoire'); 
+  }
+  
+  protected function addElementToPlaylist($playlist, $element)
+  {
+    $this->tests_cases->playlistAddElement($playlist->getId(), $element->getId());
+  }
+  
+  protected function removeElementFromPlaylist($playlist, $element)
+  {
+    $this->tests_cases->playlistremoveElement($playlist->getId(), $element->getId());
+  }
+  
 }

src/Muzich/CoreBundle/Tests/lib/Security/ContextTestCases.php

@@ -251,20 +251,26 @@ class ContextTestCases
   
   public function playlistAddElementResponseIs($success, $condition)
   {
+    $this->playlistAddElement(0, 0);
     return $this->ajaxResponseSatisfyConditions(
-      $this->getAjaxRequestContentResponse(
-        'GET',
-        $this->test->generateUrl('playlists_add_element', array(
-          'playlist_id' => 0,
-          'element_id'  => 0,
-          '_locale'     => 'fr'
-        ))
-      ), 
+      $this->test->getClient()->getResponse()->getContent(), 
       $success, 
       $condition
     );
   }
   
+  public function playlistAddElement($playlist_id, $element_id)
+  {
+    return $this->getAjaxRequestContentResponse(
+      'GET',
+      $this->test->generateUrl('playlists_add_element', array(
+        'playlist_id' => $playlist_id,
+        'element_id'  => $element_id,
+        '_locale'     => 'fr'
+      ))
+    );
+  }
+  
   public function playlistUpdateOrderResponseIs($success, $condition)
   {
     $this->playlistUpdateOrder(0, array());
@@ -297,20 +303,26 @@ class ContextTestCases
   
   public function playlistRemoveElementResponseIs($success, $condition)
   {
+    $this->playlistRemoveElement(0, 0);
     return $this->ajaxResponseSatisfyConditions(
-      $this->getAjaxRequestContentResponse(
-        'GET',
-        $this->test->generateUrl('playlist_remove_element', array(
-          'playlist_id' => 0,
-          'element_id'  => 0,
-          '_locale'     => 'fr'
-        ))
-      ), 
+      $this->test->getClient()->getResponse()->getContent(),
       $success, 
       $condition
     );
   }
   
+  public function playlistRemoveElement($playlist_id, $element_id)
+  {
+    return $this->getAjaxRequestContentResponse(
+      'GET',
+      $this->test->generateUrl('playlist_remove_element', array(
+        'playlist_id' => $playlist_id,
+        'element_id'  => $element_id,
+        '_locale'     => 'fr'
+      ))
+    );
+  }
+  
   public function playlistCreateResponseIs($success, $condition)
   {
     $this->playlistCreate(0, 'my_super_playlist');