o i@sddlmZddlZddlZddlZddlZddlZddlmZddl Z ddl Z ddl m Z ddlmZddlmZddlmZddlmZdd lmZeeZd ed efd d Zdede jedffddZde jeeefedffddZ de j!j"dedefddZ#de$de j!j"fddZ%dedededefddZ&dededefd d!Z'd3d#d$Z(d%efd&d'Z)d(edefd)d*Z*d+e j+e,d,ee-e.d-fde j/fd.d/Z0d0ed,ee-e.d-fde j/fd1d2Z1dS)4wrapsN)sleep)config)PCT_COVERED_KEY)test)CIVISIBILITY_LOG_FILTER_RE)TEST_FRAMEWORKS) get_loggerpathstart_directorycCs@z tjj||d}W|Stytdtj|YSw)N)startzlTried to collect relative path but it is using different drive paths on Windows, using absolute path instead)osr relpath ValueErrorlogdebugabspath)r r relative_pathrX/home/ubuntu/.local/lib/python3.10/site-packages/ddtrace/internal/ci_visibility/utils.py&get_relative_or_absolute_path_for_paths rrepo_directoryreturncCs.zt|}Wn tyYdSwt||S)N)inspectgetfile TypeErrorr)test_method_objectr file_objectrrr$get_source_file_path_for_test_method"s   r NNc CsHzt|}Wn ttfyYdSw|d}|t|d}||fS)z Get the start and end line numbers for a test method. Returns: Tuple of (start_line, end_line), with None indicating unavailable information r!r)rgetsourcelinesrOSErrorlen)rsource_lines_tuple start_lineend_linerrr get_source_lines_for_test_method+s r)span test_namecCs|s td|dSt||}|std|dSt|\}}|r#|s)td||tj||r9|tj||rD|tj |dSdS)NzdTried to collect source start/end lines for test method %s but test method object could not be foundzKTried to collect file path for test %s but it is a built-in Python functionzVTried to collect source start/end lines for test method %s but an exception was raised) rrr r) _set_tag_strr SOURCE_FILE set_metric SOURCE_START SOURCE_END)r*rr+rsource_file_pathr'r(rrr,_add_start_end_source_file_path_data_to_span=s,   r2 coverage_datacCsL|rt|vr tddS|t}t|tstddS|tj|dS)NzKTried to add total covered percentage to session span but no data was foundzSTried to add total covered percentage to session span but the format was unexpected)rrwarning isinstancefloatr.rTEST_LINES_PCT)r3r*lines_pct_valuerrr_add_pct_covered_to_spanWs    r9test_module_pathtest_suite_namecCsd|||S)Nz{}.{}.{}format)r:r;r+rrr#_generate_fully_qualified_test_namebsr>cCs d||S)Nz{}.{}r<)r:r;rrr%_generate_fully_qualified_module_namefs r?Tcstjr tddStj}|dkrtd|dStd}|r5tdt|j D]}| |q,nt dt tt}|fdd |td z||||Wntytt d |YdSw|||t|j|jd |_td dS)a]Creates a handler with a filter for CIVisibility-specific messages. The also removes the existing handlers on the DDTrace logger, to prevent double-logging. This is useful for testrunners (currently pytest) that have their own logger. NOTE: This should **only** be called from testrunner-level integrations (eg: pytest, unittest). zQCIVisibility not taking over ddtrace logger handler because debug mode is enabledNNONEzGCIVisibility not taking over ddtrace logger because level is set to: %sddtracez,CIVisibility removing DDTrace logger handlerz8Keeping DDTrace logger handler, double logging is likelycst|jSN)boolmatchname)recordlogger_name_rerrsz1take_over_logger_stream_handler..zT[Datadog CI Visibility] %(levelname)-8s %(name)s:%(filename)s:%(lineno)d %(message)szInvalid log level: %sFzlogger setup complete)ddconfig _debug_moderr_ci_visibility_log_levelupperlogging getLoggerlisthandlers removeHandlerr4recompiler StreamHandler addFilter setFormatter FormattersetLevelr addHandlerminlevel propagate)remove_ddtrace_stream_handlersr\ddtrace_loggerhandlerci_visibility_handlerrrGrtake_over_logger_stream_handlerjs>           rbargscGsddd|DS)zCombine URL path segments. NOTE: this is custom-built for its current usage in the Test Visibility codebase. Use with care. /css|] }t|dVqdS)rdN)strstrip).0segmentrrr sz#combine_url_path..)join)rcrrrcombine_url_pathsrktest_frameworkcCs"tD] }|j|kr |SqtjSrB)r valueMANUAL)rl frameworkrrr"_get_test_framework_telemetry_names  rpafter exceptions.csfdd}|S)zX Decorator to automatically retry a function if it raises specified exceptions. cstfdd}|S)Nc sdD](}z |i|WSty*}zt|st|WYd}~qd}~ww|i|SrB) Exceptionr5r)rckwargsdelaye)rqrrfrr retry_wrappeds zCretry_on_exceptions..retry_decorator..retry_wrappedr)rwrxrqrr)rwrretry_decorators z,retry_on_exceptions..retry_decoratorr)rqrrrzrryrretry_on_exceptionssr{attemptscCstddt|dD|dS)zy Decorator to automatically retry a function if it raises specified exceptions, with exponential backoff delays. cSsg|] }tdd|qS)rgS?)randomuniform)rgirrr sz?fibonacci_backoff_with_jitter_on_exceptions..r"ry)r{range)r|rrrrr+fibonacci_backoff_with_jitter_on_exceptionssr)T)2 functoolsrrrNrr}rStimertypingtrArrJ'ddtrace.contrib.internal.coverage.patchr ddtrace.extr(ddtrace.internal.ci_visibility.constantsr2ddtrace.internal.ci_visibility.telemetry.constantsr ddtrace.internal.loggerr __name__rrerUnionr tupleintr)traceSpanr2dictr9r>r?rbrkrpIterabler6type BaseExceptionCallabler{rrrrrs`              1