gitlab-foss/spec/models/note_diff_file_spec.rb

# frozen_string_literal: true

require 'spec_helper'

RSpec.describe NoteDiffFile, feature_category: :code_review_workflow do
  let(:diff_note) { create(:diff_note_on_commit) }
  let(:note_diff_file) { diff_note.note_diff_file }

  describe 'associations' do
    it { is_expected.to belong_to(:diff_note) }
  end

  describe 'validations' do
    it { is_expected.to validate_presence_of(:diff_note) }
  end

  describe '.referencing_sha' do
    let(:project) { diff_note.project }

    it 'finds note diff files by project and sha' do
      found = described_class.referencing_sha(diff_note.commit_id, project_id: project.id)

      expect(found).to contain_exactly(note_diff_file)
    end

    it 'excludes note diff files with the wrong project' do
      other_project = create(:project)

      found = described_class.referencing_sha(diff_note.commit_id, project_id: other_project.id)

      expect(found).to be_empty
    end

    it 'excludes note diff files with the wrong sha' do
      found = described_class.referencing_sha(Gitlab::Git::SHA1_BLANK_SHA, project_id: project.id)

      expect(found).to be_empty
    end
  end

  describe '#diff_export' do
    let_it_be(:encoded) { "b\xC3\xA5r" }
    let_it_be(:expected) { "bår" }

    before do
      note_diff_file.update!(diff: encoded)
    end

    context 'when diff can be encoded' do
      it 'force encodes the diff to UTF-8' do
        expect(note_diff_file.diff_export).to eq(expected)
        expect(note_diff_file.diff_export.encoding).to eq(Encoding::UTF_8)
      end
    end

    context 'when diff cannot be encoded' do
      it 'returns the raw diff' do
        allow(note_diff_file).to receive(:force_encode_utf8).with(encoded).and_raise(ArgumentError)

        expect(note_diff_file.diff_export).to eq(encoded)
      end
    end
  end
end