1 files changed, 56 insertions, 0 deletions
diff --git a/spec/lib/mail_handler/mail_handler_spec.rb b/spec/lib/mail_handler/mail_handler_spec.rb
index 6213d3fdf..805fd6800 100644
--- a/spec/lib/mail_handler/mail_handler_spec.rb
+++ b/spec/lib/mail_handler/mail_handler_spec.rb
@@ -21,3 +21,59 @@ describe 'when creating a mail object from raw data' do
     end
 
 end
+
+describe 'when asked for the from name' do
+
+    it 'should return nil if there is a blank "From" field' do
+        mail_data = load_file_fixture('incoming-request-plain.email')
+        mail_data.gsub!('EMAIL_FROM', '')
+        mail = MailHandler.mail_from_raw_email(mail_data)
+        MailHandler.get_from_name(mail).should == nil
+    end
+
+    it 'should correctly return an encoded name from the from field' do
+        mail = get_fixture_mail('quoted-subject-iso8859-1.email')
+        MailHandler.get_from_name(mail).should == 'Coordenação de Relacionamento, Pesquisa e Informação/CEDI'
+    end
+
+    it 'should get a name from a "From" field with a name and address' do
+        mail = get_fixture_mail('incoming-request-oft-attachments.email')
+        MailHandler.get_from_name(mail).should == 'Public Authority'
+    end
+
+    it 'should return nil from a "From" field that is just a name'do
+        mail = get_fixture_mail('track-response-webshield-bounce.email')
+        MailHandler.get_from_name(mail).should == nil
+    end
+
+end
+
+describe 'when asked for the from address' do
+
+    it 'should return nil if there is a blank "From" field' do
+        mail_data = load_file_fixture('incoming-request-plain.email')
+        mail_data.gsub!('EMAIL_FROM', '')
+        mail = MailHandler.mail_from_raw_email(mail_data)
+        MailHandler.get_from_address(mail).should == nil
+    end
+
+    it 'should correctly return an address from a mail that has an encoded name in the from field' do
+        mail = get_fixture_mail('quoted-subject-iso8859-1.email')
+        MailHandler.get_from_address(mail).should == 'geraldinequango@localhost'
+    end
+
+    it 'should return nil if there is no address in the "From" field' do
+        mail = get_fixture_mail('track-response-webshield-bounce.email')
+        MailHandler.get_from_address(mail).should == nil
+    end
+
+    it 'should return the "From" email address if there is one' do
+         mail = get_fixture_mail('track-response-abcmail-oof.email')
+        MailHandler.get_from_address(mail).should == 'Name.Removed@example.gov.uk'
+    end
+
+    it 'should get an address from a "From" field with a name and address' do
+        mail = get_fixture_mail('incoming-request-oft-attachments.email')
+        MailHandler.get_from_address(mail).should == 'public@authority.gov.uk'
+    end
+end