Fix code origin setup race (#7266)

Co-authored-by: datadog-bits <263423550+datadog-bits@users.noreply.github.com>

Author: P403n1x87

tests/debugger/test_debugger_inproduct_enablement.py

@@ -3,7 +3,7 @@
 # Copyright 2021 Datadog, Inc.
 
 import tests.debugger.utils as debugger
-from utils import context, features, logger, scenarios, slow
+from utils import context, features, interfaces, logger, scenarios, slow
 import json
 import time
 
@@ -220,8 +220,14 @@ def test_inproduct_enablement_code_origin(self):
 class Test_Debugger_InProduct_Enablement_Code_Origin_Default_On(debugger.BaseDebuggerTest):
     def setup_code_origin_enabled_by_default(self):
         self.initialize_weblog_remote_config()
+        threshold = self._get_max_trace_file_number()
+        self._span_found = False
         self.send_weblog_request("/")
-        self.code_origin_enabled_by_default = self.wait_for_code_origin_span(TIMEOUT)
+        interfaces.agent.wait_for(
+            lambda data: self._wait_for_code_origin_span(data, threshold=threshold),
+            timeout=TIMEOUT,
+        )
+        self.code_origin_enabled_by_default = self._span_found
 
     def test_code_origin_enabled_by_default(self):
         self.assert_setup_ok()