Add minimal tests for get/getg

Author: jhart-r7

lib/msf/ui/console/command_dispatcher/core.rb

@@ -2317,19 +2317,19 @@ def cmd_get(*args)
       global = true
     end
 
+    # No arguments?  No cookie.
+    if args.empty?
+      global ? cmd_getg_help : cmd_get_help
+      return false
+    end
+
     # Determine which data store we're operating on
     if (active_module && !global)
       datastore = active_module.datastore
     else
       datastore = framework.datastore
     end
 
-    # No arguments?  No cookie.
-    if args.empty?
-      global ? cmd_getg_help : cmd_get_help
-      return false
-    end
-
     args.each { |var| print_line("#{var} => #{datastore[var]}") }
   end
 

spec/lib/msf/ui/console/command_dispatcher/core_spec.rb

@@ -95,4 +95,25 @@ def cell(table, row, column)
       end
     end
   end
+
+  it { is_expected.to respond_to :cmd_get }
+  it { is_expected.to respond_to :cmd_getg }
+
+  describe "#cmd_get" do
+    describe "without arguments" do
+      it "should show a help message" do
+        core.cmd_get
+        @output.join.should =~ /Usage: get /
+      end
+    end
+  end
+
+  describe "#cmd_getg" do
+    describe "without arguments" do
+      it "should show a help message" do
+        core.cmd_getg
+        @output.join.should =~ /Usage: getg /
+      end
+    end
+  end
 end