Add context_info to SQL Server activity samples (DBMON-4106) (#17648)

* Add context_info to activity samples

* change log entry

* removed change log entry

* Update sqlserver/tests/test_activity.py

Co-authored-by: Justin <justin.james.iso@gmail.com>

* linter errors

---------

Co-authored-by: root <root@ip-172-29-198-131.ec2.internal>
Co-authored-by: Justin <justin.james.iso@gmail.com>

Author: 3 people

sqlserver/datadog_checks/sqlserver/activity.py

@@ -136,6 +136,7 @@
     "row_count",
     "query_hash",
     "query_plan_hash",
+    "context_info",
 ]
 
 
@@ -333,6 +334,10 @@ def _sanitize_row(row, obfuscated_statement=None):
         # remove deobfuscated sql text from event
         if 'statement_text' in row:
             del row['statement_text']
+        # convert `context_info` to string, since the default decoding format
+        # for json is utf-8, and `context_info` need not comply to utf-8
+        if 'context_info' in row:
+            row['context_info'] = row['context_info'].hex()
         return row
 
     @staticmethod

sqlserver/tests/test_activity.py

@@ -99,6 +99,9 @@ def test_collect_load_activity(
     def run_test_query(c, q):
         cur = c.cursor()
         cur.execute("USE {}".format(database))
+        # 0xFF can't be decoded to Unicode, which makes it good test data,
+        # since Unicode is a default format
+        cur.execute("SET CONTEXT_INFO 0xff")
         cur.execute(q)
 
     # run the test query once before the blocking test to ensure that if it's
@@ -164,6 +167,7 @@ def run_test_query(c, q):
         assert blocked_row['procedure_name'], "missing procedure name"
     assert re.match(match_pattern, blocked_row['text'], re.IGNORECASE), "incorrect blocked query"
     assert blocked_row['database_name'] == "datadog_test", "incorrect database_name"
+    assert blocked_row['context_info'] == "ff", "incorrect context_info"
     assert blocked_row['id'], "missing session id"
     assert blocked_row['now'], "missing current timestamp"
     assert blocked_row['last_request_start_time'], "missing last_request_start_time"