1 files changed, 16 insertions, 0 deletions
diff --git a/spec/lib/bulk_imports/network_error_spec.rb b/spec/lib/bulk_imports/network_error_spec.rb
index 11f555fee09..54d6554df96 100644
--- a/spec/lib/bulk_imports/network_error_spec.rb
+++ b/spec/lib/bulk_imports/network_error_spec.rb
@@ -46,6 +46,22 @@ RSpec.describe BulkImports::NetworkError, :clean_gitlab_redis_cache do
         expect(exception.retriable?(tracker)).to eq(false)
       end
     end
+
+    context 'when entity is passed' do
+      it 'increments entity cache key' do
+        entity = create(:bulk_import_entity)
+        exception = described_class.new('Error!')
+
+        allow(exception).to receive(:cause).and_return(SocketError.new('Error!'))
+
+        expect(Gitlab::Cache::Import::Caching)
+          .to receive(:increment)
+          .with("bulk_imports/#{entity.id}/network_error/SocketError")
+          .and_call_original
+
+        exception.retriable?(entity)
+      end
+    end
   end
 
   describe '#retry_delay' do