application_insights 0.5.3 → 0.5.4

data/test/application_insights/channel/test_queue_base.rb

@@ -1,89 +1,89 @@
-require_relative '../../../lib/application_insights/channel/queue_base'
-require_relative '../../../lib/application_insights/channel/sender_base'
-require 'test/unit'
-
-include ApplicationInsights::Channel
-
-class TestQueueBase < Test::Unit::TestCase
-  def test_initialize
-    sender = SenderBase.new 'http://tempuri.org'
-    item = QueueBase.new sender
-    assert_equal sender.queue, item
-    assert_equal 500, item.max_queue_length
-    assert_equal sender, item.sender
-  end
-
-  def test_max_queue_length_works_as_expected
-    sender = SenderBase.new 'http://tempuri.org'
-    item = QueueBase.new sender
-    assert_equal 500, item.max_queue_length
-    item.max_queue_length = 42
-    assert_equal 42, item.max_queue_length
-  end
-
-  def test_sender_works_as_expected
-    sender = SenderBase.new 'http://tempuri.org'
-    item = QueueBase.new sender
-    assert_equal sender, item.sender
-  end
-
-  def test_push_will_not_enqueue_nil
-    sender = SenderBase.new 'http://tempuri.org'
-    item = QueueBase.new sender
-    queue = item.instance_variable_get('@queue')
-    item.push nil
-    assert queue.empty?
-  end
-
-  def test_push_works_as_expected
-    sender = SenderBase.new 'http://tempuri.org'
-    item = MockQueueBase.new sender
-    item.max_queue_length = 2
-    queue = item.instance_variable_get('@queue')
-    [[7, 0], [8, 1], [9, 2]].each do |i, count|
-      item.push i
-      assert_equal count, item.flush_count
-    end
-    assert_equal 3, queue.length
-    temp = []
-    queue.length.times do |i|
-      temp.push queue.pop
-    end
-    assert_equal [7, 8, 9], temp
-  end
-
-  def test_pop_works_as_expected
-    sender = SenderBase.new 'http://tempuri.org'
-    item = QueueBase.new sender
-    queue = item.instance_variable_get('@queue')
-    [7, 8, 9].each do |i|
-      queue.push i
-    end
-    [7, 8, 9, nil].each do |i|
-      assert_equal i, item.pop
-    end
-  end
-
-  def test_empty_works_as_expected
-    sender = SenderBase.new 'http://tempuri.org'
-    queue = QueueBase.new sender
-    assert_equal true, queue.empty?
-    queue.push 1
-    assert_equal false, queue.empty?
-    queue.pop
-    assert_equal true, queue.empty?
-  end
-end
-
-class MockQueueBase < QueueBase
-  def initialize(sender)
-    super sender
-    @flush_count = 0
-  end
-
-  attr_accessor :flush_count
-
-  def flush
-    @flush_count += 1
-  end
+require_relative '../../../lib/application_insights/channel/queue_base'
+require_relative '../../../lib/application_insights/channel/sender_base'
+require 'test/unit'
+
+include ApplicationInsights::Channel
+
+class TestQueueBase < Test::Unit::TestCase
+  def test_initialize
+    sender = SenderBase.new 'http://tempuri.org'
+    item = QueueBase.new sender
+    assert_equal sender.queue, item
+    assert_equal 500, item.max_queue_length
+    assert_equal sender, item.sender
+  end
+
+  def test_max_queue_length_works_as_expected
+    sender = SenderBase.new 'http://tempuri.org'
+    item = QueueBase.new sender
+    assert_equal 500, item.max_queue_length
+    item.max_queue_length = 42
+    assert_equal 42, item.max_queue_length
+  end
+
+  def test_sender_works_as_expected
+    sender = SenderBase.new 'http://tempuri.org'
+    item = QueueBase.new sender
+    assert_equal sender, item.sender
+  end
+
+  def test_push_will_not_enqueue_nil
+    sender = SenderBase.new 'http://tempuri.org'
+    item = QueueBase.new sender
+    queue = item.instance_variable_get('@queue')
+    item.push nil
+    assert queue.empty?
+  end
+
+  def test_push_works_as_expected
+    sender = SenderBase.new 'http://tempuri.org'
+    item = MockQueueBase.new sender
+    item.max_queue_length = 2
+    queue = item.instance_variable_get('@queue')
+    [[7, 0], [8, 1], [9, 2]].each do |i, count|
+      item.push i
+      assert_equal count, item.flush_count
+    end
+    assert_equal 3, queue.length
+    temp = []
+    queue.length.times do |i|
+      temp.push queue.pop
+    end
+    assert_equal [7, 8, 9], temp
+  end
+
+  def test_pop_works_as_expected
+    sender = SenderBase.new 'http://tempuri.org'
+    item = QueueBase.new sender
+    queue = item.instance_variable_get('@queue')
+    [7, 8, 9].each do |i|
+      queue.push i
+    end
+    [7, 8, 9, nil].each do |i|
+      assert_equal i, item.pop
+    end
+  end
+
+  def test_empty_works_as_expected
+    sender = SenderBase.new 'http://tempuri.org'
+    queue = QueueBase.new sender
+    assert_equal true, queue.empty?
+    queue.push 1
+    assert_equal false, queue.empty?
+    queue.pop
+    assert_equal true, queue.empty?
+  end
+end
+
+class MockQueueBase < QueueBase
+  def initialize(sender)
+    super sender
+    @flush_count = 0
+  end
+
+  attr_accessor :flush_count
+
+  def flush
+    @flush_count += 1
+  end
 end

data/test/application_insights/channel/test_sender_base.rb

@@ -1,87 +1,87 @@
-require_relative '../../../lib/application_insights/channel/queue_base'
-require_relative '../../../lib/application_insights/channel/sender_base'
-require 'socket'
-require 'test/unit'
-require 'thread'
-
-include ApplicationInsights::Channel
-
-class TestSenderBase < Test::Unit::TestCase
-  def test_initialize
-    sender = SenderBase.new 'http://tempuri.org'
-    assert_equal 'http://tempuri.org', sender.service_endpoint_uri
-    assert_nil sender.queue
-    assert_equal 100, sender.send_buffer_size
-  end
-
-  def test_service_endpoint_uri_works_as_expected
-    sender = SenderBase.new 'http://tempuri.org'
-    assert_equal 'http://tempuri.org', item.service_endpoint_uri
-    item.service_endpoint_uri = 'http://live.com'
-    assert_equal 'http://live.com', item.service_endpoint_uri
-  end
-
-  def test_service_endpoint_uri_works_as_expected
-    sender = SenderBase.new 'http://tempuri.org'
-    assert_nil sender.queue
-    temp = Object.new
-    sender.queue = temp
-    assert_equal temp, sender.queue
-  end
-
-  def test_send_buffer_size_works_as_expected
-    sender = SenderBase.new 'http://tempuri.org'
-    assert_equal 100, sender.send_buffer_size
-    sender.send_buffer_size = 42
-    assert_equal 42, sender.send_buffer_size
-  end
-
-  def test_send_works_as_expected_with_400_code
-    thread, port = execute_server '400 BadRequest'
-    sender = SenderBase.new 'http://localhost:' + port.to_s + '/track'
-    sender.queue = []
-    sender.send([1, 2])
-    thread.join
-    assert_equal [], sender.queue
-  end
-
-  def test_send_works_as_expected_with_500_code
-    thread, port = execute_server '500 InternalServerError'
-    sender = SenderBase.new 'http://localhost:' + port.to_s + '/track'
-    sender.queue = []
-    sender.send([1, 2])
-    thread.join
-    assert_equal [], sender.queue
-  end
-
-  def execute_server(code)
-    port = 50000 + Random.rand(10000)
-    thread = Thread.new {
-      server = TCPServer.new(port)
-      client = server.accept
-      request = ''
-      read_buffer_size = 64
-      while TRUE
-        temp = client.recv(read_buffer_size)
-        request += temp
-        break if temp.length < read_buffer_size
-      end
-      request = request.split(/\n/)[-1]
-      response = request
-      headers = [
-          "HTTP/1.1 " + code,
-          "Content-Type: application/json",
-          "Content-Length: #{response.length}\r\n\r\n"
-      ].join("\r\n")
-      client.puts headers
-      client.puts response
-      client.close
-    }
-
-    while thread.status != 'sleep'
-      sleep 0.1
-    end
-
-    return thread, port
-  end
-end
+require_relative '../../../lib/application_insights/channel/queue_base'
+require_relative '../../../lib/application_insights/channel/sender_base'
+require 'socket'
+require 'test/unit'
+require 'thread'
+
+include ApplicationInsights::Channel
+
+class TestSenderBase < Test::Unit::TestCase
+  def test_initialize
+    sender = SenderBase.new 'http://tempuri.org'
+    assert_equal 'http://tempuri.org', sender.service_endpoint_uri
+    assert_nil sender.queue
+    assert_equal 100, sender.send_buffer_size
+  end
+
+  def test_service_endpoint_uri
+    sender = SenderBase.new 'http://tempuri.org'
+    assert_equal 'http://tempuri.org', sender.service_endpoint_uri
+    sender.service_endpoint_uri = 'http://live.com'
+    assert_equal 'http://live.com', sender.service_endpoint_uri
+  end
+
+  def test_sender_queue_attribute
+    sender = SenderBase.new 'http://tempuri.org'
+    assert_nil sender.queue
+    temp = Object.new
+    sender.queue = temp
+    assert_equal temp, sender.queue
+  end
+
+  def test_send_buffer_size_works_as_expected
+    sender = SenderBase.new 'http://tempuri.org'
+    assert_equal 100, sender.send_buffer_size
+    sender.send_buffer_size = 42
+    assert_equal 42, sender.send_buffer_size
+  end
+
+  def test_send_works_as_expected_with_400_code
+    thread, port = execute_server '400 BadRequest'
+    sender = SenderBase.new 'http://localhost:' + port.to_s + '/track'
+    sender.queue = []
+    sender.send([1, 2])
+    thread.join
+    assert_equal [], sender.queue
+  end
+
+  def test_send_works_as_expected_with_500_code
+    thread, port = execute_server '500 InternalServerError'
+    sender = SenderBase.new 'http://localhost:' + port.to_s + '/track'
+    sender.queue = []
+    sender.send([1, 2])
+    thread.join
+    assert_equal [], sender.queue
+  end
+
+  def execute_server(code)
+    port = 50000 + Random.rand(10000)
+    thread = Thread.new {
+      server = TCPServer.new(port)
+      client = server.accept
+      request = ''
+      read_buffer_size = 64
+      while true
+        temp = client.recv(read_buffer_size)
+        request += temp
+        break if temp.length < read_buffer_size
+      end
+      request = request.split(/\n/)[-1]
+      response = request
+      headers = [
+          "HTTP/1.1 " + code,
+          "Content-Type: application/json",
+          "Content-Length: #{response.length}\r\n\r\n"
+      ].join("\r\n")
+      client.puts headers
+      client.puts response
+      client.close
+    }
+
+    while thread.status != 'sleep'
+      sleep 0.1
+    end
+
+    return thread, port
+  end
+end

data/test/application_insights/channel/test_synchronous_queue.rb

@@ -1,28 +1,28 @@
-require_relative '../../../lib/application_insights/channel/sender_base'
-require_relative '../../../lib/application_insights/channel/synchronous_queue'
-require_relative '../mock_sender'
-require 'test/unit'
-
-include ApplicationInsights::Channel
-
-class TestSynchronousQueue < Test::Unit::TestCase
-  def test_initialize
-    SynchronousQueue.new(MockSynchronousSender.new)
-  end
-
-  def test_flush_works_as_expected
-    sender = MockSynchronousSender.new
-    sender.send_buffer_size = 2
-    queue = SynchronousQueue.new sender
-    queue.max_queue_length = 3
-    (1..7).to_a.each do |i|
-      queue.push i
-    end
-    assert_equal [[1, 2], [3], [4, 5], [6]], sender.buffer
-    temp = []
-    queue.instance_variable_get('@queue').length.times do |i|
-      temp.push queue.instance_variable_get('@queue').pop
-    end
-    assert_equal [7], temp
-  end
+require_relative '../../../lib/application_insights/channel/sender_base'
+require_relative '../../../lib/application_insights/channel/synchronous_queue'
+require_relative '../mock_sender'
+require 'test/unit'
+
+include ApplicationInsights::Channel
+
+class TestSynchronousQueue < Test::Unit::TestCase
+  def test_initialize
+    SynchronousQueue.new(MockSynchronousSender.new)
+  end
+
+  def test_flush_works_as_expected
+    sender = MockSynchronousSender.new
+    sender.send_buffer_size = 2
+    queue = SynchronousQueue.new sender
+    queue.max_queue_length = 3
+    (1..7).to_a.each do |i|
+      queue.push i
+    end
+    assert_equal [[1, 2], [3], [4, 5], [6]], sender.buffer
+    temp = []
+    queue.instance_variable_get('@queue').length.times do |i|
+      temp.push queue.instance_variable_get('@queue').pop
+    end
+    assert_equal [7], temp
+  end
 end

data/test/application_insights/channel/test_synchronous_sender.rb

@@ -1,11 +1,11 @@
-require_relative '../../../lib/application_insights/channel/synchronous_sender'
-require 'test/unit'
-
-include ApplicationInsights::Channel
-
-class TestSynchronousSender < Test::Unit::TestCase
-  def test_initialize
-    sender = SynchronousSender.new
-    assert_equal 'https://dc.services.visualstudio.com/v2/track', sender.service_endpoint_uri
-  end
+require_relative '../../../lib/application_insights/channel/synchronous_sender'
+require 'test/unit'
+
+include ApplicationInsights::Channel
+
+class TestSynchronousSender < Test::Unit::TestCase
+  def test_initialize
+    sender = SynchronousSender.new
+    assert_equal 'https://dc.services.visualstudio.com/v2/track', sender.service_endpoint_uri
+  end
 end

data/test/application_insights/channel/test_telemetry_channel.rb

@@ -1,102 +1,126 @@
-require_relative '../../../lib/application_insights/channel/telemetry_channel'
-require_relative '../../../lib/application_insights/channel/telemetry_context'
-require_relative '../../../lib/application_insights/channel/synchronous_queue'
-require_relative '../../../lib/application_insights/channel/synchronous_sender'
-require 'test/unit'
-
-include ApplicationInsights::Channel
-
-class TestTelemetryChannel < Test::Unit::TestCase
-  def test_initialize
-    channel = TelemetryChannel.new
-    assert_not_nil channel.context
-    assert_not_nil channel.queue
-    assert_not_nil channel.sender
-  end
-
-  def test_context_works_as_expected
-    context = TelemetryContext.new
-    channel = TelemetryChannel.new
-    assert_not_nil channel.context
-    assert_not_same context, channel.context
-    channel = TelemetryChannel.new context
-    assert_same context, channel.context
-  end
-
-  def test_queue_works_as_expected
-    queue = SynchronousQueue.new SynchronousSender.new
-    channel = TelemetryChannel.new
-    assert_not_nil channel.queue
-    assert_not_same queue, channel.queue
-    channel = TelemetryChannel.new nil, queue
-    assert_same queue, channel.queue
-  end
-
-  def test_sender_works_as_expected
-    queue = SynchronousQueue.new SynchronousSender.new
-    channel = TelemetryChannel.new
-    assert_not_nil channel.sender
-    assert_not_same queue.sender, channel.sender
-    channel = TelemetryChannel.new nil, queue
-    assert_same queue.sender, channel.sender
-  end
-
-  def test_flush_works_as_expected
-    queue = MockTelemetryChannelQueue.new SynchronousSender.new
-    channel = TelemetryChannel.new nil, queue
-    assert_equal 0, queue.flush_count
-    channel.flush
-    assert_equal 1, queue.flush_count
-  end
-
-  def test_write_works_as_expected
-    queue = MockTelemetryChannelQueue.new SynchronousSender.new
-    context = TelemetryContext.new
-    context.instrumentation_key = 'instrumentation key'
-    channel = TelemetryChannel.new context, queue
-    expected = MockTelemetryItemData.new
-    channel.write expected
-    assert_equal 1, queue.queue.count
-    actual = queue.queue[0]
-    assert_not_nil actual
-    assert_equal 1, actual.ver
-    assert_equal 100, actual.sample_rate
-    assert_equal 'Microsoft.ApplicationInsights.MockTelemetryItem', actual.name
-    assert_not_nil actual.time
-    assert_equal 'instrumentation key', actual.i_key
-    assert_not_nil actual.tags
-    assert_equal 1, actual.tags.count
-    assert_equal 'rb:'+ ApplicationInsights::VERSION, actual.tags['ai.internal.sdkVersion']
-    assert_not_nil actual.data
-    assert_equal 'MockTelemetryItemData', actual.data.base_type
-    assert_same expected, actual.data.base_data
-  end
-end
-
-class MockTelemetryItemData
-  def initialize
-    @properties = {}
-  end
-
-  attr_accessor :properties
-end
-
-class MockTelemetryChannelQueue < QueueBase
-  def initialize(sender)
-    super sender
-    @queue = []
-    @flush_count = 0
-  end
-
-  attr_accessor :flush_count
-
-  attr_accessor :queue
-
-  def push(data)
-    @queue.push data
-  end
-
-  def flush
-    @flush_count += 1
-  end
-end
+require_relative '../../../lib/application_insights/channel/telemetry_channel'
+require_relative '../../../lib/application_insights/channel/telemetry_context'
+require_relative '../../../lib/application_insights/channel/synchronous_queue'
+require_relative '../../../lib/application_insights/channel/synchronous_sender'
+require 'test/unit'
+
+include ApplicationInsights::Channel
+
+class TestTelemetryChannel < Test::Unit::TestCase
+  def test_initialize
+    channel = TelemetryChannel.new
+    assert_not_nil channel.context
+    assert_not_nil channel.queue
+    assert_not_nil channel.sender
+  end
+
+  def test_context_works_as_expected
+    context = TelemetryContext.new
+    channel = TelemetryChannel.new
+    assert_not_nil channel.context
+    assert_not_same context, channel.context
+    channel = TelemetryChannel.new context
+    assert_same context, channel.context
+  end
+
+  def test_queue_works_as_expected
+    queue = SynchronousQueue.new SynchronousSender.new
+    channel = TelemetryChannel.new
+    assert_not_nil channel.queue
+    assert_not_same queue, channel.queue
+    channel = TelemetryChannel.new nil, queue
+    assert_same queue, channel.queue
+  end
+
+  def test_sender_works_as_expected
+    queue = SynchronousQueue.new SynchronousSender.new
+    channel = TelemetryChannel.new
+    assert_not_nil channel.sender
+    assert_not_same queue.sender, channel.sender
+    channel = TelemetryChannel.new nil, queue
+    assert_same queue.sender, channel.sender
+  end
+
+  def test_flush_works_as_expected
+    queue = MockTelemetryChannelQueue.new SynchronousSender.new
+    channel = TelemetryChannel.new nil, queue
+    assert_equal 0, queue.flush_count
+    channel.flush
+    assert_equal 1, queue.flush_count
+  end
+
+  def test_write_works_as_expected
+    queue = MockTelemetryChannelQueue.new SynchronousSender.new
+    context = TelemetryContext.new
+    context.instrumentation_key = 'instrumentation key'
+    channel = TelemetryChannel.new context, queue
+    expected = MockTelemetryItemData.new
+    channel.write expected
+    assert_equal 1, queue.queue.count
+    actual = queue.queue[0]
+    assert_not_nil actual
+    assert_equal 1, actual.ver
+    assert_equal 100, actual.sample_rate
+    assert_equal 'Microsoft.ApplicationInsights.MockTelemetryItem', actual.name
+    assert_not_nil actual.time
+    assert_equal 'instrumentation key', actual.i_key
+    assert_not_nil actual.tags
+    assert_equal 1, actual.tags.count
+    assert_equal 'rb:'+ ApplicationInsights::VERSION, actual.tags['ai.internal.sdkVersion']
+    assert_not_nil actual.data
+    assert_equal 'MockTelemetryItemData', actual.data.base_type
+    assert_same expected, actual.data.base_data
+  end
+
+  def test_get_tags_works_as_expected
+    queue = MockTelemetryChannelQueue.new SynchronousSender.new
+    context = TelemetryContext.new
+    context.application.ver = 'ver'
+    context.cloud.role_name = 'role name'
+    context.device.id = 'device id'
+    context.user.id = 'user id'
+    context.session.id = 'session id'
+    context.location.ip = 'ip'
+    context.operation.id = 'operation id'
+    channel = TelemetryChannel.new context, queue
+    expected = MockTelemetryItemData.new
+    channel.write expected
+
+    assert_equal 1, queue.queue.count
+    tags = queue.queue[0].tags
+    assert_equal 'rb:'+ ApplicationInsights::VERSION, tags['ai.internal.sdkVersion']
+    assert_equal 'ver', tags['ai.application.ver']
+    assert_equal 'role name', tags['ai.cloud.role']
+    assert_equal 'device id', tags['ai.device.id']
+    assert_equal 'user id', tags['ai.user.id']
+    assert_equal 'session id', tags['ai.session.id']
+    assert_equal 'ip', tags['ai.location.ip']
+    assert_equal 'operation id', tags['ai.operation.id']
+  end
+end
+
+class MockTelemetryItemData
+  def initialize
+    @properties = {}
+  end
+
+  attr_accessor :properties
+end
+
+class MockTelemetryChannelQueue < QueueBase
+  def initialize(sender)
+    super sender
+    @queue = []
+    @flush_count = 0
+  end
+
+  attr_accessor :flush_count, :queue
+
+  def push(data)
+    @queue.push data
+  end
+
+  def flush
+    @flush_count += 1
+  end
+end