middle_squid 1.0

data/test/test_builder.rb

@@ -0,0 +1,89 @@
+require File.expand_path '../helper', __FILE__
+
+class TestBuilder < MiniTest::Test
+  class FakeAdapter < MiddleSquid::Adapter
+    def say_hello
+      @options[:hello]
+    end
+  end
+
+  def setup
+    @obj = MiddleSquid::Builder.new
+  end
+
+  def test_from_file
+    path = File.expand_path '../resources', __FILE__
+    file = path + '/hello.rb'
+
+    stdout, stderr = capture_io do
+       @obj = MiddleSquid::Builder.from_file file
+    end
+
+    assert_equal "hello #{@obj}\n", stdout
+  end
+
+  def test_default_adapter
+    assert_equal MiddleSquid::Adapters::Squid, @obj.adapter.class
+    assert_same @obj.adapter, @obj.adapter
+  end
+
+  def test_custom_adapter
+    @obj.use FakeAdapter, hello: 'world'
+    assert_equal FakeAdapter, @obj.adapter.class
+    assert_equal 'world', @obj.adapter.say_hello
+  end
+
+  def test_database
+    @obj.database ':memory:'
+  end
+
+  def test_blacklist
+    assert_empty @obj.blacklists
+
+    bl = @obj.blacklist 'hello'
+    assert_instance_of MiddleSquid::BlackList, bl
+    assert_equal 'hello', bl.category
+
+    assert_equal [bl], @obj.blacklists
+  end
+
+  def test_define_action
+    assert_empty @obj.custom_actions
+
+    world = proc { :world }
+    @obj.define_action :hello, &world
+
+    assert_equal({:hello => world}, @obj.custom_actions)
+  end
+
+  def test_define_helper
+    assert_equal @obj.method(:define_helper), @obj.method(:define_action)
+  end
+
+  def test_define_action_noblock
+    error = assert_raises ArgumentError do
+      @obj.define_action :hello
+    end
+
+    assert_equal 'no block given', error.message
+    assert_empty @obj.custom_actions
+  end
+
+  def test_handler
+    assert_nil @obj.handler
+
+    handler = lambda {}
+    @obj.run handler
+
+    assert_equal handler, @obj.handler
+  end
+
+  def test_handler_nocall
+    error = assert_raises ArgumentError do
+      @obj.run 1
+    end
+
+    assert_equal 'the handler must respond to #call', error.message
+    assert_nil @obj.handler
+  end
+end

data/test/test_cli.rb

@@ -0,0 +1,105 @@
+require File.expand_path '../helper', __FILE__
+
+class TestCLI < MiniTest::Test
+  def test_start
+    path = File.expand_path '../resources', __FILE__
+    conf = File.join path, 'hello.rb'
+
+    stdout, stderr = capture_io do
+      EM.run {
+        MiddleSquid::CLI.start(%W[start -C #{conf}])
+        EM.next_tick { EM.stop }
+      }
+    end
+
+    assert_match /\Ahello #<MiddleSquid:.+>\Z/, stdout
+  end
+
+  def test_start_relative
+    absolute = File.expand_path '../resources', __FILE__
+    path = Pathname.new(absolute).relative_path_from(Pathname.new(Dir.home))
+
+    conf = File.join '~', path, 'hello.rb'
+
+    capture_io do
+      EM.run {
+        MiddleSquid::CLI.start(%W[start -C #{conf}])
+        EM.next_tick { EM.stop }
+      }
+    end
+  end
+
+  def test_default_config
+    default = File.join Dir.home, 'middle_squid.rb'
+
+    error = assert_raises Errno::ENOENT do
+      MiddleSquid::CLI.start %W[start]
+    end
+
+    assert_equal "No such file or directory @ rb_sysopen - #{default}", error.message
+  end
+
+  def test_index
+    path = File.expand_path '../resources', __FILE__
+    conf = File.join path, 'hello.rb'
+    list = File.join path, 'black'
+
+    stdout, stderr = capture_io do
+      MiddleSquid::CLI.start(%W[index #{list} -C #{conf} --full])
+    end
+
+    assert_match /\Ahello #<MiddleSquid:.+>$/, stdout
+    assert_match "reading #{list}", stdout
+  end
+
+  def test_index_relative_path
+    absolute = File.expand_path '../resources', __FILE__
+    path = Pathname.new(absolute).relative_path_from(Pathname.new(Dir.home))
+
+    conf = File.join '~', path, 'hello.rb'
+    list = File.join '~', path, 'black'
+
+    stdout, stderr = capture_io do
+      MiddleSquid::CLI.start(%W[index #{list} -C #{conf} --full])
+    end
+
+    assert_match /\Ahello #<MiddleSquid:.+>$/, stdout
+    assert_match "reading #{absolute}/black", stdout
+  end
+
+  def test_index_multiple
+    path = File.expand_path '../resources', __FILE__
+    config = File.join path, 'hello.rb'
+    list_1 = File.join path, 'black'
+    list_2 = File.join path, 'gray'
+
+    stdout, stderr = capture_io do
+      MiddleSquid::CLI.start(%W[index #{list_1} #{list_2} -C #{config} --full])
+    end
+
+    assert_match "reading #{list_1}", stdout
+    assert_match "reading #{list_2}", stdout
+  end
+
+  def test_version
+    stdout, stderr = capture_io do
+      MiddleSquid::CLI.start %w[version]
+    end
+
+    assert_match /MiddleSquid #{MiddleSquid::VERSION}/, stdout
+    assert_match /Copyright/, stdout
+    assert_match /GNU General Public License/, stdout
+    assert_match /version 3/, stdout
+    assert_empty stderr
+  end
+
+  def test_help
+    stdout, stderr = capture_io do
+      MiddleSquid::CLI.start %w[--help]
+    end
+
+    assert_match /MiddleSquid commands:/, stdout
+    assert_match /Options:/, stdout
+    assert_empty stderr
+  end
+end

data/test/test_database.rb

@@ -0,0 +1,20 @@
+require File.expand_path '../helper', __FILE__
+
+class TestDatabase < MiniTest::Test
+  include MiddleSquid::Database
+
+  def test_db
+    assert_instance_of SQLite3::Database, db()
+  end
+
+  def test_setup
+    before = db()
+    refute before.closed?
+
+    MiddleSquid::Database.setup ':memory:'
+
+    after = db()
+    refute_same before, after
+    assert before.closed?, 'the old database should be closed'
+  end
+end

data/test/test_hash.rb

@@ -0,0 +1,28 @@
+require File.expand_path '../helper', __FILE__
+
+class TestHash < MiniTest::Test
+  def test_sanitize_headers
+    before = {
+      'Chunky-Bacon' => nil,
+      'CONNECTION' => nil,
+      'CONTENT_ENCODING' => nil,
+      'CONTENT_LENGTH' => nil,
+      'HELLO_WORLD' => nil,
+      'HOST' => nil,
+      'TEST' => nil,
+      'Transfer-Encoding' => nil,
+      'TRANSFER_ENCODING' => nil,
+      'VERSION' => nil,
+    }
+
+    after = {
+      'Chunky-Bacon' => nil,
+      'Hello-World' => nil,
+      'Test' => nil,
+    }
+
+    before.sanitize_headers!
+
+    assert_equal after, before
+  end
+end

data/test/test_helper.rb

@@ -0,0 +1,76 @@
+require File.expand_path '../helper', __FILE__
+
+class TestHelpers < MiniTest::Test
+  FakeClass = Class.new { include MiddleSquid::Helpers }
+
+  def setup
+    @obj = FakeClass.new
+  end
+
+  def download_wrapper(uri, env)
+    bag = []
+    req = Rack::Request.new env
+
+    EM.run {
+      Fiber.new {
+        bag << @obj.download_like(req, uri)
+      }.resume
+      EM.next_tick { EM.stop }
+    }
+
+    assert_equal 1, bag.size
+    bag[0]
+  end
+
+  def test_download
+    uri = MiddleSquid::URI.parse 'http://test.com/path?query=string'
+
+    stub = stub_request(:get, uri).
+      with(:body => 'request%20body', :headers => {'User-Agent'=>'Mozilla/5.0', 'Chunky' => 'bacon', 'Content-Type'=>'test/plain'}).
+      to_return(:status => 200, :body => 'response', :headers => {'CHUNKY_BACON' => 'Hello World'})
+
+    status, headers, body = download_wrapper uri,
+      'REQUEST_METHOD' => 'GET',
+      'CONTENT_TYPE' => 'test/plain',
+      'HTTP_CHUNKY' => 'bacon',
+      'HTTP_CONNECTION' => 'ignored',
+      'HTTP_USER_AGENT' => 'Mozilla/5.0',
+      'rack.input' => StringIO.new('request%20body')
+
+    assert_requested stub
+    assert_not_requested :get, uri, :headers => {'Connection' => 'ignored'}
+
+    assert_equal 200, status
+    assert_equal({'Chunky-Bacon' => 'Hello World'}, headers)
+    assert_equal 'response', body
+  end
+
+  def test_download_method
+    uri = MiddleSquid::URI.parse 'http://test.com/'
+
+    stub = stub_request(:post, uri).
+      to_return(:status => 200, :body => '')
+
+    download_wrapper uri,
+      'REQUEST_METHOD' => 'POST',
+      'rack.input' => StringIO.new
+
+    assert_requested stub
+  end
+
+  def test_download_error
+    uri = MiddleSquid::URI.parse 'http://test.com/'
+
+    stub = stub_request(:get, uri).to_timeout
+
+    status, headers, body = download_wrapper uri,
+      'REQUEST_METHOD' => 'GET',
+      'rack.input' => StringIO.new
+
+    assert_requested stub
+
+    assert_equal 520, status
+    assert_empty headers
+    assert_equal '[MiddleSquid] WebMock timeout error', body
+  end
+end