Add #import_option and #from_file

Author: egypt

spec/lib/msf/core/data_store_spec.rb

@@ -18,10 +18,25 @@
 		subject["FoO"].should == "bar"
 		subject["foO"].should == "bar"
 	end
+	context "#to_h" do
+		it "should return a Hash with correct values" do
+			subject.to_h.should == { "foo" => "bar", "fizz" => "buzz" }
+		end
+	end
 end
 
 describe Msf::DataStore do
 
+	describe "#import_option" do
+		subject do
+			s = described_class.new
+			s.import_option("foo", "bar")
+			s.import_option("fizz", "buzz")
+			s
+		end
+		it_behaves_like "datastore"
+	end
+
 	describe "#import_options_from_hash" do
 		subject do
 			hash = { "foo" => "bar", "fizz" => "buzz" }
@@ -42,4 +57,22 @@
 		it_behaves_like "datastore"
 	end
 
+	describe "#from_file" do
+		subject do
+			ini_instance = double
+			ini_instance.stub(:group?).and_return(true)
+			ini_instance.stub(:[]).and_return( { "foo" => "bar", "fizz" => "buzz" } )
+
+			ini = stub_const("Rex::Parser::Ini", Class.new)
+			ini.stub(:from_file).and_return(ini_instance)
+
+			s = described_class.new
+			s.from_file("path")
+			s
+		end
+
+		it_behaves_like "datastore"
+	end
+
+
 end