session_spec.rb in ronin-0.2.3

- old
+ new
@@ -1,56 +1,25 @@
 require 'ronin/sessions/session'
 
 require 'spec_helper'
+require 'sessions/classes/test_session'
+require 'sessions/classes/uses_test_session'
 
 describe Sessions::Session do
-  before(:all) do
-    module TestSession
-      include Sessions::Session
-
-      setup_session do
-        parameter :var, :default => :stuff, :description => 'Test parameter'
-      end
-
-      def test_one
-        'this_is_a_test'
-      end
+  describe "require_variable" do
+    before(:each) do
+      @obj = UsesTestSession.new
+      @obj.host = 'www.example.com'
     end
 
-    class TestSessionClass
-      include TestSession
+    it "should raise a VariableMissing exception if a variable is nil" do
+      lambda {
+        @obj.host = nil
+        @obj.connect
+      }.should raise_error(Sessions::VariableMissing)
     end
 
-    class TestSessionObject
+    it "should do nothing if the variable is not nil" do
+      @obj.connect.should == @obj.host
     end
-
-    @session_obj = TestSessionObject.new
-    @session_obj.extend TestSession
-  end
-
-  it "should add self.setup_session to a Module" do
-    TestSession.methods.include?('setup_session').should == true
-  end
-
-  describe "setup_session" do
-    it "should add self.included and self.extended methods once setup_session is called" do
-      TestSession.methods.include?('included').should == true
-      TestSession.methods.include?('extended').should == true
-    end
-
-    it "should run the setup_session proc once included" do
-      TestSessionClass.var.should == :stuff
-    end
-
-    it "should run the setup_session proc once extended" do
-      @session_obj.var.should == :stuff
-    end
-  end
-
-  it "should include Parameters once included" do
-    TestSessionClass.include?(Parameters).should == true
-  end
-
-  it "should extend Parameters once extended" do
-    @session_obj.kind_of?(Parameters).should == true
   end
 end