1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
|
require "get_process_mem"
require "rspec_profiling/example"
require "rspec_profiling/vcs/git"
require "rspec_profiling/vcs/svn"
require "rspec_profiling/vcs/git_svn"
require "rspec_profiling/collectors/csv"
require "rspec_profiling/collectors/json"
module RspecProfiling
class Run
def initialize(collector = RspecProfiling.config.collector.new,
vcs = RspecProfiling.config.vcs.new,
events = RspecProfiling.config.events)
@collector = collector
@vcs = vcs
@events = events
@seed = RSpec.configuration.seed
end
def start(*args)
start_counting_queries
start_counting_requests
start_counting_events
end
def example_started(example)
start_recording_memory
example = example.example if example.respond_to?(:example)
@current_example = Example.new(example)
end
def example_finished(*args)
collector.insert({
branch: vcs.branch,
commit_hash: vcs.sha,
seed: @seed,
date: vcs.time,
file: @current_example.file,
line_number: @current_example.line_number,
description: @current_example.description,
status: @current_example.status,
exception: @current_example.exception,
time: @current_example.time,
owner_tag: @current_example.owner_tag,
query_count: @current_example.query_count,
query_time: @current_example.query_time,
request_count: @current_example.request_count,
request_time: @current_example.request_time,
events: @events,
event_counts: @current_example.event_counts,
event_times: @current_example.event_times,
event_events: @current_example.event_events,
start_memory: @start_memory,
end_memory: end_memory
})
end
alias :example_passed :example_finished
alias :example_failed :example_finished
private
attr_reader :collector, :vcs, :events, :seed, :start_memory
def end_memory
GetProcessMem.new.mb
end
def start_recording_memory
@start_memory = GetProcessMem.new.mb
end
def start_counting_queries
ActiveSupport::Notifications.subscribe("sql.active_record") do |name, start, finish, id, query|
@current_example.try(:log_query, query, start, finish)
end
end
def start_counting_requests
ActiveSupport::Notifications.subscribe("process_action.action_controller") do |name, start, finish, id, request|
@current_example.try(:log_request, request, start, finish)
end
end
def start_counting_events
events.each do |event_name|
ActiveSupport::Notifications.subscribe(event_name) do |name, start, finish, id, event|
@current_example.try(:log_event, event_name, event, start, finish)
end
end
end
end
end
|
