Finished specs for Google::Search::Web

melborne · Jul 23, 2009 · 48db81d · 48db81d
1 parent 13d7f2b
commit 48db81d
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 4 deletions.
diff --git a/lib/google-search/search/web.rb b/lib/google-search/search/web.rb
@@ -17,7 +17,7 @@ class Web < self
       #   - :off
       #
 
-      attr_accessor :safe
+      attr_accessor :safety_level
 
       ##
       # Weither or not to filter duplicate results.
@@ -28,17 +28,17 @@ class Web < self
       #:nodoc:
 
       def initialize options = {}, &block
-        @safe = options.delete :safe
+        @safe = options.delete :safety_level
         @filter = options.delete(:filter) || 1
         super :web, options, &block
       end
 
       #:nodoc:
 
       def get_uri_params
-        raise Error, "invalid safety level `#{safe}'" unless safe.nil? || SAFETY_LEVELS.include?(safe)
+        raise Error, "invalid safety level `#{safety_level}'" unless safety_level.nil? || SAFETY_LEVELS.include?(safety_level)
         super + [
-          [:safe, safe],
+          [:safe, safety_level],
           [ :filter, filter ? 1 : 0]]
       end
     end

diff --git a/spec/search_spec.rb b/spec/search_spec.rb
@@ -96,6 +96,18 @@
   end
 
   describe "#get_uri" do
+    describe "safety_level" do
+      it "should validate" do
+        @search.safety_level = :moderate
+        lambda { @search.get_uri }.should_not raise_error
+      end
+
+      it "should raise an error when invalid" do
+        @search.safety_level = :foo
+        lambda { @search.get_uri }.should raise_error(Google::Search::Error, /safety/)
+      end
+    end
+
     describe "filter" do
       it "should default to 1" do
         @search.get_uri.should include('filter=1')