Resolves issue #9

Author: peter279k

src/SystemCtl.php

@@ -14,6 +14,7 @@
 use icanhazstring\SystemCtl\Unit\Slice;
 use icanhazstring\SystemCtl\Unit\Swap;
 use icanhazstring\SystemCtl\Unit\Target;
+use icanhazstring\SystemCtl\Unit\Automount;
 use icanhazstring\SystemCtl\Unit\UnitInterface;
 
 /**
@@ -443,4 +444,36 @@ public function getDevices(?string $unitPrefix = null): array
              return new Device($unitName, $this->getCommandDispatcher());
          }, $units);
     }
+
+     /**
+     * @param string $name
+     *
+     * @return Automount
+     */
+    public function getAutomount(string $name): Automount
+    {
+        $units = $this->listUnits($name, [Automount::UNIT]);
+
+        $unitName = $this->searchForUnitInUnits($name, $units);
+
+        if (is_null($unitName)) {
+            throw UnitNotFoundException::create(Automount::UNIT, $name);
+        }
+
+        return new Automount($unitName, $this->getCommandDispatcher());
+    }
+
+    /**
+     * @param null|string $unitPrefix
+     *
+     * @return Automount[]
+     */
+    public function getAutomounts(?string $unitPrefix = null): array
+    {
+        $units = $this->listUnits($unitPrefix, [Automount::UNIT]);
+
+        return array_map(function ($unitName) {
+            return new Automount($unitName, $this->getCommandDispatcher());
+        }, $units);
+    }
 }

src/Unit/Automount.php

@@ -0,0 +1,24 @@
+<?php
+
+namespace icanhazstring\SystemCtl\Unit;
+
+/**
+ * Class Automount
+ *
+ * @package icanhazstring\SystemCtl\Unit
+ */
+class Automount extends AbstractUnit
+{
+    /**
+     * @var string
+     */
+    public const UNIT = 'automount';
+
+    /**
+     * @inheritdoc
+     */
+    protected function getUnitSuffix(): string
+    {
+        return static::UNIT;
+    }
+}

test/Integration/Unit/UnitTest.php

@@ -17,6 +17,7 @@
 use icanhazstring\SystemCtl\Unit\Slice;
 use icanhazstring\SystemCtl\Unit\Swap;
 use icanhazstring\SystemCtl\Unit\Target;
+use icanhazstring\SystemCtl\Unit\Automount;
 
 /**
  * Class UnitTest
@@ -268,4 +269,33 @@ public function testDeviceCommandsIfProcessIsUnsuccessFulShouldRaiseException()
         $this->expectException(CommandFailedException::class);
         $device->start();
     }
+
+    public function testAutomountCommandsIfProcessIsSuccessfulShouldReturnTrue()
+    {
+        $command = $this->prophesize(CommandInterface::class);
+        $command->isSuccessful()->willReturn(true);
+
+        $commandDispatcher = $this->createCommandDispatcherStub();
+        $commandDispatcher->dispatch(Argument::cetera())->willReturn($command);
+
+        $automount = new Automount('AwesomeAutomount', $commandDispatcher->reveal());
+
+        $this->assertTrue($automount->start());
+        $this->assertTrue($automount->stop());
+        $this->assertTrue($automount->enable());
+        $this->assertTrue($automount->disable());
+        $this->assertTrue($automount->reload());
+        $this->assertTrue($automount->restart());
+    }
+
+    public function testAutomountCommandsIfProcessIsUnsuccessFulShouldRaiseException()
+    {
+        $commandDispatcher = $this->createCommandDispatcherStub();
+        $commandDispatcher->dispatch(Argument::cetera())->willThrow(CommandFailedException::class);
+
+        $automount = new Automount('AwesomeAutomount', $commandDispatcher->reveal());
+
+        $this->expectException(CommandFailedException::class);
+        $automount->start();
+    }
 }

test/Unit/SystemCtlTest.php

@@ -18,6 +18,7 @@
 use icanhazstring\SystemCtl\Unit\Slice;
 use icanhazstring\SystemCtl\Unit\Swap;
 use icanhazstring\SystemCtl\Unit\Target;
+use icanhazstring\SystemCtl\Unit\Automount;
 
 /**
  * Class SystemCtlTest
@@ -238,6 +239,25 @@ public function itShouldCallCommandDispatcherWithListUnitsAndUnitPrefixOnSwapGet
         $this->assertEquals($unitName, $swap->getName());
     }
 
+    /**
+     * @test
+     */
+    public function itShouldCallCommandDispatcherWithListUnitsAndUnitPrefixOnAutomountGetting()
+    {
+        $unitName = 'testAutomount';
+        $output = ' testAutomount.automount     Active running';
+        $commandDispatcherStub = $this->buildCommandDispatcherStub();
+        $commandDispatcherStub
+            ->dispatch(...['list-units', $unitName . '*'])
+            ->willReturn($this->buildCommandStub($output));
+
+        $systemctl = (new SystemCtl())->setCommandDispatcher($commandDispatcherStub->reveal());
+
+        $automount = $systemctl->getAutomount($unitName);
+        $this->assertInstanceOf(Automount::class, $automount);
+        $this->assertEquals($unitName, $automount->getName());
+    }
+
     /**
      * @test
      */
@@ -411,4 +431,21 @@ public function itShouldThrowAnExceptionIfNoDeviceCouldBeFound()
         $this->expectException(UnitNotFoundException::class);
         $systemctl->getSocket($unitName);
     }
+
+    /**
+     * @test
+     */
+    public function itShouldThrowAnExceptionIfNoAutomountCouldBeFound()
+    {
+        $unitName = 'testAutomount';
+        $commandDispatcherStub = $this->buildCommandDispatcherStub();
+        $commandDispatcherStub
+            ->dispatch(...['list-units', $unitName . '*'])
+            ->willReturn($this->buildCommandStub(''));
+
+        $systemctl = (new SystemCtl())->setCommandDispatcher($commandDispatcherStub->reveal());
+
+        $this->expectException(UnitNotFoundException::class);
+        $systemctl->getAutomount($unitName);
+    }
 }

test/Unit/Unit/AbstractUnitTest.php

@@ -70,6 +70,9 @@ public function itShouldReturnCorrectNameDataProvider(): array
             [
                 'name' => 'test1.mount',
             ],
+            [
+                'name' => 'test1.automount',
+            ],
             [
                 'name' => 'test1.device',
             ],
@@ -116,6 +119,10 @@ public function itDetectsMultiInstanceUnitsCorrectlyDataProvider(): array
                 'name' => '[email protected]',
                 'isMultiInstance' => true,
             ],
+            [
+                'name' => '[email protected]',
+                'isMultiInstance' => true,
+            ],
             [
                 'name' => 'test1.service',
                 'isMultiInstance' => false,

test/Unit/Utils/OutputFetcherTest.php

@@ -49,6 +49,8 @@ public function itDeterminesTheCorrectAmountOfUnitsDataProvider(): array
   nonservice.target            active running
   superservice.mount           active running
   awesomeservice.mount         active running
+  superservice.automount       active running
+  awesomeservice.automount     active running
   nonservice.timer             active running
   nonservice.socket            active running
   nonservice.device            active running
@@ -107,6 +109,11 @@ public function itDeterminesTheCorrectAmountOfUnitsDataProvider(): array
                 'suffix' => 'mount',
                 'amount' => 2,
             ],
+            [
+                'output' => $output,
+                'suffix' => 'automount',
+                'amount' => 2,
+            ],
             [
                 'output' => $output,
                 'suffix' => 'notThere',
@@ -146,6 +153,8 @@ public function itOnlyExtractsTheUnitNamesDataProvider(): array
   a-target.target          Active running
   super.mount        Active running
   awesome.mount      Active running
+  super.automount        Active running
+  awesome.automount      Active running
   nonservice.timer          Active running
   nonservice.socket          Active running
   nonservice.device          Active running
@@ -234,6 +243,14 @@ public function itOnlyExtractsTheUnitNamesDataProvider(): array
                     'awesome',
                 ],
             ],
+            [
+                'output' => $output,
+                'suffix' => 'automount',
+                'units'  => [
+                    'super',
+                    'awesome',
+                ],
+            ],
         ];
     }
 }