22
33import static com .datadog .debugger .agent .ConfigurationAcceptor .Source .REMOTE_CONFIG ;
44import static com .datadog .debugger .util .TestHelper .setFieldInConfig ;
5- import static datadog .trace .api .DDTags .DD_STACK_CODE_ORIGIN_FRAME ;
6- import static datadog .trace .api .DDTags .DD_STACK_CODE_ORIGIN_TYPE ;
5+ import static datadog .trace .api .DDTags .DD_CODE_ORIGIN_FRAME ;
6+ import static datadog .trace .api .DDTags .DD_CODE_ORIGIN_PREFIX ;
7+ import static datadog .trace .api .DDTags .DD_CODE_ORIGIN_TYPE ;
78import static java .lang .String .format ;
89import static java .util .Arrays .asList ;
910import static org .junit .jupiter .api .Assertions .assertEquals ;
@@ -191,16 +192,16 @@ private void waitForInstrumentation() {
191192 private static void checkEntrySpanTags (MutableSpan span , boolean includeSnapshot ) {
192193 String keys = format ("Existing keys for %s: %s" , span .getOperationName (), ldKeys (span ));
193194
194- assertEquals (span .getTag (DD_STACK_CODE_ORIGIN_TYPE ), "entry" , keys );
195- assertKeyPresent (span , DD_STACK_CODE_ORIGIN_TYPE );
196- assertKeyPresent (span , format (DD_STACK_CODE_ORIGIN_FRAME , 0 , "file" ));
197- assertKeyPresent (span , format (DD_STACK_CODE_ORIGIN_FRAME , 0 , "line" ));
198- assertKeyPresent (span , format (DD_STACK_CODE_ORIGIN_FRAME , 0 , "method" ));
199- assertKeyPresent (span , format (DD_STACK_CODE_ORIGIN_FRAME , 0 , "signature" ));
200- assertKeyPresent (span , format (DD_STACK_CODE_ORIGIN_FRAME , 0 , "type" ));
195+ assertEquals (span .getTag (DD_CODE_ORIGIN_TYPE ), "entry" , keys );
196+ assertKeyPresent (span , DD_CODE_ORIGIN_TYPE );
197+ assertKeyPresent (span , format (DD_CODE_ORIGIN_FRAME , 0 , "file" ));
198+ assertKeyPresent (span , format (DD_CODE_ORIGIN_FRAME , 0 , "line" ));
199+ assertKeyPresent (span , format (DD_CODE_ORIGIN_FRAME , 0 , "method" ));
200+ assertKeyPresent (span , format (DD_CODE_ORIGIN_FRAME , 0 , "signature" ));
201+ assertKeyPresent (span , format (DD_CODE_ORIGIN_FRAME , 0 , "type" ));
201202
202203 if (includeSnapshot ) {
203- assertKeyPresent (span , format (DD_STACK_CODE_ORIGIN_FRAME , 0 , "snapshot_id" ));
204+ assertKeyPresent (span , format (DD_CODE_ORIGIN_FRAME , 0 , "snapshot_id" ));
204205 }
205206 }
206207
@@ -224,26 +225,26 @@ private static void checkExitSpanTags(MutableSpan span, boolean includeSnapshot)
224225 String keys =
225226 format ("Existing keys for %s: %s" , span .getOperationName (), new TreeSet <>(ldKeys (span )));
226227
227- assertKeyPresent (span , DD_STACK_CODE_ORIGIN_TYPE );
228- assertKeyPresent (span , format (DD_STACK_CODE_ORIGIN_FRAME , 0 , "file" ));
229- assertKeyPresent (span , format (DD_STACK_CODE_ORIGIN_FRAME , 0 , "line" ));
230- assertKeyPresent (span , format (DD_STACK_CODE_ORIGIN_FRAME , 0 , "method" ));
231- assertKeyPresent (span , format (DD_STACK_CODE_ORIGIN_FRAME , 0 , "type" ));
228+ assertKeyPresent (span , DD_CODE_ORIGIN_TYPE );
229+ assertKeyPresent (span , format (DD_CODE_ORIGIN_FRAME , 0 , "file" ));
230+ assertKeyPresent (span , format (DD_CODE_ORIGIN_FRAME , 0 , "line" ));
231+ assertKeyPresent (span , format (DD_CODE_ORIGIN_FRAME , 0 , "method" ));
232+ assertKeyPresent (span , format (DD_CODE_ORIGIN_FRAME , 0 , "type" ));
232233 if (includeSnapshot ) {
233- assertKeyPresent (span , format (DD_STACK_CODE_ORIGIN_FRAME , 0 , "snapshot_id" ));
234+ assertKeyPresent (span , format (DD_CODE_ORIGIN_FRAME , 0 , "snapshot_id" ));
234235 }
235236
236237 MutableSpan rootSpan = span .getLocalRootSpan ();
237- assertEquals (rootSpan .getTag (DD_STACK_CODE_ORIGIN_TYPE ), "entry" , keys );
238- assertNotNull (rootSpan .getTag (format (DD_STACK_CODE_ORIGIN_FRAME , 1 , "file" )));
239- assertNotNull (rootSpan .getTag (format (DD_STACK_CODE_ORIGIN_FRAME , 1 , "line" )));
240- assertNotNull (rootSpan .getTag (format (DD_STACK_CODE_ORIGIN_FRAME , 1 , "method" )));
241- assertNotNull (rootSpan .getTag (format (DD_STACK_CODE_ORIGIN_FRAME , 1 , "type" )));
238+ assertEquals (rootSpan .getTag (DD_CODE_ORIGIN_TYPE ), "entry" , keys );
239+ assertNotNull (rootSpan .getTag (format (DD_CODE_ORIGIN_FRAME , 1 , "file" )));
240+ assertNotNull (rootSpan .getTag (format (DD_CODE_ORIGIN_FRAME , 1 , "line" )));
241+ assertNotNull (rootSpan .getTag (format (DD_CODE_ORIGIN_FRAME , 1 , "method" )));
242+ assertNotNull (rootSpan .getTag (format (DD_CODE_ORIGIN_FRAME , 1 , "type" )));
242243 }
243244
244245 private static Set <String > ldKeys (MutableSpan span ) {
245246 return span .getTags ().keySet ().stream ()
246- .filter (key -> key .startsWith ("_dd.ld" ) || key . startsWith ( "_dd.stack" ))
247+ .filter (key -> key .startsWith (DD_CODE_ORIGIN_PREFIX ))
247248 .collect (Collectors .toCollection (TreeSet ::new ));
248249 }
249250
0 commit comments