[#SDV-202] Extend unit tests for LodgingChecklist

ministryofjustice · Feb 4, 2015 · be3345b · be3345b
1 parent 42482ff
commit be3345b
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 0 deletions.
diff --git a/tests/OpgTest/Core/Model/Entity/Document/LodgingChecklistTest.php b/tests/OpgTest/Core/Model/Entity/Document/LodgingChecklistTest.php
@@ -32,6 +32,16 @@ public function testGetSetDirection()
         $this->assertEquals(Document::DIRECTION_INTERNAL, $this->checklist->setDirection(Document::DIRECTION_INTERNAL)->getDirection());
     }
 
+    public function testGetSetTotalAssets()
+    {
+        $expected = 1024.00;
+
+        $this->assertEmpty($this->checklist->getTotalAssets());
+        $this->assertTrue($this->checklist->setTotalAssets($expected) instanceof LodgingChecklist);
+
+        $this->assertEquals($expected, $this->checklist->getTotalAssets());
+    }
+
     public function testGetSetStartEndDates()
     {
         $expectedStart = new \DateTime('1970-10-01');

diff --git a/tests/OpgTest/Core/Model/Entity/LineItem/LineItemTest.php b/tests/OpgTest/Core/Model/Entity/LineItem/LineItemTest.php
@@ -2,6 +2,7 @@
 
 namespace OpgTest\Core\Model\Entity\LineItem;
 
+use Opg\Core\Model\Entity\Document\LodgingChecklist;
 use Opg\Core\Model\Entity\LineItem\LineItem;
 
 class LineItemTest extends \PHPUnit_Framework_TestCase
@@ -22,6 +23,14 @@ public function testGetSetId()
         $this->assertEquals($expected, $this->lineItem->getId());
     }
 
+    public function testGetSetDocument()
+    {
+        $expected = new LodgingChecklist();
+        $this->assertEmpty($this->lineItem->getDocument());
+        $this->assertTrue($this->lineItem->setDocument($expected) instanceof LineItem);
+        $this->assertEquals($expected, $this->lineItem->getDocument());
+    }
+
     public function testGetSetLineItemName()
     {
         $expected = 'line item name';