diff options
Diffstat (limited to 'spec/lib/gitlab/diff/file_spec.rb')
-rw-r--r-- | spec/lib/gitlab/diff/file_spec.rb | 42 |
1 files changed, 42 insertions, 0 deletions
diff --git a/spec/lib/gitlab/diff/file_spec.rb b/spec/lib/gitlab/diff/file_spec.rb index 1800d2d6b60..4b437397688 100644 --- a/spec/lib/gitlab/diff/file_spec.rb +++ b/spec/lib/gitlab/diff/file_spec.rb @@ -51,6 +51,48 @@ RSpec.describe Gitlab::Diff::File do project.commit(branch_name).diffs.diff_files.first end + describe 'initialize' do + context 'when file is ipynb with a change after transformation' do + let(:commit) { project.commit("f6b7a707") } + let(:diff) { commit.raw_diffs.first } + let(:diff_file) { described_class.new(diff, diff_refs: commit.diff_refs, repository: project.repository) } + + context 'and :jupyter_clean_diffs is enabled' do + before do + stub_feature_flags(jupyter_clean_diffs: true) + end + + it 'recreates the diff by transforming the files' do + expect(diff_file.diff.diff).not_to include('"| Fake') + end + end + + context 'but :jupyter_clean_diffs is disabled' do + before do + stub_feature_flags(jupyter_clean_diffs: false) + end + + it 'does not recreate the diff' do + expect(diff_file.diff.diff).to include('"| Fake') + end + end + end + + context 'when file is ipynb, but there only changes that are removed' do + let(:commit) { project.commit("2b5ef814") } + let(:diff) { commit.raw_diffs.first } + let(:diff_file) { described_class.new(diff, diff_refs: commit.diff_refs, repository: project.repository) } + + before do + stub_feature_flags(jupyter_clean_diffs: true) + end + + it 'does not recreate the diff' do + expect(diff_file.diff.diff).to include('execution_count') + end + end + end + describe '#diff_lines' do let(:diff_lines) { diff_file.diff_lines } |