bbmb 2.0.0

data/test/util/test_polling_manager.rb

@@ -0,0 +1,312 @@
+#!/usr/bin/env ruby
+# Util::TestPollingManager -- bbmb.ch -- 14.09.2006 -- hwyss@ywesee.com
+
+$: << File.expand_path('../../lib', File.dirname(__FILE__))
+
+require 'test/unit'
+require 'flexmock'
+require 'bbmb/util/polling_manager'
+require 'fileutils'
+
+module BBMB
+  module Util
+    class TestFileMission < Test::Unit::TestCase
+      include FlexMock::TestCase
+      def setup
+        @datadir = File.expand_path('../data', File.dirname(__FILE__))
+        BBMB.config = flexmock('config')
+        BBMB.config.should_receive(:bbmb_dir).and_return(@datadir)
+        @mission = FileMission.new
+        @dir = File.expand_path('../data/poll', File.dirname(__FILE__))
+        FileUtils.mkdir_p(@dir)
+        @mission.directory = @dir
+      end
+      def test_poll
+        path = File.join(@dir, 'test.csv')
+        File.open(path, 'w') { |fh| fh.puts 'data' }
+        @mission.poll { |name, io|
+          assert_equal('test.csv', name)
+          assert_equal("data\n", io.read)
+        }
+        assert_equal(true, File.exist?(path))
+
+        @mission.glob_pattern = '*.xls'
+        @mission.poll { |name, io|
+          flunk "glob_pattern *.xls should not match any files, matched #{name}"
+        }
+
+        @mission.glob_pattern = '*.csv'
+        @mission.poll { |name, io|
+          assert_equal('test.csv', name)
+          assert_equal("data\n", io.read)
+        }
+      end
+      def test_poll_path
+        path = File.join(@dir, 'test.csv')
+        File.open(path, 'w') { |fh| fh.puts 'data' }
+        @mission.poll_path(path) { |name, io|
+          assert_equal('test.csv', name)
+          assert_equal("data\n", io.read)
+        }
+        assert_equal(true, File.exist?(path))
+      end
+      def test_poll_path__delete
+        path = File.join(@dir, 'test.csv')
+        File.open(path, 'w') { |fh| fh.puts 'data' }
+        @mission.delete = true
+        @mission.poll_path(path) { |name, io|
+          assert_equal('test.csv', name)
+          assert_equal("data\n", io.read)
+        }
+        assert_equal(false, File.exist?(path))
+      end
+      def test_poll_path__backup
+        path = File.join(@dir, 'test.csv')
+        File.open(path, 'w') { |fh| fh.puts 'data' }
+        bdir = @mission.backup_dir = File.expand_path('../backup', @dir)
+        bpath = File.join(bdir, 'test.csv')
+        @mission.poll_path(path) { |name, io|
+          assert_equal('test.csv', name)
+          assert_equal("data\n", io.read)
+        }
+        assert_equal(true, File.exist?(path))
+        assert_equal(true, File.exist?(bpath))
+        assert_equal("data\n", File.read(bpath))
+      end
+      def test_poll_path__backup__error
+        path = File.join(@dir, 'test.csv')
+        File.open(path, 'w') { |fh| fh.puts 'data' }
+        bdir = @mission.backup_dir = File.expand_path('../backup', @dir)
+        bpath = File.join(bdir, 'test.csv')
+        flexstub(BBMB::Util::Mail).should_receive(:notify_error).times(1)
+        @mission.poll_path(path) { |name, io|
+          assert_equal('test.csv', name)
+          assert_equal("data\n", io.read)
+          raise "some error"
+        }
+        assert_equal(true, File.exist?(path))
+        assert_equal(true, File.exist?(bpath))
+        assert_equal("data\n", File.read(bpath))
+      end
+    end
+    class TestFtpMission < Test::Unit::TestCase
+      include FlexMock::TestCase
+      def setup
+        @datadir = File.expand_path('../data', File.dirname(__FILE__))
+        BBMB.config = flexmock('config')
+        BBMB.config.should_receive(:bbmb_dir).and_return(@datadir)
+        @mission = FtpMission.new
+        @mission.directory = "ftp://user:pass@ftp.server.com/path/to/dir"
+      end
+      def test_poll
+        session = flexmock 'ftp'
+        session.should_receive(:login).with('user', 'pass').times(2)
+        session.should_receive(:chdir).with('/path/to/dir').times(2)
+        session.should_receive(:nlst).and_return %w{test.csv test.txt}
+        session.should_receive(:get).and_return { |remote, local|
+          assert_equal('test.txt', remote)
+          File.open(local, 'w') { |fh| fh.puts "data" }
+        }
+        flexmock(Net::FTP).should_receive(:open).and_return { |host, block| 
+          assert_equal('ftp.server.com', host)
+          block.call session
+        }
+        @mission.pattern = '.*\.txt'
+        @mission.poll { |name, io|
+          assert_equal('test.txt', name)
+          assert_equal("data\n", io.read)
+        }
+
+        @mission.pattern = '.*\.xls'
+        @mission.poll { |name, io|
+          flunk "pattern .*\.xls should not match any files, matched #{name}"
+        }
+      end
+      def test_poll_remote
+        @mission.backup_dir = '/backup/dir'
+        session = flexmock 'ftp'
+        session.should_receive(:get).and_return { |remote, local|
+          assert_equal('test.txt', remote)
+          assert_equal('/backup/dir/test.txt', local)
+        }
+        assert_equal('/backup/dir/test.txt', 
+                     @mission.poll_remote(session, 'test.txt'))
+      end
+      def test_poll_remote__delete
+        @mission.backup_dir = '/backup/dir'
+        @mission.delete = true
+        session = flexmock 'ftp'
+        session.should_receive(:get).and_return { |remote, local|
+          assert_equal('test.txt', remote)
+          assert_equal('/backup/dir/test.txt', local)
+        }
+        session.should_receive(:delete).with('test.txt').times(1)
+        assert_equal('/backup/dir/test.txt', 
+                     @mission.poll_remote(session, 'test.txt'))
+      end
+    end
+    class TestPopMission < Test::Unit::TestCase
+      include FlexMock::TestCase
+      def setup
+        @mission = PopMission.new
+        @mission.host = "mail.ywesee.com"
+        @mission.user = "data@bbmb.ch"
+        @mission.pass = "test"
+      end
+      def test_poll_message__normal
+        message = RMail::Message.new
+        part1 = RMail::Message.new
+        part1.body = "inline text"
+        message.add_part(part1)
+        part2 = RMail::Message.new
+        part2.body = "attached data"
+        part2.header.add("Content-Type",'TEXT/plain', nil,
+                         'NAME' => "=?ISO-8859-1?Q?ywsarti.csv?=")
+        message.add_part(part2)
+        blk_called = false
+        @mission.poll_message(message) { |filename, data|
+          assert_equal('ywsarti.csv', filename) 
+          assert_equal('attached data', data) 
+          blk_called = true
+        }
+        assert(blk_called, "poll_message never called its block")
+      end
+      def test_poll_message__many_parameters
+        message = RMail::Message.new
+        part1 = RMail::Message.new
+        part1.body = "inline text"
+        message.add_part(part1)
+        part2 = RMail::Message.new
+        part2.body = "attached data"
+        part2.header.add("Content-Type",'TEXT/plain', nil,
+                         [['NAME', "ywsarti.csv"], ["foo", "b.r"]])
+        message.add_part(part2)
+        blk_called = false
+        @mission.poll_message(message) { |filename, data|
+          assert_equal('ywsarti.csv', filename) 
+          assert_equal('attached data', data) 
+          blk_called = true
+        }
+        assert(blk_called, "poll_message never called its block")
+      end
+      def test_poll_message__no_quotes
+        message = RMail::Message.new
+        part1 = RMail::Message.new
+        part1.body = "inline text"
+        message.add_part(part1)
+        part2 = RMail::Message.new
+        part2.body = "attached data"
+        part2.header.add("Content-Type",'text/plain', nil,
+                         [['filename', "ywsarti"], ["foo", "bar"]])
+        message.add_part(part2)
+        blk_called = false
+        @mission.poll_message(message) { |filename, data|
+          assert_equal('ywsarti', filename) 
+          assert_equal('attached data', data) 
+          blk_called = true
+        }
+        assert(blk_called, "poll_message never called its block")
+      end
+      def test_poll
+        src = <<-EOS
+Content-Type: multipart/mixed; boundary="=-1158308026-727155-3822-1761-1-="
+MIME-Version: 1.0
+
+
+--=-1158308026-727155-3822-1761-1-=
+
+inline text
+--=-1158308026-727155-3822-1761-1-=
+Content-Disposition: attachment; filename="ywsarti.csv"
+
+attached data
+--=-1158308026-727155-3822-1761-1-=--
+        EOS
+        flexstub(Net::POP3).should_receive(:start).with('mail.ywesee.com', 110, 
+          'data@bbmb.ch', 
+          'test', Proc).and_return { |host, port, user, pass, block|
+          pop = flexmock('pop')
+          pop.should_receive(:each_mail).and_return { |block2|
+            mail = flexmock('mail')
+            mail.should_receive(:pop).and_return(src)
+            mail.should_receive(:delete)
+            block2.call(mail)
+          }
+          block.call(pop)
+        }
+        @mission.poll { |name, data|
+          assert_equal('ywsarti.csv', name)
+          assert_equal('attached data', data)
+        }
+      end
+      def test_poll__error
+        src = <<-EOS
+Content-Type: multipart/mixed; boundary="=-1158308026-727155-3822-1761-1-="
+MIME-Version: 1.0
+
+
+--=-1158308026-727155-3822-1761-1-=
+
+inline text
+--=-1158308026-727155-3822-1761-1-=
+Content-Type: text/csv; filename="ywsarti.csv"
+
+attached data
+--=-1158308026-727155-3822-1761-1-=--
+        EOS
+        flexstub(Net::POP3).should_receive(:start).with('mail.ywesee.com', 110, 
+          'data@bbmb.ch', 
+          'test', Proc).and_return { |host, port, user, pass, block|
+          pop = flexmock('pop')
+          pop.should_receive(:each_mail).and_return { |block2|
+            mail = flexmock('mail')
+            mail.should_receive(:pop).and_return(src)
+            mail.should_receive(:delete)
+            block2.call(mail)
+          }
+          block.call(pop)
+        }
+        flexstub(BBMB::Util::Mail).should_receive(:notify_error)\
+          .times(1).and_return { assert true }
+        @mission.poll { |name, data|
+          raise "some error"
+        }
+      end
+    end
+    class TestPollingManager < Test::Unit::TestCase
+      include FlexMock::TestCase
+      def setup
+        @manager = PollingManager.new
+        @datadir = File.expand_path('data', File.dirname(__FILE__))
+      end
+      def test_load_sources
+        FileUtils.mkdir_p(@datadir)
+        path = File.join(@datadir, 'polling.yml')
+        mission = PopMission.new
+        mission.host = "mail.ywesee.com"
+        mission.user = "data@bbmb.ch"
+        mission.pass = "test"
+        File.open(path, 'w') { |fh| fh.puts mission.to_yaml }
+        config = flexmock("config")
+        config.should_receive(:polling_file).and_return(path)
+        BBMB.config = config
+        @manager.load_sources { |mission|
+          assert_equal("mail.ywesee.com", mission.host)
+          assert_equal("data@bbmb.ch", mission.user)
+          assert_equal("test", mission.pass)
+        }
+      ensure
+        FileUtils.rm_r(@datadir) if(File.exist?(@datadir))
+      end
+      def test_poll_sources
+        flexstub(@manager).should_receive(:load_sources).and_return { |block|
+          source = flexmock("source")
+          source.should_receive(:poll)
+          block.call(source)
+          assert(true)
+        }
+        @manager.poll_sources
+      end
+    end
+  end
+end

data/test/util/test_server.rb

@@ -0,0 +1,189 @@
+#!/usr/bin/env ruby
+# Util::TestServer -- bbmb.ch -- 22.09.2006 -- hwyss@ywesee.com
+
+$: << File.expand_path('../../lib', File.dirname(__FILE__))
+$: << File.expand_path('..', File.dirname(__FILE__))
+
+require 'test/unit'
+require 'bbmb'
+require 'bbmb/util/server'
+require 'stub/persistence'
+require 'flexmock'
+
+module BBMB
+  module Util
+class TestServer < Test::Unit::TestCase
+  include FlexMock::TestCase
+  def setup
+    @server = Server.new
+    Model::Customer.instances.clear
+    Model::Product.instances.clear
+  end
+  def test_inject_order__unknown_customer
+    assert_raises(RuntimeError) {
+      @server.inject_order('12345', [], {})
+    }
+  end
+  def test_inject_order
+    pr1 = Model::Product.new 1
+    pr1.pcode = '1234567'
+    pr2 = Model::Product.new 2
+    pr2.ean13 = '1234567890123'
+    pr3 = Model::Product.new 3
+    pr3.ean13 = '2345678901234'
+    pr3.pcode = '2345678'
+    customer = Model::Customer.new('12345')
+    flexmock(customer).should_receive(:inject_order).times(1).and_return { |order|
+      assert_instance_of(Model::Order, order)
+      ps1, ps2, ps3 = order.positions
+      assert_instance_of(Model::Order::Position, ps1)
+      assert_instance_of(Model::Order::Position, ps2)
+      assert_instance_of(Model::Order::Position, ps3)
+      assert_equal(3, ps1.quantity)
+      assert_equal(pr1, ps1.product)
+      assert_equal(4, ps2.quantity)
+      assert_equal(pr2, ps2.product)
+      assert_equal(5, ps3.quantity)
+      assert_equal(pr3, ps3.product)
+      assert_equal('My Comment', order.comment)
+      assert_equal('76543', order.reference)
+    }
+    prods = [
+      {:quantity => 3, :pcode => '1234567'},
+      {:quantity => 4, :ean13 => '1234567890123'},
+      {:quantity => 5, :pcode => '2345678', :ean13 => '2345678901234'},
+    ]
+    infos = {
+      :comment => 'My Comment',
+      :reference => '76543',
+    }
+    assert_nothing_raised {
+      @server.inject_order('12345', prods, infos)
+    }
+  end
+  def test_inject_order__customer_by_ean13
+    pr1 = Model::Product.new 1
+    pr1.pcode = '1234567'
+    pr2 = Model::Product.new 2
+    pr2.ean13 = '1234567890123'
+    pr3 = Model::Product.new 3
+    pr3.ean13 = '2345678901234'
+    pr3.pcode = '2345678'
+    customer = Model::Customer.new('12345')
+    customer.ean13 = '1234567890123'
+    flexmock(customer).should_receive(:inject_order).times(1).and_return { |order|
+      assert_instance_of(Model::Order, order)
+      ps1, ps2, ps3 = order.positions
+      assert_instance_of(Model::Order::Position, ps1)
+      assert_instance_of(Model::Order::Position, ps2)
+      assert_instance_of(Model::Order::Position, ps3)
+      assert_equal(3, ps1.quantity)
+      assert_equal(pr1, ps1.product)
+      assert_equal(4, ps2.quantity)
+      assert_equal(pr2, ps2.product)
+      assert_equal(5, ps3.quantity)
+      assert_equal(pr3, ps3.product)
+      assert_equal('My Comment', order.comment)
+      assert_equal('76543', order.reference)
+    }
+    prods = [
+      {:quantity => 3, :pcode => '1234567'},
+      {:quantity => 4, :ean13 => '1234567890123'},
+      {:quantity => 5, :pcode => '2345678', :ean13 => '2345678901234'},
+    ]
+    infos = {
+      :comment => 'My Comment',
+      :reference => '76543',
+    }
+    flexmock(BBMB::Util::Mail).should_receive(:send_order)\
+      .with(BBMB::Model::Order).times(1)
+    flexmock(BBMB::Util::TargetDir).should_receive(:send_order)\
+      .with(BBMB::Model::Order).times(1)
+    assert_nothing_raised {
+      @server.inject_order('1234567890123', prods, infos, :deliver => true)
+    }
+  end
+  def test_rename_user__new
+    BBMB.config = flexmock('config')
+    BBMB.config.should_receive(:auth_domain).times(1).and_return('ch.bbmb')
+    BBMB.auth = flexmock('auth')
+    session = flexmock('yus-session')
+    BBMB.auth.should_receive(:autosession).times(1).and_return { |domain, block|
+      assert_equal('ch.bbmb', domain)
+      block.call(session)
+    }
+    session.should_receive(:create_entity).times(1).and_return { |email|
+      assert_equal('test@bbmb.ch', email)
+    }
+    @server.rename_user(nil, 'test@bbmb.ch')
+  end
+  def test_rename_user__existing
+    BBMB.config = flexmock('config')
+    BBMB.config.should_receive(:auth_domain).times(1).and_return('ch.bbmb')
+    BBMB.auth = flexmock('auth')
+    session = flexmock('yus-session')
+    BBMB.auth.should_receive(:autosession).times(1).and_return { |domain, block|
+      assert_equal('ch.bbmb', domain)
+      block.call(session)
+    }
+    session.should_receive(:rename).times(1).and_return { |previous, email|
+      assert_equal('old@bbmb.ch', previous)
+      assert_equal('test@bbmb.ch', email)
+    }
+    @server.rename_user('old@bbmb.ch', 'test@bbmb.ch')
+  end
+  def test_rename_user__same
+    assert_nothing_raised { 
+      @server.rename_user('test@bbmb.ch', 'test@bbmb.ch')
+    }
+  end
+  def test_run_invoicer
+    BBMB.logger = flexmock('logger')
+    BBMB.logger.should_ignore_missing
+    flexstub(Mail).should_receive(:notify_error).times(1).and_return { |error|
+      assert_instance_of(RuntimeError, error)
+    }
+    flexstub(Invoicer).should_receive(:run).times(1).and_return { |range|
+      assert_instance_of(Range, range)
+      raise "notify an error!"
+    }
+    invoicer = @server.run_invoicer
+    Timeout.timeout(5) { 
+      until(invoicer.status == 'sleep')
+        sleep 0.1
+      end
+    }
+    invoicer.wakeup
+    assert_equal('run', invoicer.status)
+    until(invoicer.status == 'sleep')
+      sleep 0.1
+    end
+    invoicer.exit
+  end
+  def test_run_updater
+    BBMB.config = flexmock('config')
+    BBMB.config.should_receive(:update_hour).and_return(0)
+    BBMB.logger = flexmock('logger')
+    BBMB.logger.should_ignore_missing
+    flexstub(Mail).should_receive(:notify_error).times(1).and_return { |error|
+      assert_instance_of(RuntimeError, error)
+    }
+    flexstub(Updater).should_receive(:run).times(1).and_return {
+      raise "notify an error!"
+    }
+    updater = @server.run_updater
+    Timeout.timeout(5) { 
+      until(updater.status == 'sleep')
+        sleep 0.1
+      end
+    }
+    updater.wakeup
+    assert_equal('run', updater.status)
+    until(updater.status == 'sleep')
+      sleep 0.1
+    end
+    updater.exit
+  end
+end
+  end
+end