diff options
author | Kamil Trzcinski <ayufan@ayufan.eu> | 2017-04-04 23:52:59 +0200 |
---|---|---|
committer | Kamil Trzcinski <ayufan@ayufan.eu> | 2017-04-06 14:32:39 +0200 |
commit | 062806e47da00dde37a2a62d7a5c6ede8341582c (patch) | |
tree | bd701e49203feee98d17de8c1444339dc70c56cc /spec/support/query_recorder.rb | |
parent | 0a4b853f2ceb0ad6c7394d1056f9139e53b38688 (diff) | |
download | gitlab-ce-062806e47da00dde37a2a62d7a5c6ede8341582c.tar.gz |
Define baseline for test for pipelines serializer
Diffstat (limited to 'spec/support/query_recorder.rb')
-rw-r--r-- | spec/support/query_recorder.rb | 14 |
1 files changed, 11 insertions, 3 deletions
diff --git a/spec/support/query_recorder.rb b/spec/support/query_recorder.rb index e40d5ebd9a8..55b531b4cf7 100644 --- a/spec/support/query_recorder.rb +++ b/spec/support/query_recorder.rb @@ -1,21 +1,29 @@ module ActiveRecord class QueryRecorder - attr_reader :log + attr_reader :log, :cached def initialize(&block) @log = [] + @cached = [] ActiveSupport::Notifications.subscribed(method(:callback), 'sql.active_record', &block) end def callback(name, start, finish, message_id, values) - return if %w(CACHE SCHEMA).include?(values[:name]) - @log << values[:sql] + if values[:name]&.include?("CACHE") + @cached << values[:sql] + elsif !values[:name]&.include?("SCHEMA") + @log << values[:sql] + end end def count @log.count end + def cached_count + @cached.count + end + def log_message @log.join("\n\n") end |