entp-astrotrain 0.2.0

data/test/message_test.rb

@@ -0,0 +1,424 @@
+require File.join(File.dirname(__FILE__), "test_helper")
+
+class Astrotrain::MessageTest < Astrotrain::TestCase
+  describe "mapping" do
+    describe "against default domain" do
+      before :all do
+        Astrotrain::Mapping.transaction do
+          Astrotrain::LoggedMail.all.destroy!
+          Astrotrain::Mapping.all.destroy!
+          @mapping  = Astrotrain::Mapping.create!(:email_user => 'xyz')
+          @mapping2 = Astrotrain::Mapping.create!(:email_user => 'xyz',  :email_domain => 'sample.com')
+        end
+      end
+
+      describe "without mapping" do
+        before do
+          Astrotrain::LoggedMail.all.destroy!
+        end
+
+        it "doesn't log message" do
+          @msg = Astrotrain::Message.receive(mail(:basic))
+          @log = Astrotrain::LoggedMail.first
+          assert_nil @log
+        end
+
+        it "logs message if Astrotrain::LoggedMail.log_processed" do
+          Astrotrain::LoggedMail.log_processed = true
+          @msg = Astrotrain::Message.receive(mail(:basic))
+          @log = Astrotrain::LoggedMail.first
+          assert @log
+          assert @log.error_message.blank?
+        end
+
+        it "calls pre_mapping callback" do
+          Astrotrain::LoggedMail.log_processed = true
+          callback_msg = nil
+          Astrotrain.callback(:pre_mapping) do |message|
+            callback_msg = message
+          end
+
+          @msg = Astrotrain::Message.receive(mail(:mapped))
+          assert_equal callback_msg, @msg
+        end
+
+        it "calls pre_processing callback" do
+          Astrotrain::LoggedMail.log_processed = true
+          callback_msg, callback_map = nil
+          Astrotrain.callback(:pre_processing) do |message, mapping|
+            callback_msg = message
+            callback_map = mapping
+          end
+
+          @msg = Astrotrain::Message.receive(mail(:mapped))
+          @log = Astrotrain::LoggedMail.first
+          assert_equal callback_msg, @msg
+          assert_equal callback_map, @log.mapping
+        end
+
+        it "calls post_processing callback" do
+          Astrotrain::LoggedMail.log_processed = true
+          callback_msg, callback_map, callback_log = nil
+          Astrotrain.callback(:post_processing) do |message, mapping, log|
+            callback_msg = message
+            callback_map = mapping
+            callback_log = log
+          end
+
+          @msg = Astrotrain::Message.receive(mail(:mapped))
+          @log = Astrotrain::LoggedMail.first
+          assert_equal callback_msg, @msg
+          assert_equal callback_map, @log.mapping
+          assert_equal callback_log, @log
+        end
+
+        after do
+          Astrotrain::LoggedMail.log_processed = false
+          Astrotrain.clear_callbacks
+        end
+      end
+
+      describe "erroring" do
+        before do
+          Astrotrain::LoggedMail.all.destroy!
+        end
+
+        it "logs message without mappping" do
+          stub(Astrotrain::Mapping).match { raise RuntimeError }
+          @msg = Astrotrain::Message.receive(mail(:basic))
+          @log = Astrotrain::LoggedMail.first
+          assert @log
+          assert_nil @log.delivered_at
+          assert_match /RuntimeError/, @log.error_message
+          assert_nil @log.mapping
+        end
+
+        it "logs message with mappping" do
+          stub(Astrotrain::Mapping).match {@mapping}
+          stub(@mapping).process { raise RuntimeError }
+          @msg = Astrotrain::Message.receive(mail(:basic))
+          @log = Astrotrain::LoggedMail.first
+          assert @log
+          assert_nil @log.delivered_at
+          assert_match /RuntimeError/, @log.error_message
+          assert_equal @mapping, @log.mapping
+        end
+      end
+
+      describe "with mapping" do
+        before :all do
+          @msg = Astrotrain::Message.receive(mail(:mapped))
+          @log = Astrotrain::LoggedMail.first
+        end
+
+        it "does not log message" do
+          assert_nil @log
+        end
+      end
+    end
+  end
+
+  describe "parsing" do
+    before :all do
+      @body = "---------- Forwarded message ----------\nblah blah"
+    end
+
+    describe "basic, single sender/recipient" do
+      before :all do
+        @raw     = mail(:basic)
+        @message = Astrotrain::Message.parse(@raw)
+      end
+
+      it "parses TMail::Mail headers" do
+        expected = {'mime-version' => '1.0', 'content-type' => 'text/plain; charset=ISO-8859-1', 'to' => 'Processor <processor@astrotrain.com>',
+          'x-custom' => 'reply', 'content-transfer-encoding' => '7bit', 'content-disposition' => 'inline', 'message-id' => '<a16be7390810161014n52b603e9k1aa6bb803c6735aa@mail.gmail.com>'}
+        assert_equal expected, @message.headers
+      end
+
+      it "#parse parses TMail::Mail object from raw text" do
+        assert_kind_of TMail::Mail, @message.mail
+      end
+
+      it "recognizes Delivered-To and To: headers as recipients" do
+        assert_equal %w(processor@astrotrain.com), @message.recipients
+      end
+
+      it "recognizes From: header as sender" do
+        assert_equal %(Bob <user@example.com>), @message.sender
+      end
+
+      it "recognizes Subject: header" do
+        assert_equal 'Fwd: blah blah', @message.subject
+      end
+
+      it "recognizes message body" do
+        assert_equal @body, @message.body
+      end
+
+      it "retains raw message" do
+        assert_equal @raw, @message.raw
+      end
+    end
+    
+    describe "iso 8859 1 encoded headers" do
+      before :all do
+        @raw     = mail("iso-8859-1")
+        @message = Astrotrain::Message.parse(@raw)
+      end
+
+      it "recognizes From: header with strange encoding" do
+        assert_equal %(Matthéw <user@example.com>), @message.sender
+      end
+    end
+
+    describe "gb2312 encoded body" do
+      before :all do
+        @raw     = mail("gb2312_encoding")
+        @message = Astrotrain::Message.parse(@raw)
+      end
+
+      it "converts to UTF-8" do
+        assert_equal "Dear Sirs, \r\nWe are given to understand that you are  Manufacturer of  plstic  Bottles\r\nAdd： blah China", 
+          @message.body
+      end
+    end
+
+    describe "gb2312 encoded body with invalid charset in mime version header" do
+      before :all do
+        @raw     = mail("gb2312_encoding_invalid")
+        @message = Astrotrain::Message.parse(@raw)
+      end
+
+      it "converts to UTF-8" do
+        assert_equal "Dear Sirs, \r\nWe are given to understand that you are  Manufacturer of  plstic  Bottles\r\nAdd： blah China",
+          @message.body
+      end
+    end
+
+    describe "utf 8 encoded headers" do
+      before :all do
+        @raw     = mail("utf-8")
+        @message = Astrotrain::Message.parse(@raw)
+      end
+      
+      it "recognizes From: header with strange encoding" do
+        assert_equal %(isnard naiké <user@example.com>), @message.sender
+      end
+    end
+
+    describe "multipart message with name property on Content Type" do
+      before :all do
+        @raw     = mail(:multipart)
+        @message = Astrotrain::Message.parse(@raw)
+      end
+
+      it "#parse parses TMail::Mail object from raw text" do
+        assert_kind_of TMail::Mail, @message.mail
+      end
+
+      it "recognizes Delivered-To/To: headers as recipient" do
+        assert_equal %w(foo@example.com), @message.recipients
+      end
+
+      it "recognizes message body" do
+        assert_equal "Testing out rich emails with attachments!\nThis one has a name property on Content-Type.\n[state:hold responsible:rick]\n\n",
+          @message.body
+      end
+
+      it "retrieves attachments" do
+        assert_equal 1, @message.attachments.size
+      end
+
+      it "retrieves attachment filename" do
+        assert_equal 'bandit.jpg', @message.attachments.first.filename
+      end
+
+      it "retrieves attachment content_type" do
+        assert_equal 'image/jpeg', @message.attachments.first.content_type
+      end
+    end
+
+    describe "multipart message with filename property on Content Disposition" do
+      before :all do
+        @raw     = mail(:multipart2)
+        @message = Astrotrain::Message.parse(@raw)
+      end
+
+      it "#parse parses TMail::Mail object from raw text" do
+        assert_kind_of TMail::Mail, @message.mail
+      end
+
+      it "recognizes Delivered-To/To: headers as recipient" do
+        assert_equal %w(foo@example.com), @message.recipients
+      end
+
+      it "recognizes message body" do
+        assert_equal "Testing out rich emails with attachments!\nThis one has NO name property on Content-Type.\n[state:hold responsible:rick]\n\n", 
+          @message.body
+      end
+
+      it "retrieves attachments" do
+        assert_equal 1, @message.attachments.size
+      end
+
+      it "retrieves attachment filename" do
+        assert_equal 'bandit.jpg', @message.attachments.first.filename
+      end
+
+      it "retrieves attachment content_type" do
+        assert_equal 'image/jpeg', @message.attachments.first.content_type
+      end
+    end
+
+    describe "apple multipart message" do
+      before :all do
+        @raw     = mail(:apple_multipart)
+        @message = Astrotrain::Message.parse(@raw)
+      end
+
+      it "#parse parses TMail::Mail object from raw text" do
+        assert_kind_of TMail::Mail, @message.mail
+      end
+
+      it "recognizes To: header as recipient" do
+        assert_equal %w(foo@example.com), @message.recipients
+      end
+
+      it "recognizes message body" do
+        assert_equal "Let's have a test here:\r\n\r\n\r\n\nYum\r\n\r\n\r\nOn Feb 10, 2009, at 3:37 PM, Tender Support wrote:\r\n\r\n> // Add your reply above here\r\n> ==================================================\r\n> From: Tyler Durden\r\n> Subject: Email attachments and file upload\r\n>\r\n> not at the moment ... let me test\r\n>\r\n> View this Discussion online: http://foobar.com\r\n> .\r\n\r\n\r\n\r\n\r\n--Apple-Mail-7-451386929--", 
+          @message.body
+      end
+
+      it "retrieves attachments" do
+        assert_equal 1, @message.attachments.size
+      end
+
+      it "retrieves attachment filename" do
+        assert_equal 'logo.gif', @message.attachments.first.filename
+      end
+
+      it "retrieves attachment content_type" do
+        assert_equal 'image/gif', @message.attachments.first.content_type
+      end
+    end
+
+    describe "multiple sender/recipients" do
+      before :all do
+        @raw     = mail(:multiple)
+        @message = Astrotrain::Message.parse(@raw)
+      end
+
+      it "#parse parses TMail::Mail object from raw text" do
+        assert_kind_of TMail::Mail, @message.mail
+      end
+
+      it "recognizes To: headers as recipients" do
+        assert_equal %w(processor@astrotrain.com other@example.com), @message.recipients
+      end
+
+      it "recognizes To: headers as recipients with custom header order" do
+        assert_equal %w(other@example.com processor@astrotrain.com), @message.recipients(%w(to original_to delivered_to))
+      end
+
+      it "recognizes From: header as sender" do
+        assert_equal %(user@example.com, boss@example.com), @message.sender
+      end
+
+      it "recognizes Subject: header" do
+        assert_equal 'Fwd: blah blah', @message.subject
+      end
+
+      it "recognizes message body" do
+        assert_equal @body, @message.body
+      end
+
+      it "retains raw message" do
+        assert_equal @raw, @message.raw
+      end
+    end
+
+    describe "recipients in the body" do
+      before :all do
+        @raw     = mail(:multiple_with_body_recipients)
+        @message = Astrotrain::Message.parse(@raw)
+      end
+
+      it "recognizes in-body emails and To: headers as recipients" do
+        assert_equal %w(processor+foobar@astrotrain.com processor+blah@astrotrain.com processor@astrotrain.com other@example.com), 
+          @message.recipients
+      end
+    end
+
+    describe "with only HTML body" do
+      before :all do
+        @raw     = mail(:html)
+        @message = Astrotrain::Message.parse(@raw)
+      end
+
+      it "parses emtpy body" do
+        assert_equal '', @message.body
+      end
+
+      it "parses HTML body" do
+        assert_equal "<p>ABC</p>\n------", @message.html
+      end
+    end
+
+    describe "with X Original To header" do
+      before :all do
+        @raw     = mail(:custom)
+        @message = Astrotrain::Message.parse(@raw)
+      end
+
+      it "#parse parses TMail::Mail object from raw text" do
+        assert_kind_of TMail::Mail, @message.mail
+      end
+
+      it "recognizes X-Original-to: header as recipient" do
+        assert_equal %w(processor-reply-57@custom.com processor-delivered@astrotrain.com processor@astrotrain.com), @message.recipients
+      end
+
+      it "recognizes Delivered-To: header as recipient with custom header order" do
+        assert_equal %w(processor-delivered@astrotrain.com processor-reply-57@custom.com processor@astrotrain.com), @message.recipients(%w(delivered_to original_to to))
+      end
+
+      it "recognizes To: header as recipient with custom header order" do
+        assert_equal %w(processor@astrotrain.com processor-reply-57@custom.com processor-delivered@astrotrain.com), @message.recipients(%w(to original_to delivered_to))
+      end
+
+      it "recognizes From: header as sender" do
+        assert_equal %(user@example.com, boss@example.com), @message.sender
+      end
+
+      it "recognizes Subject: header" do
+        assert_equal 'Fwd: blah blah', @message.subject
+      end
+
+      it "recognizes message body" do
+        assert_equal @body, @message.body
+      end
+
+      it "retains raw message" do
+        assert_equal @raw, @message.raw
+      end
+    end
+
+    describe "with multiple Delivered To headers" do
+      before :all do
+        @raw     = mail(:multiple_delivered_to)
+        @message = Astrotrain::Message.parse(@raw)
+      end
+
+      it "recognizes Delivered-to: header as recipient" do
+        assert_equal %w(processor-reply-57@custom.com processor-delivered@astrotrain.com processor@astrotrain.com), @message.recipients
+      end
+    end
+  end
+
+  describe "queueing" do
+    it "writes contents queue path" do
+      filename = Astrotrain::Message.queue("boo!")
+      assert_equal 'boo!', IO.read(filename)
+    end
+  end
+end

data/test/test_helper.rb

@@ -0,0 +1,54 @@
+$testing = true
+$LOAD_PATH.unshift File.join(File.dirname(__FILE__), '..', 'lib')
+
+require "rubygems"
+require "context"
+require 'rr'
+require 'astrotrain'
+require 'astrotrain/api'
+require 'sinatra/test'
+
+module Astrotrain
+  load File.dirname(__FILE__) do
+    DataMapper.setup(:default, {
+      :adapter  => "mysql",
+      :database => "astrotrain_test",
+      :username => "root",
+      :host     => "localhost"
+    })
+  end
+
+  LoggedMail.log_path = Astrotrain.root / 'messages'
+  FileUtils.rm_rf   LoggedMail.log_path
+  FileUtils.mkdir_p LoggedMail.log_path
+
+  Message.queue_path = root / 'fixtures' / 'queue'
+  FileUtils.rm_rf   Message.queue_path
+  FileUtils.mkdir_p Message.queue_path
+
+  class TestCase < Test::Unit::TestCase
+    include RR::Adapters::RRMethods
+
+    before do
+      RR.reset
+    end
+
+    after do
+      RR.verify
+    end
+
+    def mail(filename)
+      IO.read(File.join(File.dirname(__FILE__), 'fixtures', "#{filename}.txt"))
+    end
+  end
+
+  class ApiTestCase < TestCase
+    include Sinatra::Test
+  end
+end
+
+begin
+  require 'ruby-debug'
+  Debugger.start
+rescue LoadError
+end

data/test/transport_test.rb

@@ -0,0 +1,111 @@
+require File.join(File.dirname(__FILE__), "test_helper")
+
+class Astrotrain::TransportTest < Astrotrain::TestCase
+  describe "HttpPost" do
+    before :all do
+      @post    = 'http://example.com'
+      @message = Astrotrain::Message.parse(mail(:custom))
+      @mapping = Astrotrain::Mapping.new(:destination => @post, :transport => 'http_post')
+    end
+
+    before do
+      @trans           = Astrotrain::Mapping::HttpPost.new(@message, @mapping, @message.recipients(%w(delivered_to)).first)
+      @expected_fields = @trans.fields.merge(:emails => @message.recipients(%w(original_to to)) * ",")
+    end
+
+    it "sets #fields" do
+      expected = {:subject => @message.subject, :from => @message.sender, :to => @message.recipients(%w(delivered_to)).first, :body => @message.body, :html => @message.html, :emails => @message.recipients(%w(original_to to)),
+        "headers[reply-to]" => "reply-to-me@example.com", 'headers[message-id]' => '<a16be7390810161014n52b603e9k1aa6bb803c6735aa@mail.gmail.com>', 'headers[to]' => "processor@astrotrain.com",
+        "headers[mime-version]"=>"1.0", "headers[content-type]"=>"text/plain; charset=ISO-8859-1", "headers[content-disposition]"=>"inline", "headers[content-transfer-encoding]"=>"7bit"}
+      assert_equal expected, @trans.fields
+    end
+
+    it "adds attachments to #fields" do
+      @multipart = Astrotrain::Message.parse(mail(:multipart))
+      @trans     = Astrotrain::Mapping::HttpPost.new(@multipart, @mapping, @multipart.recipients.first)
+      assert_equal @multipart.attachments.first, @trans.fields[:attachments_0]
+    end
+
+    it "sets fields with mapping separator set" do
+      @message = Astrotrain::Message.parse(mail(:reply))
+      @mapping.separator = "=" * 5
+      @trans   = Astrotrain::Mapping::HttpPost.new(@message, @mapping, @message.recipients.first)
+      assert_equal "blah blah", @trans.fields[:body]
+    end
+
+    describe "when processing" do
+      before do
+        stub(Astrotrain::Mapping::HttpPost).new {@trans}
+      end
+
+      it "makes http post request" do
+        mock(RestClient).post(@mapping.destination, @expected_fields)
+        @trans.process
+      end
+
+      it "makes http post request from Transport" do
+        mock(RestClient).post(@mapping.destination, @expected_fields)
+        Astrotrain::Mapping::Transport.process(@message, @mapping, @message.recipients.first)
+      end
+
+      before :all do
+        Astrotrain::Mapping::Transport.processing = true
+      end
+
+      after :all do
+        Astrotrain::Mapping::Transport.processing = false
+      end
+    end
+  end
+
+  describe "jabber" do
+    before :all do
+      @dest    = 'foo@bar.com'
+      @message = Astrotrain::Message.parse(mail(:custom))
+      @mapping = Astrotrain::Mapping.new(:destination => @dest, :transport => 'jabber')
+    end
+
+    before do
+      @trans   = Astrotrain::Mapping::Jabber.new(@message, @mapping, @message.recipients(%w(delivered_to)).first)
+    end
+
+    it "sets #content" do
+      expected = "From: %s\nTo: %s\nSubject: %s\nEmails: %s\n%s" % [@message.sender, @message.recipients(%w(delivered_to)).first, @message.subject, @message.recipients(%w(original_to to)) * ", ", @message.body]
+      assert_equal expected, @trans.content
+    end
+
+    it "sets content with mapping separator set" do
+      @message = Astrotrain::Message.parse(mail(:reply))
+      @mapping.separator = "=" * 5
+      @trans   = Astrotrain::Mapping::Jabber.new(@message, @mapping, @message.recipients(%w(delivered_to)).first)
+      expected = "From: %s\nTo: %s\nSubject: %s\nEmails: %s\n%s" % [@message.sender, @message.recipients(%w(delivered_to)).first, @message.subject, '', "blah blah"]
+      assert_equal expected, @trans.content
+    end
+
+    describe "when processing" do
+      before do
+        @conn = Object.new
+        stub(@trans).connection { @conn }
+        stub(Astrotrain::Mapping::Jabber).new { @trans }
+      end
+
+      it "makes jabber delivery" do
+        mock(@conn).deliver(@mapping.destination, @trans.content)
+        @trans.process
+      end
+
+      it "makes jabber delivery from Transport" do
+        mock(@conn).deliver(@mapping.destination, @trans.content)
+        Astrotrain::Mapping::Transport.process(@message, @mapping, @message.recipients(%w(delivered_to)).first)
+      end
+
+      before :all do
+        Astrotrain::Mapping::Transport.processing = true
+      end
+
+      after :all do
+        Astrotrain::Mapping::Transport.processing = false
+      end
+    end
+  end
+end

metadata

@@ -0,0 +1,115 @@
+--- !ruby/object:Gem::Specification 
+name: entp-astrotrain
+version: !ruby/object:Gem::Version 
+  version: 0.2.0
+platform: ruby
+authors: 
+- technoweenie
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2009-09-21 00:00:00 -07:00
+default_executable: 
+dependencies: []
+
+description: 
+email: technoweenie@gmail.com
+executables: []
+
+extensions: []
+
+extra_rdoc_files: 
+- LICENSE
+- README
+files: 
+- .gitignore
+- LICENSE
+- README
+- Rakefile
+- VERSION
+- astrotrain.gemspec
+- config/sample.rb
+- lib/astrotrain.rb
+- lib/astrotrain/api.rb
+- lib/astrotrain/logged_mail.rb
+- lib/astrotrain/mapping.rb
+- lib/astrotrain/mapping/http_post.rb
+- lib/astrotrain/mapping/jabber.rb
+- lib/astrotrain/mapping/transport.rb
+- lib/astrotrain/message.rb
+- lib/astrotrain/tmail.rb
+- lib/vendor/rest-client/README.rdoc
+- lib/vendor/rest-client/Rakefile
+- lib/vendor/rest-client/bin/restclient
+- lib/vendor/rest-client/foo.diff
+- lib/vendor/rest-client/lib/rest_client.rb
+- lib/vendor/rest-client/lib/rest_client/net_http_ext.rb
+- lib/vendor/rest-client/lib/rest_client/payload.rb
+- lib/vendor/rest-client/lib/rest_client/request_errors.rb
+- lib/vendor/rest-client/lib/rest_client/resource.rb
+- lib/vendor/rest-client/rest-client.gemspec
+- lib/vendor/rest-client/spec/base.rb
+- lib/vendor/rest-client/spec/master_shake.jpg
+- lib/vendor/rest-client/spec/payload_spec.rb
+- lib/vendor/rest-client/spec/request_errors_spec.rb
+- lib/vendor/rest-client/spec/resource_spec.rb
+- lib/vendor/rest-client/spec/rest_client_spec.rb
+- tasks/doc.thor
+- tasks/merb.thor
+- test/api_test.rb
+- test/fixtures/apple_multipart.txt
+- test/fixtures/basic.txt
+- test/fixtures/custom.txt
+- test/fixtures/fwd.txt
+- test/fixtures/gb2312_encoding.txt
+- test/fixtures/gb2312_encoding_invalid.txt
+- test/fixtures/html.txt
+- test/fixtures/iso-8859-1.txt
+- test/fixtures/mapped.txt
+- test/fixtures/multipart.txt
+- test/fixtures/multipart2.txt
+- test/fixtures/multiple.txt
+- test/fixtures/multiple_delivered_to.txt
+- test/fixtures/multiple_with_body_recipients.txt
+- test/fixtures/reply.txt
+- test/fixtures/utf-8.txt
+- test/logged_mail_test.rb
+- test/mapping_test.rb
+- test/message_test.rb
+- test/test_helper.rb
+- test/transport_test.rb
+has_rdoc: false
+homepage: http://github.com/entp/astrotrain
+licenses: 
+post_install_message: 
+rdoc_options: 
+- --charset=UTF-8
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.3.5
+signing_key: 
+specification_version: 3
+summary: email => http post
+test_files: 
+- test/api_test.rb
+- test/logged_mail_test.rb
+- test/mapping_test.rb
+- test/message_test.rb
+- test/test_helper.rb
+- test/transport_test.rb