Class: Datadog::CI::TestDiscovery::Component

Inherits:
Object
  • Object
show all
Defined in:
lib/datadog/ci/test_discovery/component.rb

Overview

Test discovery mode component that manages test discovery output and lifecycle

Instance Method Summary collapse

Constructor Details

#initialize(enabled:, output_path:) ⇒ Component

Returns a new instance of Component.



13
14
15
16
17
18
19
20
21
22
 
# File 'lib/datadog/ci/test_discovery/component.rb', line 13

def initialize(
  enabled:,
  output_path:
)
  @enabled = enabled
  @output_path = output_path

  @buffer = []
  @buffer_mutex = Mutex.new
end

Instance Method Details

#configure(library_settings, test_session) ⇒ Object 



24
25
26
 
# File 'lib/datadog/ci/test_discovery/component.rb', line 24

def configure(library_settings, test_session)
  # This method is noop for this component, it is present for compatibility with other components
end

#disable_features_for_test_discovery!(settings) ⇒ Object 



32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
 
# File 'lib/datadog/ci/test_discovery/component.rb', line 32

def disable_features_for_test_discovery!(settings)
  return unless @enabled

  Datadog.logger.debug("ATTENTION! Running in test discovery mode, disabling all features")

  # in test discovery mode don't send anything to Datadog
  settings.ci.discard_traces = true

  # Disable all feature flags when in test discovery mode
  settings.telemetry.enabled = false
  settings.ci.itr_enabled = false
  settings.ci.git_metadata_upload_enabled = false
  settings.ci.retry_failed_tests_enabled = false
  settings.ci.retry_new_tests_enabled = false
  settings.ci.test_management_enabled = false
  settings.ci.agentless_logs_submission_enabled = false
  settings.ci.impacted_tests_detection_enabled = false
end

#enabled? ⇒ Boolean

Returns:

  • (Boolean) 


28
29
30
 
# File 'lib/datadog/ci/test_discovery/component.rb', line 28

def enabled?
  @enabled
end

#finish ⇒ Object 



70
71
72
73
74
75
76
 
# File 'lib/datadog/ci/test_discovery/component.rb', line 70

def finish
  return unless @enabled

  @buffer_mutex.synchronize do
    flush_buffer_unsafe if @buffer.any?
  end
end

#record_test(name:, suite:, module_name:, parameters:, source_file:) ⇒ Object 



78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
 
# File 'lib/datadog/ci/test_discovery/component.rb', line 78

def record_test(name:, suite:, module_name:, parameters:, source_file:)
  test_info = {
    "name" => name,
    "suite" => suite,
    "module" => module_name,
    "parameters" => parameters,
    "suiteSourceFile" => source_file
  }

  Datadog.logger.debug { "Discovered test: #{test_info}" }

  @buffer_mutex.synchronize do
    @buffer << test_info

    flush_buffer_unsafe if @buffer.size >= Ext::TestDiscovery::MAX_BUFFER_SIZE
  end
end

#shutdown! ⇒ Object 



96
97
98
99
100
101
102
 
# File 'lib/datadog/ci/test_discovery/component.rb', line 96

def shutdown!
  return unless @enabled

  @buffer_mutex.synchronize do
    flush_buffer_unsafe if @buffer.any?
  end
end

#start ⇒ Object 



51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
 
# File 'lib/datadog/ci/test_discovery/component.rb', line 51

def start
  return unless @enabled

  if @output_path.nil? || @output_path&.empty?
    @output_path = Ext::TestDiscovery::DEFAULT_OUTPUT_PATH
  end

  # thanks RBS for this weirdness
  output_path = @output_path
  return unless output_path

  output_dir = File.dirname(output_path)
  FileUtils.mkdir_p(output_dir) unless Dir.exist?(output_dir)

  Datadog.logger.debug { "Test discovery output path: #{output_path}" }

  @buffer_mutex.synchronize { @buffer.clear }
end