agent_spec.rb in spidr-0.2.0

- old
+ new

@@ -18,25 +18,44 @@
     @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')]
+    previous_history = Set[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
+  it "should convert new histories to an Set of URIs" do
     agent = Agent.new
     previous_history = ['http://www.example.com']
+    expected_history = Set[URI('http://www.example.com')]
 
     agent.history = previous_history
     agent.history.should_not == previous_history
-    agent.history.should == previous_history.map { |url| URI(url) }
+    agent.history.should == expected_history
   end
 
+  it "should be able to restore the failures" do
+    agent = Agent.new
+    previous_failures = Set[URI('http://localhost/')]
+
+    agent.failures = previous_failures
+    agent.failures.should == previous_failures
+  end
+
+  it "should convert new histories to a Set of URIs" do
+    agent = Agent.new
+    previous_failures = ['http://localhost/']
+    expected_failures = Set[URI('http://localhost/')]
+
+    agent.failures = previous_failures
+    agent.failures.should_not == previous_failures
+    agent.failures.should == expected_failures
+  end
+
   it "should be able to restore the queue" do
     agent = Agent.new
     previous_queue = [URI('http://www.example.com')]
 
     agent.queue = previous_queue
@@ -44,40 +63,14 @@
   end
 
   it "should convert new queues to an Array of URIs" do
     agent = Agent.new
     previous_queue = ['http://www.example.com']
+    expected_queue = [URI('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
+    agent.queue.should == expected_queue
   end
 
   it "should provide a to_hash method that returns the queue and history" do
     hash = @agent.to_hash
 