clean up tests

Author: joeyzhao2018

tests/test_tracing.py

@@ -408,15 +408,16 @@ def test_extracts_durable_trace_context_from_latest_checkpoint_operation_map(sel
             "CheckpointToken": "token",
             "InitialExecutionState": {
                 "Operations": {
-                    "0": {
-                        "Type": "EXECUTION",
-                        "ExecutionDetails": {
-                            "InputPayload": {"order_id": "ORD-300"}
-                        },
-                    },
+                    "0": {"Type": "EXECUTION"},
                     "1": {
                         "Name": "_datadog_0",
-                        "StepDetails": {"Result": {TraceHeader.TRACE_ID: "999"}},
+                        "StepDetails": {
+                            "Result": {
+                                TraceHeader.TRACE_ID: "999",
+                                TraceHeader.PARENT_ID: "888",
+                                TraceHeader.SAMPLING_PRIORITY: "1",
+                            }
+                        },
                     },
                     "2": {
                         "Name": "_datadog_1",
@@ -431,35 +432,7 @@ def test_extracts_durable_trace_context_from_latest_checkpoint_operation_map(sel
         self.assertEqual(source, "event")
         self.assertEqual(ctx, Context(trace_id=123, span_id=321, sampling_priority=1))
 
-    @with_trace_propagation_style("datadog")
-    def test_extracts_durable_trace_context_from_input_payload_when_no_checkpoint(self):
-        lambda_ctx = get_mock_context()
-        headers = {
-            TraceHeader.TRACE_ID: "777",
-            TraceHeader.PARENT_ID: "888",
-            TraceHeader.SAMPLING_PRIORITY: "1",
-        }
-
-        event = {
-            "DurableExecutionArn": "arn:aws:lambda:us-east-2:123456789012:function:demo:1/durable-execution/demo/first",
-            "CheckpointToken": "token",
-            "InitialExecutionState": {
-                "Operations": {
-                    "0": {
-                        "Type": "EXECUTION",
-                        "ExecutionDetails": {
-                            "InputPayload": json.dumps({"headers": headers})
-                        },
-                    }
-                }
-            },
-        }
-
-        ctx, source, _ = extract_dd_trace_context(event, lambda_ctx)
-
-        self.assertEqual(source, "event")
-        self.assertEqual(ctx, Context(trace_id=777, span_id=888, sampling_priority=1))
-
+    
     @with_trace_propagation_style("datadog")
     def test_with_extractor_function(self):
         def extractor_foo(event, context):