add a method to set host

ArkEcosystem · Jul 31, 2024 · a8237d6 · a8237d6
1 parent 9e8a09e
commit a8237d6
Show file tree

Hide file tree

Showing 2 changed files with 38 additions and 0 deletions.
diff --git a/src/ArkClient.php b/src/ArkClient.php
@@ -101,6 +101,23 @@ public function __call($name, $args)
         }
     }
 
+    /**
+     * Set the host for the given type.
+     *
+     * @param string $host
+     * @param string $type
+     *
+     * @throws InvalidArgumentException if the type is not 'api', 'transactions', or 'evm'
+     */
+    public function setHost(string $host, string $type): void
+    {
+        if (! in_array($type, ['api', 'transactions', 'evm'], true)) {
+            throw new \InvalidArgumentException('Invalid host type.');
+        }
+
+        $this->hosts[$type] = $host;
+    }
+
     /**
      * @return array{
      *  api: string,

diff --git a/tests/ArkClientTest.php b/tests/ArkClientTest.php
@@ -101,6 +101,27 @@ public function should_accept_custom_handler()
         $this->assertSame($handler, $connection->getHttpClient()->getConfig('handler'));
     }
 
+    /** @test */
+    public function should_set_host()
+    {
+        $client = $this->getClient();
+
+        $newHost = 'https://new-host.com/api';
+        $client->setHost($newHost, 'api');
+
+        $this->assertSame($newHost, $client->getHosts()['api']);
+    }
+
+    /** @test */
+    public function should_throw_exception_if_host_type_is_invalid()
+    {
+        $client = $this->getClient();
+
+        $this->expectException(\InvalidArgumentException::class);
+
+        $client->setHost('https://new-host.com/api', 'other');
+    }
+
     /**
      * Get a new client instance.
      *