1 files changed, 44 insertions, 3 deletions

diff --git a/spec/serializers/diff_viewer_entity_spec.rb b/spec/serializers/diff_viewer_entity_spec.rb
index 53601fcff61..84d2bdceb78 100644
--- a/spec/serializers/diff_viewer_entity_spec.rb
+++ b/spec/serializers/diff_viewer_entity_spec.rb
@@ -12,10 +12,51 @@ RSpec.describe DiffViewerEntity do
   let(:diff) { commit.raw_diffs.first }
   let(:diff_file) { Gitlab::Diff::File.new(diff, diff_refs: diff_refs, repository: repository) }
   let(:viewer) { diff_file.simple_viewer }
+  let(:options) { {} }
 
-  subject { described_class.new(viewer).as_json }
+  subject { described_class.new(viewer).as_json(options) }
 
-  it 'serializes diff file viewer' do
-    expect(subject.with_indifferent_access).to match_schema('entities/diff_viewer')
+  context 'when add_ignore_all_white_spaces is enabled' do
+    before do
+      stub_feature_flags(add_ignore_all_white_spaces: true)
+    end
+
+    it 'serializes diff file viewer' do
+      expect(subject.with_indifferent_access).to match_schema('entities/diff_viewer')
+    end
+
+    it 'contains whitespace_only attribute' do
+      expect(subject.with_indifferent_access).to include(:whitespace_only)
+    end
+
+    context 'when whitespace_only option is true' do
+      let(:options) { { whitespace_only: true } }
+
+      it 'returns the whitespace_only attribute true' do
+        expect(subject.with_indifferent_access[:whitespace_only]).to eq true
+      end
+    end
+
+    context 'when whitespace_only option is false' do
+      let(:options) { { whitespace_only: false } }
+
+      it 'returns the whitespace_only attribute false' do
+        expect(subject.with_indifferent_access[:whitespace_only]).to eq false
+      end
+    end
+  end
+
+  context 'when add_ignore_all_white_spaces is disabled ' do
+    before do
+      stub_feature_flags(add_ignore_all_white_spaces: false)
+    end
+
+    it 'serializes diff file viewer' do
+      expect(subject.with_indifferent_access).to match_schema('entities/diff_viewer')
+    end
+
+    it 'does not contain whitespace_only attribute' do
+      expect(subject.with_indifferent_access).not_to include(:whitespace_only)
+    end
   end
 end