use polymorphic_url instead of url_for

lib/rewritten/document.rb

@@ -1,26 +1,29 @@
 module Rewritten
   module Document
 
-    def rewritten_url=(new_translation)
-      Rewritten.add_translation(new_translation, url_for(self)) 
-    end
 
     def rewritten_url
       return "" unless persisted? 
-      Rewritten.get_current_translation(url_for(self))
+      Rewritten.get_current_translation(polymorphic_url(self, :only_path => true))
+    end
+
+    def rewritten_url=(new_url)
+      if !new_url.nil? && new_url != "" && new_url != rewritten_url
+        Rewritten.add_translation(new_url, polymorphic_url(self, :only_path => true)) 
+      end
     end
 
     def rewritten_urls
       return [] unless persisted? 
-      Rewritten.get_all_translations(url_for(self))
+      Rewritten.get_all_translations(polymorphic_url(self, :only_path => true))
     end
 
     def has_rewritten_url?
-      Rewritten.exist_translation_for?(url_for(self))
+      Rewritten.exist_translation_for?(polymorphic_url(self, :only_path => true))
     end
 
     def remove_rewritten_urls
-      Rewritten.remove_all_translations(url_for(self))
+      Rewritten.remove_all_translations(polymorphic_url(self, :only_path => true))
     end
 
   end

test/rewritten/document_test.rb

@@ -7,7 +7,7 @@ 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
 
@@ -35,7 +35,6 @@ def @instance.persisted?; false; end
     Rewritten.add_translation('/foo/baz', '/products/123')
 
     @instance.rewritten_urls.must_equal ['/foo/bar', '/foo/baz']         
-
   end
 
   it 'must add a new translation' do
@@ -52,6 +51,17 @@ def @instance.persisted?; false; end
     @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