Add unit test for a custom filename attribute suffix.

dustin10 · Jun 29, 2015 · 4a9d075 · 4a9d075
1 parent cb290db
commit 4a9d075
Showing 1 changed file with 30 additions and 0 deletions.
diff --git a/Tests/Mapping/PropertyMappingFactoryTest.php b/Tests/Mapping/PropertyMappingFactoryTest.php
@@ -317,6 +317,36 @@ public function testFromFieldReturnsNullOnInvalidFieldName()
         $this->assertNull($mapping);
     }
 
+    public function testCustomFileNameProperty()
+    {
+        $mappings = array(
+            'dummy_file' => array(
+                'namer' => false,
+                'directory_namer' => false
+            )
+        );
+
+        $this->metadata
+            ->expects($this->once())
+            ->method('isUploadable')
+            ->with('Vich\UploaderBundle\Tests\DummyEntity')
+            ->will($this->returnValue(true));
+
+        $this->metadata
+            ->expects($this->once())
+            ->method('getUploadableField')
+            ->with('Vich\UploaderBundle\Tests\DummyEntity')
+            ->will($this->returnValue(array(
+                'mapping'           => 'dummy_file',
+                'propertyName'      => 'file'
+            )));
+
+        $factory = new PropertyMappingFactory($this->container, $this->metadata, $mappings, '_suffix');
+        $mapping = $factory->fromField(new DummyEntity(), 'file');
+
+        $this->assertEquals('file_suffix', $mapping->getFileNamePropertyName());
+    }
+
     public function testConfiguredNamersAreRetrievedFromContainer()
     {
         $mappings = array(