test/rewritten/document_test.rb in rewritten-0.12.0 vs test/rewritten/document_test.rb in rewritten-0.12.1

@@ -5,11 +5,11 @@
   before do
     class Product
       include Rewritten::Document
     end
     @instance = Product.new
-    def @instance.url_for(object); '/products/123'; end
+    def @instance.polymorphic_url(object, options={}); '/products/123'; end
     def @instance.persisted?; true; end
   end
 
   it 'must add the rewritten methods to the class' do
     @instance.respond_to?(:rewritten_url=).must_equal true
@@ -33,11 +33,10 @@
 
     Rewritten.add_translation('/foo/bar', '/products/123')
     Rewritten.add_translation('/foo/baz', '/products/123')
 
     @instance.rewritten_urls.must_equal ['/foo/bar', '/foo/baz']         
-
   end
 
   it 'must add a new translation' do
     @instance.has_rewritten_url?.must_equal false
     @instance.rewritten_url = '/foo/baz'  
@@ -48,9 +47,20 @@
   it 'must remove all translation' do
     @instance.rewritten_url = '/foo/bar'  
     @instance.rewritten_url = '/foo/baz'  
     @instance.remove_rewritten_urls
     @instance.rewritten_urls.must_equal []
+  end
+
+  it 'must won\'t add blank and similar translations' do
+    @instance.rewritten_url = '/foo/bar'  
+    @instance.rewritten_urls.must_equal ['/foo/bar']
+    @instance.rewritten_url = nil
+    @instance.rewritten_urls.must_equal ['/foo/bar']
+    @instance.rewritten_url = ""
+    @instance.rewritten_urls.must_equal ['/foo/bar']
+    @instance.rewritten_url = "/foo/bar"
+    @instance.rewritten_urls.must_equal ['/foo/bar']
   end
 
 end