Merge pull request #122 from davidrushton/feature/diskspace-custom-fi…

…lesystem Added filesystemName to UsedDiskSpaceCheck
spatie · Sep 22, 2022 · 46d1953 · 46d1953
2 parents 2c0ff43 + d6403f5
commit 46d1953
Show file tree

Hide file tree

Showing 3 changed files with 26 additions and 1 deletion.
diff --git a/src/Checks/Checks/UsedDiskSpaceCheck.php b/src/Checks/Checks/UsedDiskSpaceCheck.php
@@ -13,6 +13,15 @@ class UsedDiskSpaceCheck extends Check
 
     protected int $errorThreshold = 90;
 
+    protected ?string $filesystemName = null;
+
+    public function filesystemName(string $filesystemName): self
+    {
+        $this->filesystemName = $filesystemName;
+
+        return $this;
+    }
+
     public function warnWhenUsedSpaceIsAbovePercentage(int $percentage): self
     {
         $this->warningThreshold = $percentage;
@@ -48,7 +57,7 @@ public function run(): Result
 
     protected function getDiskUsagePercentage(): int
     {
-        $process = Process::fromShellCommandline('df -P .');
+        $process = Process::fromShellCommandline('df -P ' . ( $this->filesystemName ?: '.' ));
 
         $process->run();
 

diff --git a/tests/Checks/UsedDiskSpaceCheckTest.php b/tests/Checks/UsedDiskSpaceCheckTest.php
@@ -46,6 +46,16 @@
         ->getNotificationMessage()->toEqual('The disk is almost full (91% used).');
 });
 
+it('can set a custom filesystem name', function () {
+    $filesystem = '/mnt/temp';
+
+    $check = FakeUsedDiskSpaceCheck::new()
+        ->filesystemName('/mnt/temp');
+
+    expect($check->getFilesystemName())
+        ->toEqual($filesystem);
+});
+
 it('can report the real disk space used', function () {
     $result = UsedDiskSpaceCheck::new()->run();
 

diff --git a/tests/TestClasses/FakeUsedDiskSpaceCheck.php b/tests/TestClasses/FakeUsedDiskSpaceCheck.php
@@ -19,4 +19,10 @@ public function getDiskUsagePercentage(): int
     {
         return $this->fakeDiskUsagePercentage;
     }
+
+    public function getFilesystemName(): string|null
+    {
+        return $this->filesystemName;
+    }
+
 }