Add ProjecFile and TaskFile models

1 files changed, 6 insertions, 0 deletions

diff --git a/tests/units/Base.php b/tests/units/Base.php
index 22c6304f..eb9fc68b 100644
--- a/tests/units/Base.php
+++ b/tests/units/Base.php
@@ -67,6 +67,12 @@ abstract class Base extends PHPUnit_Framework_TestCase
             ->setMethods(array('getType', 'getSelectedTypes'))
             ->getMock();
 
+        $this->container['objectStorage'] = $this
+            ->getMockBuilder('\Kanboard\Core\ObjectStorage\FileStorage')
+            ->setConstructorArgs(array($this->container))
+            ->setMethods(array('put', 'moveFile', 'remove', 'moveUploadedFile'))
+            ->getMock();
+
         $this->container['sessionStorage'] = new SessionStorage;
         $this->container->register(new ActionProvider);