From cc517ea037f3dd3a69f6095afa34b0923a80abba Mon Sep 17 00:00:00 2001
From: Ivan Ramirez <ivan.ramirez@gmail.com>
Date: Thu, 19 Sep 2019 17:31:37 +0800
Subject: [PATCH] Add support for exporting Custom Events to Azure App
 Insights.

Enable exporter for custom  events using the same logging pipeline available in OC.
---
 .../ext/azure/events_exporter/__init__.py     | 24 +++++++
 .../tests/test_azure_event_exporter.py        | 69 +++++++++++++++++++
 2 files changed, 93 insertions(+)
 create mode 100644 contrib/opencensus-ext-azure/opencensus/ext/azure/events_exporter/__init__.py
 create mode 100644 contrib/opencensus-ext-azure/tests/test_azure_event_exporter.py

diff --git a/contrib/opencensus-ext-azure/opencensus/ext/azure/events_exporter/__init__.py b/contrib/opencensus-ext-azure/opencensus/ext/azure/events_exporter/__init__.py
new file mode 100644
index 000000000..1805f5808
--- /dev/null
+++ b/contrib/opencensus-ext-azure/opencensus/ext/azure/events_exporter/__init__.py
@@ -0,0 +1,24 @@
+from opencensus.ext.azure.common import utils
+from opencensus.ext.azure.common.protocol import Data, Envelope, Event
+from opencensus.ext.azure.log_exporter import AzureLogHandler
+
+
+class AzureEventHandler(AzureLogHandler):
+    def __init__(self, **options):
+        super(AzureEventHandler, self).__init__(**options)
+
+    def log_record_to_envelope(self, record):
+        envelope = Envelope(
+            iKey=self.options.instrumentation_key,
+            tags=dict(utils.azure_monitor_context),
+            time=utils.timestamp_to_iso_str(record.created),
+        )
+
+        envelope.name = "Microsoft.ApplicationInsights.Event"
+        data = Event(
+            name=record.msg,
+            properties=record.args,
+            measurements=None,
+        )
+        envelope.data = Data(baseData=data, baseType='EventData')
+        return envelope
diff --git a/contrib/opencensus-ext-azure/tests/test_azure_event_exporter.py b/contrib/opencensus-ext-azure/tests/test_azure_event_exporter.py
new file mode 100644
index 000000000..1c970deee
--- /dev/null
+++ b/contrib/opencensus-ext-azure/tests/test_azure_event_exporter.py
@@ -0,0 +1,69 @@
+import logging
+import os
+import shutil
+import unittest
+
+import mock
+
+from opencensus.ext.azure import events_exporter
+
+TEST_FOLDER = os.path.abspath('.test.logs')
+
+
+def setUpModule():
+    os.makedirs(TEST_FOLDER)
+
+
+def tearDownModule():
+    shutil.rmtree(TEST_FOLDER)
+
+
+class TestAzureEventHandler(unittest.TestCase):
+    def test_log_record_to_envelope(self):
+        handler = events_exporter.AzureEventHandler(
+            instrumentation_key='12345678-1234-5678-abcd-12345678abcd',
+            storage_path=os.path.join(TEST_FOLDER, self.id()),
+        )
+        envelope = handler.log_record_to_envelope(mock.MagicMock(
+            msg="test"
+        ))
+        self.assertEqual(
+            envelope.iKey,
+            '12345678-1234-5678-abcd-12345678abcd')
+
+        self.assertEqual(
+            envelope.name,
+            'Microsoft.ApplicationInsights.Event')
+
+        self.assertEqual(
+            envelope.get("data").baseData.get("name"),
+            "test")
+        handler.close()
+
+    def test_log_record_to_envelope_with_properties(self):
+        handler = events_exporter.AzureEventHandler(
+            instrumentation_key='12345678-1234-5678-abcd-12345678abcd',
+            storage_path=os.path.join(TEST_FOLDER, self.id()),
+        )
+        envelope = handler.log_record_to_envelope(mock.MagicMock(
+            msg="test",
+            args={"sku": "SKU-12312"}
+        ))
+        self.assertEqual(
+            len(envelope.get("data").baseData.get("properties")),
+            1)
+        handler.close()
+
+    @mock.patch('requests.post', return_value=mock.Mock())
+    def test_export(self, requests_mock):
+        logger = logging.getLogger(self.id())
+        handler = events_exporter.AzureEventHandler(
+            instrumentation_key='12345678-1234-5678-abcd-12345678abcd',
+            storage_path=os.path.join(TEST_FOLDER, self.id()),
+        )
+        logger.addHandler(handler)
+        logger.warning('test_metric')
+        handler.close()
+        self.assertEqual(len(requests_mock.call_args_list), 1)
+        post_body = requests_mock.call_args_list[0][1]['data']
+        self.assertTrue('test_metric' in post_body)