fix: Fix addition of custom commands

Author: valfirst

src/main/java/io/appium/java_client/AppiumDriver.java

@@ -31,6 +31,7 @@
 import org.openqa.selenium.UnsupportedCommandException;
 import org.openqa.selenium.WebDriverException;
 import org.openqa.selenium.remote.CapabilityType;
+import org.openqa.selenium.remote.CommandInfo;
 import org.openqa.selenium.remote.DriverCommand;
 import org.openqa.selenium.remote.ErrorHandler;
 import org.openqa.selenium.remote.ExecuteMethod;
@@ -242,22 +243,23 @@ public Map<String, Object> getStatus() {
      * @param methodName The name of custom appium command.
      */
     public void addCommand(HttpMethod httpMethod, String url, String methodName) {
+        CommandInfo commandInfo;
         switch (httpMethod) {
             case GET:
-                MobileCommand.commandRepository.put(methodName, MobileCommand.getC(url));
+                commandInfo = MobileCommand.getC(url);
                 break;
             case POST:
-                MobileCommand.commandRepository.put(methodName, MobileCommand.postC(url));
+                commandInfo = MobileCommand.postC(url);
                 break;
             case DELETE:
-                MobileCommand.commandRepository.put(methodName, MobileCommand.deleteC(url));
+                commandInfo = MobileCommand.deleteC(url);
                 break;
             default:
                 throw new WebDriverException(String.format("Unsupported HTTP Method: %s. Only %s methods are supported",
                         httpMethod,
                         Arrays.toString(HttpMethod.values())));
         }
-        ((AppiumCommandExecutor) getCommandExecutor()).refreshAdditionalCommands();
+        ((AppiumCommandExecutor) getCommandExecutor()).defineCommand(methodName, commandInfo);
     }
 
     @Override

src/main/java/io/appium/java_client/remote/AppiumCommandExecutor.java

@@ -181,7 +181,7 @@ private Response createSession(Command command) throws IOException {
                     + "Please make sure your server is up to date.");
         }
         setCommandCodec(new AppiumW3CHttpCommandCodec());
-        refreshAdditionalCommands();
+        getAdditionalCommands().forEach(this::defineCommand);
         setResponseCodec(dialect.getResponseCodec());
         Response response = result.createResponse();
         if (this.appiumClientConfig != null && this.appiumClientConfig.isDirectConnectEnabled()) {
@@ -191,8 +191,8 @@ private Response createSession(Command command) throws IOException {
         return response;
     }
 
-    public void refreshAdditionalCommands() {
-        getAdditionalCommands().forEach(this::defineCommand);
+    public void defineCommand(String commandName, CommandInfo info) {
+        super.defineCommand(commandName, info);
     }
 
     @SuppressWarnings("unchecked")