Custom control mapping tests

Author: AydinHassan

src/CliMenu.php

@@ -205,7 +205,7 @@ private function selectFirstItem() : void
     public function addCustomControlMapping(string $input, callable $callable) : void
     {
         if (isset($this->defaultControlMappings[$input]) || isset($this->customControlMappings[$input])) {
-            throw new \InvalidArgumentException('Cannot rebind this input.');
+            throw new \InvalidArgumentException('Cannot rebind this input');
         }
 
         $this->customControlMappings[$input] = $callable;
@@ -217,7 +217,7 @@ public function addCustomControlMapping(string $input, callable $callable) : voi
     public function removeCustomControlMapping(string $input) : void
     {
         if (!isset($this->customControlMappings[$input])) {
-            throw new \InvalidArgumentException('This input is not registered.');
+            throw new \InvalidArgumentException('This input is not registered');
         }
 
         unset($this->customControlMappings[$input]);

test/CliMenuTest.php

@@ -377,6 +377,71 @@ public function testAskPasswordStyle() : void
         static::assertEquals('red', $password->getStyle()->getFg());
     }
 
+    public function testAddCustomControlMappingThrowsExceptionWhenOverwritingExistingDefaultControls() : void
+    {
+        $this->expectException(\InvalidArgumentException::class);
+        $this->expectExceptionMessage('Cannot rebind this input');
+
+        $menu = new CliMenu('PHP School FTW', []);
+        $menu->addCustomControlMapping(' ', function () {
+        });
+    }
+
+    public function testAddCustomControlMappingThrowsExceptionWhenAttemptingToOverwriteAddedCustomControlMap() : void
+    {
+        $this->expectException(\InvalidArgumentException::class);
+        $this->expectExceptionMessage('Cannot rebind this input');
+
+        $menu = new CliMenu('PHP School FTW', []);
+        $menu->addCustomControlMapping('c', function () {
+        });
+        $menu->addCustomControlMapping('c', function () {
+        });
+    }
+
+    public function testAddCustomControlMapping() : void
+    {
+        $this->terminal->expects($this->once())
+            ->method('read')
+            ->willReturn('c');
+
+        $style = $this->getStyle($this->terminal);
+
+        $action = function (CliMenu $menu) {
+            $menu->close();
+        };
+        $item = new SelectableItem('Item 1', $action);
+
+        $menu = new CliMenu('PHP School FTW', [$item], $this->terminal, $style);
+        $menu->addCustomControlMapping('c', $action);
+        $menu->open();
+
+        static::assertStringEqualsFile($this->getTestFile(), $this->output->fetch());
+    }
+
+    public function testRemoveCustomControlMappingThrowsExceptionIfNoSuchMappingExists() : void
+    {
+        $this->expectException(\InvalidArgumentException::class);
+        $this->expectExceptionMessage('This input is not registered');
+
+        $menu = new CliMenu('PHP School FTW', []);
+        $menu->removeCustomControlMapping('c');
+    }
+
+    public function testRemoveCustomControlMapping() : void
+    {
+        $action = function (CliMenu $menu) {
+            $menu->close();
+        };
+
+        $menu = new CliMenu('PHP School FTW', [], $this->terminal);
+        $menu->addCustomControlMapping('c', $action);
+        self::assertSame(['c' => $action], $this->readAttribute($menu, 'customControlMappings'));
+        
+        $menu->removeCustomControlMapping('c');
+        self::assertSame([], $this->readAttribute($menu, 'customControlMappings'));
+    }
+
     private function getTestFile() : string
     {
         return sprintf('%s/res/%s.txt', __DIR__, $this->getName());

test/res/testAddCustomControlMapping.txt

@@ -0,0 +1,9 @@
+
+
+                                                  
+    PHP School FTW                                
+    ==========================================    
+    ● Item 1                                      
+                                                  
+
+