Merge pull request #179 from ministryofjustice/SDV-61_Crec

Sdv 61 crec

src/Opg/Core/Model/Entity/Document/Crec.php

@@ -0,0 +1,58 @@
+<?php
+
+namespace Opg\Core\Model\Entity\Document;
+
+use Doctrine\ORM\Mapping as ORM;
+use JMS\Serializer\Annotation\Exclude;
+use JMS\Serializer\Annotation\Type;
+use JMS\Serializer\Annotation\Accessor;
+use JMS\Serializer\Annotation\ReadOnly;
+use Opg\Core\Model\Entity\Document\Decorators\SystemType;
+use Opg\Core\Model\Entity\Document\Decorators\HasSystemType;
+
+/**
+ * @ORM\Entity
+ * ORM\entity(repositoryClass="Application\Model\Repository\DocumentRepository")
+ *
+ * Class Correspondence
+ * @package Opg\Core\Model\Entity\Document
+ */
+class Crec extends Document implements HasSystemType
+{
+    use SystemType;
+
+    /**
+     * @Type("string")
+     * @Accessor(getter="getDirection")
+     * @ReadOnly
+     */
+    protected $direction = self::DOCUMENT_INTERNAL_CORRESPONDENCE;
+
+    /**
+     * @ORM\Column(type="integer", nullable=true)
+     * @Type("integer")
+     * @var int
+     */
+    protected $riskScore;
+
+    /**
+     * @return int
+     */
+    public function getRiskScore()
+    {
+        return $this->riskScore;
+    }
+
+    /**
+     * @param int $riskScore
+     *
+     * @return $this
+     */
+    public function setRiskScore($riskScore)
+    {
+        $this->riskScore = intval($riskScore);
+
+        return $this;
+    }
+
+}

src/Opg/Core/Model/Entity/Document/Document.php

@@ -37,6 +37,7 @@
  *     "lodging_checklist" = "Opg\Core\Model\Entity\Document\LodgingChecklist",
  *     "action_log"        = "Opg\Core\Model\Entity\Document\ActionLog",
  *     "annual_report"     = "Opg\Core\Model\Entity\Document\AnnualReport",
+ *     "crec"              = "Opg\Core\Model\Entity\Document\Crec"
  * })
  *
  * Class Document

tests/OpgTest/Core/Model/Entity/Document/CrecTest.php

@@ -0,0 +1,31 @@
+<?php
+
+namespace OpgTest\Core\Model\Entity\Document;
+
+use Opg\Core\Model\Entity\Document\Crec;
+
+class CrecTest extends \PHPUnit_Framework_TestCase
+{
+    /** @var  Crec */
+    protected $crec;
+
+    public function setUp()
+    {
+        $this->crec = new Crec();
+    }
+
+    public function testSetUp()
+    {
+        $this->assertTrue($this->crec instanceof Crec);
+    }
+
+    public function testGetSetTotalAssets()
+    {
+        $expected = 42;
+
+        $this->assertEmpty($this->crec->getRiskScore());
+        $this->assertTrue($this->crec->setRiskScore($expected) instanceof Crec);
+
+        $this->assertEquals($expected, $this->crec->getRiskScore());
+    }
+}