diff --git a/tests/command_completion.cpp b/tests/command_completion.cpp
index c92bad0..8e8570f 100644
--- a/tests/command_completion.cpp
+++ b/tests/command_completion.cpp
@@ -21,7 +21,7 @@
     int expectedContextLength;
     SECTION("no prefix")
     {
-        expectedCompletions = {"cd", "copy", "create", "delete", "set", "commit", "get", "ls", "discard", "help", "describe", "move", "dump", "prepare", "exec", "cancel"};
+        expectedCompletions = {"cd", "copy", "create", "delete", "set", "commit", "get", "ls", "discard", "help", "describe", "move", "dump", "prepare", "exec", "cancel", "switch"};
         expectedContextLength = 0;
         SECTION("no space")
         {
@@ -75,6 +75,13 @@
         expectedContextLength = 0;
     }
 
+    SECTION("switch datastore targets")
+    {
+        input = "switch ";
+        expectedCompletions = {"running", "startup", "operational"};
+        expectedContextLength = 0;
+    }
+
     SECTION("dump")
     {
         input = "dump ";