Fixing tests after setg/unsetg changes

routersploit/test/test_completer.py

@@ -89,7 +89,7 @@ class RoutersploitCompleterTest(RoutersploitTestCase):
         self.set_module()
         self.rsf.send("\t\t")
         self.assertPrompt(
-            'back   check  exec   exit   help   run    set    show   \r\n',
+            'back   check  exec   exit   help   run    set    setg   show   \r\n',
             self.module_prompt('FTP Bruteforce')
         )
 
@@ -121,18 +121,29 @@ class RoutersploitCompleterTest(RoutersploitTestCase):
         self.set_module()
         self.rsf.send("s\t\t")
         self.assertPrompt(
-            'set    show   \r\n',
+            'set    setg   show   \r\n',
             self.module_prompt('FTP Bruteforce')
         )
 
     def test_complete_set_raw_2(self):
         self.set_module()
-        self.rsf.send("se\t\t")
+        self.rsf.send("se\t")
         self.assertPrompt(
             self.module_prompt('FTP Bruteforce'),
-            'set ',
+            'se\at',
         )
 
+    def test_complete_set_raw_3(self):
+        self.set_module()
+        self.rsf.send("set\t\t")
+        self.assertPrompt(
+            'set    setg   \r\n',
+            self.module_prompt('FTP Bruteforce'),
+        )
+
+    def test_complete_setg(self):  # TODO: cover setg/unsetg
+        pass
+
     def test_complete_set(self):
         self.set_module()
         self.rsf.send("set \t\t")

routersploit/test/test_interpreter.py

@@ -187,7 +187,7 @@ class RoutersploitInterpreterTest(RoutersploitTestCase):
     def test_suggested_commands_with_loaded_module(self):
         self.assertEqual(
             self.interpreter.suggested_commands(),
-            ['run', 'back', 'set ', 'show ', 'check', 'exec', 'help', 'exit']  # Extra space at the end because of following param
+            ['run', 'back', 'set ', 'setg ', 'show ', 'check', 'exec ', 'help', 'exit']  # Extra space at the end because of following param
         )
 
     def test_suggested_commands_without_loaded_module(self):