segregate 0.1.0 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 8ebb52ffc2a9364921143e72be4e33c036fe241b
-  data.tar.gz: 4dd0ee7bd05e7718fb57918f5ae416783a2266d5
+  metadata.gz: 7421d31435304199cb1032dbe326420d13a75a69
+  data.tar.gz: eea29b9926c873d0c66a2bbf6a9106aa847b2e5c
 SHA512:
-  metadata.gz: 92bf21b2a568e476884189262dd195ab31487d4e3451104f492b4d0e8a8b62dc9897e85a718c812086c85edb5cf2d9574a4ca05b973407ad45c5665395d6b6e6
-  data.tar.gz: 65eb168efe658013e9dfd428814792d3a4369c94c14b3ea52ae03f965a4492ec354381f505b61880fc6343f3411a7dc95c13e91cc85e19e7dbee04f489065f04
+  metadata.gz: 6d52b0c763d8705420c29d53722c70b74f84549fb5fafc426a4fb303b6057cd39b71b96904e413485027d226cf8d93840e0bde29469a8699efe609684576eb41
+  data.tar.gz: bdc263d1000ecaca777f3a808fb4c3e0ee6cbd3d93d87f3f2a0692611ba4a2e5312d2faf7336b3a7c5ff9ff0a092199602c54085f9573294b9661cc7b0cbac95

data/lib/segregate/version.rb

@@ -1,4 +1,4 @@
 class Segregate
-	VERSION = "0.1.0".freeze
+	VERSION = "0.2.0".freeze
   DATE = "2014-02-03".freeze
 end

data/lib/segregate.rb

@@ -18,7 +18,9 @@ class Segregate
     @uri.respond_to?(meth, include_private) || super
   end
 
-  def initialize
+  def initialize callback = nil
+    @callback = callback
+
     @uri = nil
     @request_method = nil
     @status_code = nil
@@ -94,6 +96,7 @@ class Segregate
   end
 
   def read_first_line data
+    @callback.on_message_begin self if @callback.respond_to?(:on_message_begin)
     line = read data
 
     if line =~ REQUEST_LINE
@@ -114,6 +117,8 @@ class Segregate
     @request_method, url, @http_version[0], @http_version[1] = line.scan(REQUEST_LINE).flatten
     @http_version.map! {|v| v.to_i}
     @uri = URI.parse url
+
+    @callback.on_request_line self if @callback.respond_to?(:on_request_line)
   end
 
   def parse_status_line line
@@ -121,6 +126,8 @@ class Segregate
     @http_version[0], @http_version[1], code, @status_phrase = line.scan(STATUS_LINE).flatten
     @http_version.map! {|v| v.to_i}
     @status_code = code.to_i
+
+    @callback.on_status_line self if @callback.respond_to?(:on_status_line)
   end
 
   def read_headers data
@@ -134,6 +141,8 @@ class Segregate
         @header_order << key.downcase
       end
     end
+
+    @callback.on_headers_complete self if @callback.respond_to?(:on_headers_complete) && @headers_complete
   end
 
   def read_body data
@@ -142,6 +151,8 @@ class Segregate
     elsif headers['content-encoding'] == 'chunked'
       parse_chunked_data data
     end
+
+    @callback.on_body_complete self if @callback.respond_to?(:on_body_complete) && @body_complete
   end
 
   def parse_body data

data/spec/segregate_spec.rb

@@ -396,4 +396,49 @@ describe Segregate do
       end
     end
   end
+
+  context 'a new parser has been created with a callback object' do
+    before(:each) do
+      @callback_object = double
+      @parser = Segregate.new @callback_object
+    end
+
+    describe 'on_message_begin' do
+      it 'calls the callback object' do
+        @callback_object.should_receive(:on_message_begin).with(@parser)
+        @parser.parse "GET /endpoint HTTP/1.1\r\n"
+      end
+    end
+
+    describe 'on_request' do
+      it 'calls the callback object' do
+        @callback_object.should_receive(:on_request_line).with(@parser)
+        @parser.parse "GET /endpoint HTTP/1.1\r\n"
+      end
+    end
+
+    describe 'on_response' do
+      it 'calls the callback object' do
+        @callback_object.should_receive(:on_status_line).with(@parser)
+        @parser.parse "HTTP/1.1 200 OK\r\n"
+      end
+    end
+
+    describe 'on_headers_complete' do
+      it 'calls the callback object' do
+        @callback_object.should_receive(:on_headers_complete).with(@parser)
+        @parser.parse "GET /endpoint HTTP/1.1\r\n"
+        @parser.parse "Host: www.google.com\r\n\r\n"
+      end
+    end
+
+    describe 'on_body_complete' do
+      it 'calls the callback object' do
+        @callback_object.should_receive(:on_body_complete).with(@parser)
+        @parser.parse "GET /endpoint HTTP/1.1\r\n"
+        @parser.parse "Content-Length: 8\r\n\r\n"
+        @parser.parse "TestData\r\n\r\n"
+      end
+    end
+  end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: segregate
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.0
 platform: ruby
 authors:
 - Ben Slaughter