demarches-normaliennes/spec/services/clamav_service_spec.rb

describe ClamavService do
  describe '.safe_file?' do
    let(:path_file) { '/tmp/plop.txt' }

    subject { ClamavService.safe_file?(path_file) }

    context "when ClamAV is enabled" do
      before do
        @saved_clamav = Rails.configuration.x.clamav.enabled
        Rails.configuration.x.clamav.enabled = true

        client = double("ClamAV::Client", execute: [response])
        allow(ClamAV::Client).to receive(:new).and_return(client)
        allow(FileUtils).to receive(:chmod).with(0666, path_file).and_return(true)
      end

      after do
        Rails.configuration.x.clamav.enabled = @saved_clamav
      end

      context 'When response type is ClamAV::SuccessResponse' do
        let(:response) { ClamAV::SuccessResponse.new("OK") }
        it { expect(subject).to eq(true) }
      end

      context 'When response type is ClamAV::VirusResponse' do
        let(:response) { ClamAV::VirusResponse.new("KO", "VirusN4ame") }
        it { expect(subject).to eq(false) }
      end

      context 'When response type is ClamAV::ErrorResponse' do
        let(:response) { ClamAV::ErrorResponse.new("File not found") }
        it { expect { subject }.to raise_error("ClamAV ErrorResponse : File not found") }
      end
    end

    context "when ClamAV is disabled" do
      before do
        @saved_clamav = Rails.configuration.x.clamav.enabled
        Rails.configuration.x.clamav.enabled = false
      end

      after do
        Rails.configuration.x.clamav.enabled = @saved_clamav
      end

      it do
        expect(ClamAV::Client).not_to receive(:new)
        expect(FileUtils).not_to receive(:chmod)
        expect(subject).to eq(true)
      end
    end
  end
end