Update assertions to check payload types.

Author: a-feld

tests/agent_features/test_serverless_mode.py

@@ -12,6 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+import json
 import pytest
 
 from newrelic.api.application import application_instance
@@ -69,6 +70,9 @@ def _test():
     # Validate that something is printed to stdout
     assert out
 
+    # Verify that the payload is loadable JSON
+    payload = json.loads(out)
+
 
 def test_no_cat_headers(serverless_application):
     @background_task(

tests/agent_unittests/test_agent_protocol.py

@@ -532,7 +532,7 @@ def test_serverless_protocol_finalize(capsys):
     assert payload[:2] == [1, "NR_LAMBDA_MONITORING"]
 
     data = serverless_payload_decode(payload[2])
-    assert data["data"] == {"metric_data": "[1,2,3]"}
+    assert data["data"] == {"metric_data": [1,2,3]}
 
     assert data["metadata"]["foo"] == "bar"
     assert data["metadata"]["agent_version"] != "x"

tests/agent_unittests/test_http_client.py

@@ -542,24 +542,24 @@ def test_serverless_mode_client():
         for method in methods:
             params = {"method": method}
             status, data = client.send_request(
-                params=params, payload=method.encode("utf-8")
+                params=params, payload=json.dumps({"method": method}).encode("utf-8"),
             )
 
             assert status == 200
             assert json.loads(data.decode("utf-8"))
 
         # Verify that missing methods aren't captured
-        status, _ = client.send_request(payload=b"*")
+        status, _ = client.send_request(payload=b"{}")
         assert status == 400
 
         # Verify that invalid methods aren't captured
-        status, _ = client.send_request(params={"method": "foo"}, payload=b"*")
+        status, _ = client.send_request(params={"method": "foo"}, payload=b"{}")
         assert status == 400
 
     payloads = client.finalize()
     assert len(payloads) == len(methods)
     for method in methods:
-        assert payloads[method] == method.encode("utf-8")
+        assert payloads[method] == {"method": method}
 
 
 @pytest.mark.parametrize(

tests/testing_support/validators/validate_serverless_data.py

@@ -12,6 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+
 from newrelic.common.object_wrapper import (
         transient_function_wrapper,
         function_wrapper)
@@ -40,6 +41,9 @@ def _validate():
                 for method in expected_methods:
                     assert method in payload
 
+                    # Verify the method is not a byte string
+                    assert isinstance(payload[method], (dict, list))
+
                 for method in forgone_methods:
                     assert method not in payload