1 files changed, 17 insertions, 15 deletions

diff --git a/tests/units/ActionTaskCloseTest.php b/tests/units/ActionTaskCloseTest.php
index 6c8e4cf1..7f2c42de 100644
--- a/tests/units/ActionTaskCloseTest.php
+++ b/tests/units/ActionTaskCloseTest.php
@@ -2,16 +2,18 @@
 
 require_once __DIR__.'/Base.php';
 
+use Event\GenericEvent;
 use Model\Task;
+use Model\TaskCreation;
 use Model\TaskFinder;
 use Model\Project;
-use Model\GithubWebhook;
+use Integration\GithubWebhook;
 
 class ActionTaskCloseTest extends Base
 {
     public function testExecutable()
     {
-        $action = new Action\TaskClose($this->registry, 3, Task::EVENT_MOVE_COLUMN);
+        $action = new Action\TaskClose($this->container, 3, Task::EVENT_MOVE_COLUMN);
         $action->setParam('column_id', 5);
 
         $event = array(
@@ -22,7 +24,7 @@ class ActionTaskCloseTest extends Base
 
         $this->assertTrue($action->isExecutable($event));
 
-        $action = new Action\TaskClose($this->registry, 3, GithubWebhook::EVENT_COMMIT);
+        $action = new Action\TaskClose($this->container, 3, GithubWebhook::EVENT_COMMIT);
 
         $event = array(
             'project_id' => 3,
@@ -34,7 +36,7 @@ class ActionTaskCloseTest extends Base
 
     public function testBadEvent()
     {
-        $action = new Action\TaskClose($this->registry, 3, Task::EVENT_UPDATE);
+        $action = new Action\TaskClose($this->container, 3, Task::EVENT_UPDATE);
         $action->setParam('column_id', 5);
 
         $event = array(
@@ -44,12 +46,12 @@ class ActionTaskCloseTest extends Base
         );
 
         $this->assertFalse($action->isExecutable($event));
-        $this->assertFalse($action->execute($event));
+        $this->assertFalse($action->execute(new GenericEvent($event)));
     }
 
     public function testBadProject()
     {
-        $action = new Action\TaskClose($this->registry, 3, Task::EVENT_MOVE_COLUMN);
+        $action = new Action\TaskClose($this->container, 3, Task::EVENT_MOVE_COLUMN);
         $action->setParam('column_id', 5);
 
         $event = array(
@@ -59,12 +61,12 @@ class ActionTaskCloseTest extends Base
         );
 
         $this->assertFalse($action->isExecutable($event));
-        $this->assertFalse($action->execute($event));
+        $this->assertFalse($action->execute(new GenericEvent($event)));
     }
 
     public function testBadColumn()
     {
-        $action = new Action\TaskClose($this->registry, 3, Task::EVENT_MOVE_COLUMN);
+        $action = new Action\TaskClose($this->container, 3, Task::EVENT_MOVE_COLUMN);
         $action->setParam('column_id', 5);
 
         $event = array(
@@ -73,20 +75,20 @@ class ActionTaskCloseTest extends Base
             'column_id' => 3,
         );
 
-        $this->assertFalse($action->execute($event));
+        $this->assertFalse($action->execute(new GenericEvent($event)));
     }
 
     public function testExecute()
     {
-        $action = new Action\TaskClose($this->registry, 1, Task::EVENT_MOVE_COLUMN);
+        $action = new Action\TaskClose($this->container, 1, Task::EVENT_MOVE_COLUMN);
         $action->setParam('column_id', 2);
 
         // We create a task in the first column
-        $t = new Task($this->registry);
-        $tf = new TaskFinder($this->registry);
-        $p = new Project($this->registry);
+        $tc = new TaskCreation($this->container);
+        $tf = new TaskFinder($this->container);
+        $p = new Project($this->container);
         $this->assertEquals(1, $p->create(array('name' => 'test')));
-        $this->assertEquals(1, $t->create(array('title' => 'test', 'project_id' => 1, 'column_id' => 1)));
+        $this->assertEquals(1, $tc->create(array('title' => 'test', 'project_id' => 1, 'column_id' => 1)));
 
         // We create an event to move the task to the 2nd column
         $event = array(
@@ -96,7 +98,7 @@ class ActionTaskCloseTest extends Base
         );
 
         // Our event should be executed
-        $this->assertTrue($action->execute($event));
+        $this->assertTrue($action->execute(new GenericEvent($event)));
 
         // Our task should be closed
         $task = $tf->getById(1);