Define only what is allowed for column restrictions

1 files changed, 4 insertions, 0 deletions

diff --git a/tests/units/Helper/BoardHelperTest.php b/tests/units/Helper/BoardHelperTest.php
index 641ca9cb..0a44f8ab 100644
--- a/tests/units/Helper/BoardHelperTest.php
+++ b/tests/units/Helper/BoardHelperTest.php
@@ -87,8 +87,12 @@ class BoardHelperTest extends Base
 
         $this->assertTrue($projectUserRole->addUser(1, 2, 'Custom Role'));
         $this->assertEquals(1, $taskCreationModel->create(array('project_id' => 1, 'title' => 'test', 'column_id' => 2)));
+        $this->assertEquals(2, $taskCreationModel->create(array('project_id' => 1, 'title' => 'test', 'column_id' => 3)));
 
         $task = $taskFinderModel->getById(1);
+        $this->assertTrue($boardHelper->isDraggable($task));
+
+        $task = $taskFinderModel->getById(2);
         $this->assertFalse($boardHelper->isDraggable($task));
     }
 }