Add Console::setLanguage method

aplus-framework · May 10, 2024 · 44e7112 · 44e7112
1 parent feb4f4a
commit 44e7112
Show file tree

Hide file tree

Showing 2 changed files with 31 additions and 4 deletions.
diff --git a/src/Console.php b/src/Console.php
@@ -58,10 +58,9 @@ class Console
      */
     public function __construct(Language $language = null)
     {
-        if ($language === null) {
-            $language = new Language('en');
+        if ($language) {
+            $this->setLanguage($language);
         }
-        $this->language = $language->addDirectory(__DIR__ . '/Languages');
         global $argv;
         $this->prepare($argv ?? []);
         $this->setDefaultCommands();
@@ -130,14 +129,30 @@ public function getArgument(int $position) : ?string
         return $this->arguments[$position] ?? null;
     }
 
+    /**
+     * Set the Language instance.
+     *
+     * @param Language|null $language
+     *
+     * @return static
+     */
+    public function setLanguage(Language $language = null) : static
+    {
+        $this->language = $language ?? new Language();
+        $this->language->addDirectory(__DIR__ . '/Languages');
+        return $this;
+    }
+
     /**
      * Get the Language instance.
      *
      * @return Language
      */
-    #[Pure]
     public function getLanguage() : Language
     {
+        if (!isset($this->language)) {
+            $this->setLanguage();
+        }
         return $this->language;
     }
 

diff --git a/tests/ConsoleTest.php b/tests/ConsoleTest.php
@@ -12,6 +12,7 @@
 use Framework\CLI\Command;
 use Framework\CLI\Streams\Stderr;
 use Framework\CLI\Streams\Stdout;
+use Framework\Language\Language;
 use PHPUnit\Framework\TestCase;
 
 final class ConsoleTest extends TestCase
@@ -29,6 +30,17 @@ protected function tearDown() : void
         Stdout::reset();
     }
 
+    public function testLanguage() : void
+    {
+        $language = new Language();
+        $console = new ConsoleMock($language);
+        self::assertSame($language, $console->getLanguage());
+        self::assertContains(
+            \realpath(__DIR__ . '/../src/Languages') . \DIRECTORY_SEPARATOR,
+            $console->getLanguage()->getDirectories()
+        );
+    }
+
     public function testEmptyLine() : void
     {
         $this->console->prepare([