application_insights 0.1.0

data/test/application_insights/channel/contracts/test_stack_frame.rb

@@ -0,0 +1,83 @@
+require_relative '../../../../lib/application_insights/channel/contracts/stack_frame'
+require 'test/unit'
+
+include ApplicationInsights::Channel
+
+class TestStackFrame < Test::Unit::TestCase
+  def test_initialize
+    item = Contracts::StackFrame.new
+    assert_not_nil item
+  end
+  
+  def test_level_works_as_expected
+    expected = 42
+    item = Contracts::StackFrame.new
+    item.level = expected
+    actual = item.level
+    assert_equal expected, actual
+    expected = 13
+    item.level = expected
+    actual = item.level
+    assert_equal expected, actual
+  end
+  
+  def test_method_works_as_expected
+    expected = 'Test string'
+    item = Contracts::StackFrame.new
+    item.method = expected
+    actual = item.method
+    assert_equal expected, actual
+    expected = 'Other string'
+    item.method = expected
+    actual = item.method
+    assert_equal expected, actual
+  end
+  
+  def test_assembly_works_as_expected
+    expected = 'Test string'
+    item = Contracts::StackFrame.new
+    item.assembly = expected
+    actual = item.assembly
+    assert_equal expected, actual
+    expected = 'Other string'
+    item.assembly = expected
+    actual = item.assembly
+    assert_equal expected, actual
+  end
+  
+  def test_file_name_works_as_expected
+    expected = 'Test string'
+    item = Contracts::StackFrame.new
+    item.file_name = expected
+    actual = item.file_name
+    assert_equal expected, actual
+    expected = 'Other string'
+    item.file_name = expected
+    actual = item.file_name
+    assert_equal expected, actual
+  end
+  
+  def test_line_works_as_expected
+    expected = 42
+    item = Contracts::StackFrame.new
+    item.line = expected
+    actual = item.line
+    assert_equal expected, actual
+    expected = 13
+    item.line = expected
+    actual = item.line
+    assert_equal expected, actual
+  end
+  
+  def test_to_json_works_as_expected
+    item = Contracts::StackFrame.new
+    item.level = 42
+    item.method = 'Test string'
+    item.assembly = 'Test string'
+    item.file_name = 'Test string'
+    item.line = 42
+    actual = item.to_json
+    expected = '{"level":42,"method":"Test string","assembly":"Test string","fileName":"Test string","line":42}'
+    assert_equal expected, actual
+  end
+end

data/test/application_insights/channel/contracts/test_user.rb

@@ -0,0 +1,70 @@
+require_relative '../../../../lib/application_insights/channel/contracts/user'
+require 'test/unit'
+
+include ApplicationInsights::Channel
+
+class TestUser < Test::Unit::TestCase
+  def test_initialize
+    item = Contracts::User.new
+    assert_not_nil item
+  end
+  
+  def test_account_acquisition_date_works_as_expected
+    expected = 'Test string'
+    item = Contracts::User.new
+    item.account_acquisition_date = expected
+    actual = item.account_acquisition_date
+    assert_equal expected, actual
+    expected = 'Other string'
+    item.account_acquisition_date = expected
+    actual = item.account_acquisition_date
+    assert_equal expected, actual
+  end
+  
+  def test_account_id_works_as_expected
+    expected = 'Test string'
+    item = Contracts::User.new
+    item.account_id = expected
+    actual = item.account_id
+    assert_equal expected, actual
+    expected = 'Other string'
+    item.account_id = expected
+    actual = item.account_id
+    assert_equal expected, actual
+  end
+  
+  def test_user_agent_works_as_expected
+    expected = 'Test string'
+    item = Contracts::User.new
+    item.user_agent = expected
+    actual = item.user_agent
+    assert_equal expected, actual
+    expected = 'Other string'
+    item.user_agent = expected
+    actual = item.user_agent
+    assert_equal expected, actual
+  end
+  
+  def test_id_works_as_expected
+    expected = 'Test string'
+    item = Contracts::User.new
+    item.id = expected
+    actual = item.id
+    assert_equal expected, actual
+    expected = 'Other string'
+    item.id = expected
+    actual = item.id
+    assert_equal expected, actual
+  end
+  
+  def test_to_json_works_as_expected
+    item = Contracts::User.new
+    item.account_acquisition_date = 'Test string'
+    item.account_id = 'Test string'
+    item.user_agent = 'Test string'
+    item.id = 'Test string'
+    actual = item.to_json
+    expected = '{"ai.user.accountAcquisitionDate":"Test string","ai.user.accountId":"Test string","ai.user.userAgent":"Test string","ai.user.id":"Test string"}'
+    assert_equal expected, actual
+  end
+end

data/test/application_insights/channel/test_queue_base.rb

@@ -0,0 +1,88 @@
+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_raises_exception_when_sender_is_nil
+    begin
+      QueueBase.new nil
+      assert false
+    rescue ArgumentError => e
+      assert_equal 'Sender was required but not provided', e.message
+    end
+  end
+
+  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
+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

@@ -0,0 +1,96 @@
+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_raises_exception_when_service_endpoint_uri_is_nil
+    begin
+      SenderBase.new nil
+      assert false
+    rescue ArgumentError => e
+      assert_equal 'Service endpoint URI was required but not provided', e.message
+    end
+  end
+
+  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 [1, 2], 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

@@ -0,0 +1,42 @@
+require_relative '../../../lib/application_insights/channel/sender_base'
+require_relative '../../../lib/application_insights/channel/synchronous_queue'
+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
+    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.data
+    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
+
+class MockSynchronousSender
+  def initialize
+    @send_buffer_size = 2
+    @data = []
+    @queue = nil
+  end
+
+  attr_accessor :send_buffer_size
+  attr_accessor :data
+  attr_accessor :queue
+
+  def send(data_to_send)
+    @data.push data_to_send
+  end
+end

data/test/application_insights/channel/test_synchronous_sender.rb

@@ -0,0 +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 'http://dc.services.visualstudio.com/v2/track', sender.service_endpoint_uri
+  end
+end

data/test/application_insights/channel/test_telemetry_channel.rb

@@ -0,0 +1,102 @@
+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:0.1.0', 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