1 files changed, 38 insertions, 3 deletions

diff --git a/spec/lib/i18n_interpolation.rb b/spec/lib/i18n_interpolation.rb
index 8c86413ad..e8d046757 100644
--- a/spec/lib/i18n_interpolation.rb
+++ b/spec/lib/i18n_interpolation.rb
@@ -1,6 +1,3 @@
-# This is a test of the set_content_type monkey patch in
-# lib/tmail_extensions.rb
-
 require File.expand_path(File.dirname(__FILE__) + '/../spec_helper')
 
 describe "when using i18n" do
@@ -11,5 +8,43 @@ describe "when using i18n" do
         result = _('Hello {{dip}}', :dip => 'hummus')
         result.should == 'Hello hummus'
     end
+
+    it "should assume that simple translations are always html safe" do
+      _("Hello").should be_html_safe
+    end
+
 end
 
+describe "gettext_interpolate" do
+    context "html unsafe string" do
+        let(:string) { "Hello {{a}}" }
+
+        it "should give an unsafe result" do
+          result = gettext_interpolate(string, :a => "foo")
+          result.should == "Hello foo"
+          result.should_not be_html_safe
+        end
+
+        it "should give an unsafe result" do
+          result = gettext_interpolate(string, :a => "foo".html_safe)
+          result.should == "Hello foo"
+          result.should_not be_html_safe
+        end        
+    end
+
+    context "html safe string" do
+        let(:string) { "Hello {{a}}".html_safe }
+
+        it "should quote the input if it's unsafe" do
+          result = gettext_interpolate(string, :a => "foo&")
+          result.should == "Hello foo&"
+          result.should be_html_safe
+        end
+
+        it "should not quote the input if it's safe" do
+          result = gettext_interpolate(string, :a => "foo&".html_safe)
+          result.should == "Hello foo&"
+          result.should be_html_safe
+        end        
+    end
+end