diff options
| -rw-r--r-- | spec/services/users/keys_count_service_spec.rb | 10 |
1 files changed, 3 insertions, 7 deletions
diff --git a/spec/services/users/keys_count_service_spec.rb b/spec/services/users/keys_count_service_spec.rb index ccfd922b8fa..59152669e20 100644 --- a/spec/services/users/keys_count_service_spec.rb +++ b/spec/services/users/keys_count_service_spec.rb @@ -11,20 +11,16 @@ describe Users::KeysCountService, :use_clean_rails_memory_store_caching do create(:personal_key, user: user) end - let(:num_queries) { 1 } - it 'returns the number of SSH keys as an Integer' do expect(service.count).to eq(1) end it 'caches the number of keys in Redis' do service.delete_cache + control_count = ActiveRecord::QueryRecorder.new { service.count }.count + service.delete_cache - recorder = ActiveRecord::QueryRecorder.new do - 2.times { service.count } - end - - expect(recorder.count).to eq(num_queries) + expect { 2.times { service.count } }.not_to exceed_query_limit(control_count + 1) end end |