spec/lib/msf/db_manager/export_spec.rb - master - githubmirror/metasploit-framework

spec/lib/msf/db_manager/export_spec.rb
107 строк3 KB
dwelch-r7
Use zeitwerk for lib/msf/core folder
07 дек 2020, 13:31
07 дек 2020, 13:311617b3e
require 'spec_helper'


RSpec.describe Msf::DBExport do
  include_context 'Msf::DBManager'

  subject(:export) do
    described_class.new(workspace)
  end

  let(:active) do
    true
  end

  let(:workspace) do
    FactoryBot.create(
        :mdm_workspace
    )
  end

  context '#extract_module_detail_info' do
    let(:report_file) do
      StringIO.new
    end

    subject(:extract_module_detail_info) do
      export.extract_module_detail_info(report_file)
    end

    context 'with Mdm::Module::Details' do
      let(:document) do
        Nokogiri::XML(report_file.string)
      end

      let(:module_detail_count) do
        2
      end

      let(:root) do
        document.root
      end

      let!(:module_details) do
        FactoryBot.create_list(
            :mdm_module_detail,
            module_detail_count
        )
      end

      before(:example) do
        report_file.write("<root>")
        extract_module_detail_info
        report_file.write("</root>")
      end

      it 'should have module_detail tag for each Mdm::Module::Detail' do
        nodes = root.xpath('module_detail')

        expect(nodes.length).to eq module_detail_count
      end

      context 'module_detail' do
        let(:module_detail) do
          module_details.first
        end

        subject(:module_detail_node) do
          root.at_xpath('module_detail')
        end

        it_should_behave_like 'Msf::DBExport#extract_module_detail_info module_detail child', 'description'

        context '/disclosure-date' do
          it 'should have Mdm::Module::Detail#disclosure_date present' do
            expect(module_detail.disclosure_date).to be_present
          end

          it 'should have Mdm::Module::Detail#disclosure_date from disclosure-date content' do
            node = module_detail_node.at_xpath('disclosure-date')

            expect(DateTime.parse(node.content)).to eq module_detail.disclosure_date
          end
        end

        it_should_behave_like 'Msf::DBExport#extract_module_detail_info module_detail child', 'file'
        it_should_behave_like 'Msf::DBExport#extract_module_detail_info module_detail child', 'fullname'
        it_should_behave_like 'Msf::DBExport#extract_module_detail_info module_detail child', 'license'
        it_should_behave_like 'Msf::DBExport#extract_module_detail_info module_detail child', 'mtime'
        it_should_behave_like 'Msf::DBExport#extract_module_detail_info module_detail child', 'mtype'
        it_should_behave_like 'Msf::DBExport#extract_module_detail_info module_detail child', 'name'
        it_should_behave_like 'Msf::DBExport#extract_module_detail_info module_detail child', 'privileged'
        it_should_behave_like 'Msf::DBExport#extract_module_detail_info module_detail child', 'rank'
        it_should_behave_like 'Msf::DBExport#extract_module_detail_info module_detail child', 'refname'

        # @todo https://www.pivotaltracker.com/story/show/48451001
      end
    end

    context 'without Mdm::Module::Details' do
      it 'should not write anything to report_file' do
        extract_module_detail_info

        expect(report_file.string).to be_empty
      end
    end
  end
end