From 62ca1ae260e4ffff60930af2d42bd8b8b15128cc Mon Sep 17 00:00:00 2001 From: Andrei Eres Date: Mon, 22 Mar 2021 15:13:47 +0300 Subject: [PATCH] Add names for department to Registry --- lib/rubocop/cop/registry.rb | 4 ++++ spec/rubocop/cop/registry_spec.rb | 7 +++++++ 2 files changed, 11 insertions(+) diff --git a/lib/rubocop/cop/registry.rb b/lib/rubocop/cop/registry.rb index 76b7a313910..492bbd21321 100644 --- a/lib/rubocop/cop/registry.rb +++ b/lib/rubocop/cop/registry.rb @@ -184,6 +184,10 @@ def names cops.map(&:cop_name) end + def names_for_department(department) + cops.select { |cop| cop.department == department.to_sym }.map(&:cop_name) + end + def ==(other) cops == other.cops end diff --git a/spec/rubocop/cop/registry_spec.rb b/spec/rubocop/cop/registry_spec.rb index 6ed390043f5..75c15430b6d 100644 --- a/spec/rubocop/cop/registry_spec.rb +++ b/spec/rubocop/cop/registry_spec.rb @@ -350,4 +350,11 @@ expect(registry.department?('Foo')).to be false end end + + describe 'names_for_department' do + it 'returns array of cops for specified department' do + expect(registry.names_for_department('Lint')) + .to eq %w[Lint/BooleanSymbol Lint/DuplicateMethods] + end + end end