Fix key for agent metadata based on schema: from 'memory_usage_mb' to 'memoryInMB'.

Author: mirelap-amazon

codeguru_profiler_agent/agent_metadata/agent_metadata.py

@@ -69,7 +69,7 @@ def fleet_info(self):
     def serialize_to_json(self, sample_weight, duration_ms, cpu_time_seconds,
                           average_num_threads, overhead_ms, memory_usage_mb, total_sample_count):
         """
-        This needs to be compliant with agent profile schema.
+        This needs to be compliant with the AgentMetadata schema that is used on the service side.
         """
         if self.json_rep is None:
             self.json_rep = {
@@ -83,7 +83,7 @@ def serialize_to_json(self, sample_weight, duration_ms, cpu_time_seconds,
                     "version": self.agent_info.version
                 },
                 "agentOverhead": {
-                    "memory_usage_mb": int(memory_usage_mb)
+                    "memoryInMB": int(memory_usage_mb)
                 },
                 "runtimeVersion": self.runtime_version,
                 "cpuTimeInSeconds": cpu_time_seconds,

test/acceptance/test_end_to_end_profile_and_save_to_file.py

@@ -92,7 +92,7 @@ def assert_valid_agent_metadata(agent_metadata):
         assert agent_metadata["agentOverhead"]
         assert agent_metadata["durationInMs"]
         assert agent_metadata["sampleWeights"]["WALL_TIME"]
-        assert type(agent_metadata["agentOverhead"]["memory_usage_mb"]) is int
+        assert type(agent_metadata["agentOverhead"]["memoryInMB"]) is int
 
         if platform.system() != "Windows":
             # Due to the issue mentioned on https://bugs.python.org/issue37859, we would skip checking agentOverhead for

test/unit/sdk_reporter/test_sdk_profile_encoder.py

@@ -122,7 +122,7 @@ def test_it_includes_the_overhead_ms_in_the_agent_metadata(self):
         assert (self.decoded_json_result()["agentMetadata"]["agentOverhead"]["timeInMs"] == 256)
 
     def test_it_includes_the_memory_overhead_in_the_agent_metadata(self):
-        assert (type(self.decoded_json_result()["agentMetadata"]["agentOverhead"]["memory_usage_mb"]) is int)
+        assert (type(self.decoded_json_result()["agentMetadata"]["agentOverhead"]["memoryInMB"]) is int)
 
     def test_it_includes_the_num_times_sampled_in_the_agent_metadata(self):
         assert (self.decoded_json_result()["agentMetadata"]["numTimesSampled"] > 0)