Class: Threatinator::Actions::Run::CoverageObserver

Inherits:
Object
  • Object
show all
Defined in:
lib/threatinator/actions/run/coverage_observer.rb

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(filename) ⇒ CoverageObserver

Returns a new instance of CoverageObserver.

[View source]

8
9
10
11
 
# File 'lib/threatinator/actions/run/coverage_observer.rb', line 8

def initialize(filename)
  @filename = filename
  @csv = nil
end

Instance Attribute Details

#filename ⇒ Object (readonly)

Returns the value of attribute filename.


7
8
9
 
# File 'lib/threatinator/actions/run/coverage_observer.rb', line 7

def filename
  @filename
end

Instance Method Details

#close ⇒ Object

[View source]

48
49
50
 
# File 'lib/threatinator/actions/run/coverage_observer.rb', line 48

def close
  @csv.close unless closed?
end

#closed? ⇒ Boolean

Returns:

  • (Boolean)
[View source]

13
14
15
16
 
# File 'lib/threatinator/actions/run/coverage_observer.rb', line 13

def closed?
  return false if @csv.nil?
  @csv.closed?
end

#log_record(status, record, events = []) ⇒ Object

Parameters:

[View source]

41
42
43
44
45
46
 
# File 'lib/threatinator/actions/run/coverage_observer.rb', line 41

def log_record(status, record, events = [])
  return if closed?
  @csv.add_row(  [
    status, events.count, record.line_number, 
    record.pos_start, record.pos_end, record.data.inspect])
end

#open ⇒ Object

[View source]

18
19
20
 
# File 'lib/threatinator/actions/run/coverage_observer.rb', line 18

def open
  @csv = ::CSV.open(@filename, "wb", :headers => [:status, :event_count, :line_number, :pos_start, :pos_end, :data], :write_headers => true)
end

#update(message, *args) ⇒ Object

Handles FeedRunner observations

[View source]

23
24
25
26
27
28
29
30
31
32
33
34
35
36
 
# File 'lib/threatinator/actions/run/coverage_observer.rb', line 23

def update(message, *args)
  case message
  when :record_missed
    log_record(:missed, args.shift, [])
  when :record_filtered
    log_record(:filtered, args.shift, [])
  when :record_parsed
    log_record(:parsed, args.shift, args.shift)
  when :end
    close
  when :start
    open
  end
end