Add method to store plugin in config after reloading

diff --git a/src/main/java/net/pterodactylus/fcp/quelaton/ReloadPluginCommand.java b/src/main/java/net/pterodactylus/fcp/quelaton/ReloadPluginCommand.java
index 8495ad2..5024267 100644 (file)
--- a/src/main/java/net/pterodactylus/fcp/quelaton/ReloadPluginCommand.java
+++ b/src/main/java/net/pterodactylus/fcp/quelaton/ReloadPluginCommand.java
@@ -13,6 +13,7 @@ public interface ReloadPluginCommand {
 
        ReloadPluginCommand waitFor(int milliseconds);
        ReloadPluginCommand purge();
+       ReloadPluginCommand addToConfig();
        Executable<Optional<PluginInfo>> plugin(String pluginClassName);
 
 }

diff --git a/src/main/java/net/pterodactylus/fcp/quelaton/ReloadPluginCommandImpl.java b/src/main/java/net/pterodactylus/fcp/quelaton/ReloadPluginCommandImpl.java
index 2c2e6c4..e544ec6 100644 (file)
--- a/src/main/java/net/pterodactylus/fcp/quelaton/ReloadPluginCommandImpl.java
+++ b/src/main/java/net/pterodactylus/fcp/quelaton/ReloadPluginCommandImpl.java
@@ -45,6 +45,12 @@ public class ReloadPluginCommandImpl implements ReloadPluginCommand {
        }
 
        @Override
+       public ReloadPluginCommand addToConfig() {
+               reloadPlugin.setStore(true);
+               return this;
+       }
+
+       @Override
        public Executable<Optional<PluginInfo>> plugin(String pluginClassName) {
                reloadPlugin.setPluginName(pluginClassName);
                return this::execute;

diff --git a/src/test/java/net/pterodactylus/fcp/quelaton/DefaultFcpClientTest.java b/src/test/java/net/pterodactylus/fcp/quelaton/DefaultFcpClientTest.java
index 0974d40..6cc61d6 100644 (file)
--- a/src/test/java/net/pterodactylus/fcp/quelaton/DefaultFcpClientTest.java
+++ b/src/test/java/net/pterodactylus/fcp/quelaton/DefaultFcpClientTest.java
@@ -2171,6 +2171,16 @@ public class DefaultFcpClientTest {
                                verifyPluginInfo(pluginInfo);
                        }
 
+                       @Test
+                       public void reloadingPluginWithStoreWorks()
+                       throws InterruptedException, ExecutionException, IOException {
+                               Future<Optional<PluginInfo>> pluginInfo =
+                                       fcpClient.reloadPlugin().addToConfig().plugin(CLASS_NAME).execute();
+                               connectAndAssert(() -> allOf(matchReloadPluginMessage(), hasItem("Store=true")));
+                               replyWithPluginInfo();
+                               verifyPluginInfo(pluginInfo);
+                       }
+
                        private Matcher<List<String>> matchReloadPluginMessage() {
                                return matchesFcpMessage(
                                        "ReloadPlugin",