loggregator_emitter 0.0.4.pre → 0.0.5.pre

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    loggregator_emitter (0.0.4.pre)
+    loggregator_emitter (0.0.5.pre)
       loggregator_messages (~> 0.0.1.pre)
 
 GEM

data/lib/loggregator_emitter/emit.rb

@@ -13,14 +13,29 @@ module LoggregatorEmitter
     end
 
     def emit(app_id, message)
-      s = UDPSocket.new
-      s.do_not_reverse_lookup = true
+      lm = create_log_message(app_id, message, LogMessage::MessageType::OUT)
+      send_message(lm)
+    end
+
+    def emit_error(app_id, message)
+      lm = create_log_message(app_id, message, LogMessage::MessageType::ERR)
+      send_message(lm)
+    end
+
+    private
+    def create_log_message(app_id, message, type)
       lm = LogMessage.new()
       lm.timestamp = Time.now.to_i
       lm.message = message
       lm.app_id = app_id
       lm.source_type = @source_type
-      lm.message_type = LogMessage::MessageType::OUT
+      lm.message_type = type
+      lm
+    end
+
+    def send_message(lm)
+      s = UDPSocket.new
+      s.do_not_reverse_lookup = true
 
       result = lm.encode.buf
       result.unpack("C*")
@@ -28,7 +43,6 @@ module LoggregatorEmitter
       s.sendmsg_nonblock(result, 0, @sockaddr_in)
     end
 
-    private
     def valid_source_type?(source_type)
       LogMessage::SourceType.constants.each do |name|
         if LogMessage::SourceType.const_get(name) == source_type

data/loggregator_emitter.gemspec

@@ -2,7 +2,7 @@
 
 Gem::Specification.new do |spec|
   spec.name          = "loggregator_emitter"
-  spec.version       = '0.0.4.pre'
+  spec.version       = '0.0.5.pre'
   spec.authors       = ["Pivotal"]
   spec.email         = ["cf-eng@pivotallabs.com"]
   spec.description   = "Library to emit data to Loggregator"

data/spec/loggregator_emitter/emit_spec.rb

@@ -22,7 +22,7 @@ describe LoggregatorEmitter do
     end
   end
 
-  describe 'Writing to Sockets' do
+  describe 'Sending To STDOUT' do
     before(:each) do
       @emitter = LoggregatorEmitter::Emitter.new('0.0.0.0:12345', LogMessage::SourceType::CLOUD_CONTROLLER)
     end
@@ -43,6 +43,36 @@ describe LoggregatorEmitter do
       expect(message.message).to eq 'Hello there!'
       expect(message.app_id).to eq "my_app_id"
       expect(message.source_type).to eq LogMessage::SourceType::CLOUD_CONTROLLER
+      expect(message.message_type).to eq LogMessage::MessageType::OUT
+
+
+      message = messages[1]
+      expect(message.message).to eq 'Hello again!'
+    end
+  end
+
+  describe 'Sending To STDOUT' do
+    before(:each) do
+      @emitter = LoggregatorEmitter::Emitter.new('0.0.0.0:12345', LogMessage::SourceType::CLOUD_CONTROLLER)
+    end
+
+    it 'successfully writes protobuffer to a socket' do
+      server = FakeLoggregatorServer.new(12345)
+      server.start
+
+      @emitter.emit_error("my_app_id", 'Hello there!')
+      @emitter.emit_error("my_app_id", 'Hello again!')
+
+      server.stop(2)
+
+      messages = server.messages
+
+      expect(messages.length).to eq 2
+      message = messages[0]
+      expect(message.message).to eq 'Hello there!'
+      expect(message.app_id).to eq "my_app_id"
+      expect(message.source_type).to eq LogMessage::SourceType::CLOUD_CONTROLLER
+      expect(message.message_type).to eq LogMessage::MessageType::ERR
 
       message = messages[1]
       expect(message.message).to eq 'Hello again!'

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: loggregator_emitter
 version: !ruby/object:Gem::Version
-  version: 0.0.4.pre
+  version: 0.0.5.pre
   prerelease: 6
 platform: ruby
 authors: