Add tab complete for full balance when depositing/withdrawing

#21
BentoBoxWorld · Sep 20, 2021 · b757037 · b757037
1 parent f8ffb4b
commit b757037
Show file tree

Hide file tree

Showing 4 changed files with 45 additions and 0 deletions.
diff --git a/src/main/java/world/bentobox/bank/commands/user/DepositCommand.java b/src/main/java/world/bentobox/bank/commands/user/DepositCommand.java
@@ -1,6 +1,8 @@
 package world.bentobox.bank.commands.user;
 
+import java.util.Collections;
 import java.util.List;
+import java.util.Optional;
 
 import net.milkbowl.vault.economy.EconomyResponse;
 import net.milkbowl.vault.economy.EconomyResponse.ResponseType;
@@ -71,4 +73,10 @@ public boolean execute(User user, String label, List<String> args) {
         return false;
     }
 
+    @Override
+    public Optional<List<String>> tabComplete(User user, String alias, List<String> args) {
+        VaultHook vault = addon.getVault();
+        String balance = String.valueOf(vault.getBalance(user, getWorld()));
+        return Optional.of(Collections.singletonList(balance));
+    }
 }
diff --git a/src/main/java/world/bentobox/bank/commands/user/WithdrawCommand.java b/src/main/java/world/bentobox/bank/commands/user/WithdrawCommand.java
@@ -1,6 +1,8 @@
 package world.bentobox.bank.commands.user;
 
+import java.util.Collections;
 import java.util.List;
+import java.util.Optional;
 
 import world.bentobox.bank.commands.AbstractBankCommand;
 import world.bentobox.bank.data.Money;
@@ -66,5 +68,10 @@ public boolean execute(User user, String label, List<String> args) {
         return true;
     }
 
+    @Override
+    public Optional<List<String>> tabComplete(User user, String alias, List<String> args) {
+        String balance = String.valueOf(addon.getBankManager().getBalance(user, getWorld()).getValue());
+        return Optional.of(Collections.singletonList(balance));
+    }
 
 }
diff --git a/src/test/java/world/bentobox/bank/commands/user/DepositCommandTest.java b/src/test/java/world/bentobox/bank/commands/user/DepositCommandTest.java
@@ -12,6 +12,8 @@
 import static org.mockito.Mockito.when;
 
 import java.util.Collections;
+import java.util.List;
+import java.util.Optional;
 import java.util.concurrent.CompletableFuture;
 
 import org.bukkit.World;
@@ -244,4 +246,14 @@ public void testCanExecuteOneArgNumberLowBalance() {
         verify(user).sendMessage(eq("bank.errors.low-balance"));
     }
 
+    /**
+     * Test method for {@link world.bentobox.bank.commands.user.DepositCommand#tabComplete(User, String, java.util.List)
+     */
+    @Test
+    public void testTabComplete() {
+        Optional<List<String>> value = dct.tabComplete(user, "", Collections.emptyList());
+        assertTrue(value.isPresent());
+        assertEquals("1000.0", value.get().get(0));
+    }
+
 }
diff --git a/src/test/java/world/bentobox/bank/commands/user/WithdrawCommandTest.java b/src/test/java/world/bentobox/bank/commands/user/WithdrawCommandTest.java
@@ -11,6 +11,8 @@
 import static org.mockito.Mockito.when;
 
 import java.util.Collections;
+import java.util.List;
+import java.util.Optional;
 import java.util.concurrent.CompletableFuture;
 
 import org.bukkit.World;
@@ -104,6 +106,8 @@ public void setUp() {
         PowerMockito.mockStatic(Util.class);
         when(Util.getWorld(any())).thenAnswer(arg -> arg.getArgument(0, World.class));
 
+        when(ic.getWorld()).thenReturn(world);
+
         wct = new WithdrawCommand(ic);
 
     }
@@ -231,4 +235,18 @@ public void testExecuteUserStringListOfStringSuccess() {
         verify(vh).deposit(eq(user), eq(123.3D), eq(world));
         verify(user).sendMessage(eq("bank.withdraw.success"), eq(TextVariables.NUMBER), eq("100.0"));
     }
+
+    /**
+     * Test method for {@link world.bentobox.bank.commands.user.WithdrawCommand#tabComplete(User, String, java.util.List)
+     */
+    @Test
+    public void testTabComplete() {
+        Optional<List<String>> value = wct.tabComplete(user, "", Collections.emptyList());
+        assertTrue(value.isPresent());
+        assertEquals("0.0", value.get().get(0));
+        when(bankManager.getBalance(any(), any())).thenReturn(new Money(12345D));
+        value = wct.tabComplete(user, "", Collections.emptyList());
+        assertEquals("12345.0", value.get().get(0));
+    }
+
 }