1 files changed, 47 insertions, 33 deletions
diff --git a/spec/models/foi_attachment_spec.rb b/spec/models/foi_attachment_spec.rb
index 88afc24a4..9583f4c76 100644
--- a/spec/models/foi_attachment_spec.rb
+++ b/spec/models/foi_attachment_spec.rb
@@ -18,45 +18,59 @@ require File.expand_path(File.dirname(__FILE__) + '/../spec_helper')
 
 describe FoiAttachment do
 
-    before(:each) do
-        load_raw_emails_data
-    end
+    describe :body= do
+
+        it "sets the body" do
+            attachment = FoiAttachment.new
+            attachment.body = "baz"
+            attachment.body.should == "baz"
+        end
+
+        it "sets the size" do
+            attachment = FoiAttachment.new
+            attachment.body = "baz"
+            attachment.body.should == "baz"
+            attachment.display_size.should == "0K"
+        end
+
+        it "reparses the body if it disappears" do
+            load_raw_emails_data
+            im = incoming_messages(:useless_incoming_message)
+            im.extract_attachments!
+            main = im.get_main_body_text_part
+            orig_body = main.body
+            main.delete_cached_file!
+            lambda {
+                im.get_main_body_text_part.body
+            }.should_not raise_error(Errno::ENOENT)
+            main.delete_cached_file!
+            main = im.get_main_body_text_part
+            main.body.should == orig_body
+        end
 
-    it "sets the body" do
-        attachment = FoiAttachment.new
-        attachment.body = "baz"
-        attachment.body.should == "baz"
-    end
-    it "sets the size" do
-        attachment = FoiAttachment.new
-        attachment.body = "baz"
-        attachment.body.should == "baz"
-        attachment.update_display_size!
-        attachment.display_size.should == "0K"
     end
-    it "reparses the body if it disappears" do
-        im = incoming_messages(:useless_incoming_message)
-        im.extract_attachments!
-        main = im.get_main_body_text_part
-        orig_body = main.body
-        main.delete_cached_file!
-        lambda {
-            im.get_main_body_text_part.body
-        }.should_not raise_error(Errno::ENOENT)
-        main.delete_cached_file!
-        main = im.get_main_body_text_part
-        main.body.should == orig_body
+
+    describe :ensure_filename! do
+
+        it 'should create a filename for an instance with a blank filename' do
+            attachment = FoiAttachment.new
+            attachment.filename = ''
+            attachment.ensure_filename!
+            attachment.filename.should == 'attachment.bin'
+        end
 
     end
-end
 
-describe FoiAttachment, "when ensuring a filename is present" do
+    describe :has_body_as_html? do
+
+        it 'should be true for a pdf attachment' do
+            FactoryGirl.build(:pdf_attachment).has_body_as_html?.should be_true
+        end
+
+        it 'should be false for an html attachment' do
+            FactoryGirl.build(:html_attachment).has_body_as_html?.should be_false
+        end
 
-    it 'should create a filename for an instance with a blank filename' do
-        attachment = FoiAttachment.new
-        attachment.filename = ''
-        attachment.ensure_filename!
-        attachment.filename.should == 'attachment.bin'
     end
 
 end