Fixed #3 - Added assumeFreeDiskSpace()

MehrAlsNix · May 23, 2015 · 41a43c7 · 41a43c7
1 parent 6bc49d5
commit 41a43c7
Show file tree

Hide file tree

Showing 3 changed files with 36 additions and 0 deletions.
diff --git a/src/Assumptions.php b/src/Assumptions.php
@@ -107,3 +107,15 @@ function assumeEnvironment(...$args)
         call_user_func_array(['MehrAlsNix\Assumptions\Assume', 'assumeEnvironment'], $args);
     }
 }
+
+if (!function_exists('assumeFreeDiskSpace')) {
+    /**
+     * Make an assumption and throw
+     * {@link MehrAlsNix\Assumptions\AssumptionViolatedException} if it fails.
+     * @param ...$args
+     */
+    function assumeFreeDiskSpace(...$args)
+    {
+        call_user_func_array(['MehrAlsNix\Assumptions\Assume', 'assumeFreeDiskSpace'], $args);
+    }
+}
diff --git a/src/Extensions/System.php b/src/Extensions/System.php
@@ -49,4 +49,17 @@ public static function assumeEnvironment($varname, $message = '')
     {
         assumeTrue((bool) getenv($varname), $message);
     }
+
+    /**
+     * @param string $directory
+     * @param string $message
+     */
+    public static function assumeFreeDiskSpace($directory, $available = null,  $message = '')
+    {
+        if ($available === null) {
+            assumeThat(disk_free_space($directory), is(not(false)), $message);
+        } else {
+            assumeThat(disk_free_space($directory), is(greaterThanOrEqualTo($available)), $message);
+        }
+    }
 }
diff --git a/tests/AssumeTest.php b/tests/AssumeTest.php
@@ -112,4 +112,15 @@ public function assumeEnvironment()
         assumeEnvironment('PATH');
         assumeEnvironment('THIS_IS_NOT_AN_ENV', 'Environment not set.');
     }
+
+    /**
+     * @test
+     * @expectedException \MehrAlsNix\Assumptions\AssumptionViolatedException
+     * @expectedExceptionMessage No free disc space.
+     */
+    public function assumeFreeDiskSpace()
+    {
+        assumeFreeDiskSpace(__DIR__);
+        assumeFreeDiskSpace(__DIR__, 9999999999999999.0, 'No free disc space.');
+    }
 }