throw exception if required attributes of UploadableField are not pre…

…sent and tests
slam310 · Jan 29, 2012 · 0fe106a · 0fe106a
1 parent 85f4139
commit 0fe106a
Show file tree

Hide file tree

Showing 2 changed files with 37 additions and 0 deletions.
diff --git a/Mapping/Annotation/UploadableField.php b/Mapping/Annotation/UploadableField.php
@@ -35,10 +35,14 @@ public function __construct(array $options)
     {
         if (isset($options['mapping'])) {
             $this->mapping = $options['mapping'];
+        } else {
+            throw new \Exception('The "mapping" attribute of UploadableField is required.');
         }
 
         if (isset($options['fileNameProperty'])) {
             $this->fileNameProperty = $options['fileNameProperty'];
+        } else {
+            throw new \Exception('The "fileNameProperty" attribute of UploadableField is required.');
         }
     }
 

diff --git a/Tests/Mapping/Annotation/UploadableFieldTest.php b/Tests/Mapping/Annotation/UploadableFieldTest.php
@@ -0,0 +1,33 @@
+<?php
+
+namespace Vich\UploaderBundle\Tests\Mapping\Annotation;
+
+use Vich\UploaderBundle\Mapping\Annotation\UploadableField;
+
+/**
+ * UploadableFieldTest.
+ *
+ * @author Dustin Dobervich <ddobervich@gmail.com>
+ */
+class UploadableFieldTest extends \PHPUnit_Framework_TestCase
+{
+    /**
+     * @expectedException \Exception
+     */
+    public function testExceptionThrownWhenNoMappingAttribute()
+    {
+        new UploadableField(array(
+            'fileNameProperty' => 'fileName'
+        ));
+    }
+
+    /**
+     * @expectedException \Exception
+     */
+    public function testExceptionThrownWhenNoFileNamePropertyAttribute()
+    {
+        new UploadableField(array(
+            'mapping' => 'file'
+        ));
+    }
+}