From b2e9e43f3dfcccac6608dc08cf20a353341811df Mon Sep 17 00:00:00 2001 From: Jon Hart Date: Fri, 16 Jan 2015 10:39:05 -0800 Subject: [PATCH] Add unit tests for RPC's getg --- spec/lib/msf/core/rpc/v10/rpc_core_spec.rb | 28 ++++++++++++++++++++++ 1 file changed, 28 insertions(+) create mode 100644 spec/lib/msf/core/rpc/v10/rpc_core_spec.rb diff --git a/spec/lib/msf/core/rpc/v10/rpc_core_spec.rb b/spec/lib/msf/core/rpc/v10/rpc_core_spec.rb new file mode 100644 index 0000000000..06cdd96025 --- /dev/null +++ b/spec/lib/msf/core/rpc/v10/rpc_core_spec.rb @@ -0,0 +1,28 @@ +# -*- coding:binary -*- +require 'spec_helper' + +require 'msf/core/rpc/v10/rpc_base' +require 'msf/core/rpc/v10/rpc_core' +require 'msf/core/rpc/v10/service' + +describe Msf::RPC::RPC_Core do + include_context 'Msf::Simple::Framework' + + let(:service) do + Msf::RPC::Service.new(framework) + end + + let(:core) do + Msf::RPC::RPC_Core.new(service) + end + + describe '#rpc_getg' do + it 'should show an empty value if the variable is unset' do + expect(core.rpc_getg('FOO')).to eq({'FOO' => ''}) + end + it 'should show the correct value if the variable is set' do + core.rpc_setg('FOO', 'BAR') + expect(core.rpc_getg('FOO')).to eq({'FOO' => 'BAR'}) + end + end +end