Include swimlane in task export

1 files changed, 10 insertions, 2 deletions

diff --git a/tests/units/TaskExportTest.php b/tests/units/TaskExportTest.php
index fc080c0a..963c031d 100644
--- a/tests/units/TaskExportTest.php
+++ b/tests/units/TaskExportTest.php
@@ -8,6 +8,7 @@ use Model\TaskExport;
 use Model\Project;
 use Model\Category;
 use Model\User;
+use Model\Swimlane;
 
 class TaskExportTest extends Base
 {
@@ -17,8 +18,13 @@ class TaskExportTest extends Base
         $p = new Project($this->container);
         $c = new Category($this->container);
         $e = new TaskExport($this->container);
+        $s = new Swimlane($this->container);
 
         $this->assertEquals(1, $p->create(array('name' => 'Export Project')));
+
+        $this->assertEquals(1, $s->create(1, 'S1'));
+        $this->assertEquals(2, $s->create(1, 'S2'));
+
         $this->assertNotFalse($c->create(array('name' => 'Category #1', 'project_id' => 1)));
         $this->assertNotFalse($c->create(array('name' => 'Category #2', 'project_id' => 1)));
         $this->assertNotFalse($c->create(array('name' => 'Category #3', 'project_id' => 1)));
@@ -34,7 +40,8 @@ class TaskExportTest extends Base
                 'color_id' => rand(0, 1) === 0 ? 'green' : 'purple',
                 'category_id' => rand(0, 3),
                 'date_due' => array_rand(array(0, date('Y-m-d'), date('Y-m-d', strtotime('+'.$i.'day')))),
-                'score' => rand(0, 21)
+                'score' => rand(0, 21),
+                'swimlane_id' => rand(0, 2),
             );
 
             $this->assertEquals($i, $tc->create($task));
@@ -44,6 +51,7 @@ class TaskExportTest extends Base
         $this->assertEquals($i, count($rows));
         $this->assertEquals('Task Id', $rows[0][0]);
         $this->assertEquals(1, $rows[1][0]);
-        $this->assertEquals('Task #'.($i - 1), $rows[$i - 1][11]);
+        $this->assertEquals('Task #'.($i - 1), $rows[$i - 1][12]);
+        $this->assertTrue(in_array($rows[$i - 1][4], array('Default swimlane', 'S1', 'S2')));
     }
 }