spec/agent_spec.rb in spidr-0.1.7 vs spec/agent_spec.rb in spidr-0.1.8

@@ -7,6 +7,82 @@
   include Helpers::Course
 
   before(:all) do
     @agent = run_course
   end
+
+  it "should provide the history" do
+    @agent.history.should_not be_empty
+  end
+
+  it "should provide the queue" do
+    @agent.queue.should be_empty
+  end
+
+  it "should be able to restore the history" do
+    agent = Agent.new
+    previous_history = [URI('http://www.example.com')]
+
+    agent.history = previous_history
+    agent.history.should == previous_history
+  end
+
+  it "should convert new histories to an Array of URIs" do
+    agent = Agent.new
+    previous_history = ['http://www.example.com']
+
+    agent.history = previous_history
+    agent.history.should_not == previous_history
+    agent.history.should == previous_history.map { |url| URI(url) }
+  end
+
+  it "should be able to restore the queue" do
+    agent = Agent.new
+    previous_queue = [URI('http://www.example.com')]
+
+    agent.queue = previous_queue
+    agent.queue.should == previous_queue
+  end
+
+  it "should convert new queues to an Array of URIs" do
+    agent = Agent.new
+    previous_queue = ['http://www.example.com']
+
+    agent.queue = previous_queue
+    agent.queue.should_not == previous_queue
+    agent.queue.should == previous_queue.map { |url| URI(url) }
+  end
+
+  it "should be able to pause spidering" do
+    count = 0
+    agent = Agent.host('spidr.rubyforge.org') do |spider|
+      spider.every_page do |page|
+        count += 1
+        spider.pause! if count >= 2
+      end
+    end
+
+    agent.should be_paused
+    agent.history.length.should == 2
+  end
+
+  it "should be able to continue spidering after being paused" do
+    agent = Agent.new do |spider|
+      spider.enqueue('http://spidr.rubyforge.org/')
+      spider.every_page do |page|
+        spider.pause!
+      end
+    end
+
+    agent.pause!
+    agent.continue!
+
+    agent.visited?('http://spidr.rubyforge.org/').should == true
+  end
+
+  it "should provide a to_hash method that returns the queue and history" do
+    hash = @agent.to_hash
+
+    hash[:queue].should be_empty
+    hash[:history].should_not be_empty
+  end
 end