add WordlistFile specs (#13)

Author: moozzi

lib/ronin/wordlists/wordlist_file.rb

@@ -73,6 +73,9 @@ def initialize(path, url: nil)
       # @raise [DownloadFailed]
       #   Failed to download the wordlist file.
       #
+      # @return [WordlistFile]
+      #   The wordlist file.
+      #
       def self.download(url,dest=Dir.pwd)
         dest_path = begin
                       Core::System.download(url,dest)

spec/wordlist_file_spec.rb

@@ -0,0 +1,82 @@
+require 'spec_helper'
+require 'ronin/wordlists/wordlist_file'
+require 'ronin/core/system'
+
+describe Ronin::Wordlists::WordlistFile do
+  subject { described_class.new(path, url: url) }
+
+  let(:path) { File.expand_path(File.join(__dir__, '..', 'spec', 'fixtures', 'wordlists')) }
+  let(:url) { "www.example.com" }
+
+  describe "#initialzie" do
+    it "must initialize #path, #name and #url" do
+      expect(subject.path).to eq(path)
+      expect(subject.name).to eq("wordlists")
+      expect(subject.url).to eq("www.example.com")
+    end
+  end
+
+  describe ".download" do
+    context "when download succeed" do
+      let(:download_result) { described_class.download("valid.com") }
+
+      before do
+        allow(Ronin::Core::System).to receive(:download).and_return("/some/path")
+      end
+
+      it "must return new WordlistFile" do
+        expect(download_result).to be_kind_of(described_class)
+        expect(download_result.path).to eq("/some/path")
+        expect(download_result.url).to eq("valid.com")
+      end
+    end
+
+    context "when download failed" do
+      before do
+        allow(Ronin::Core::System).to receive(:download).and_raise(Ronin::Wordlists::DownloadFailed, "error message")
+      end
+
+      it "must raise a DownloadFailed error" do
+        expect {
+          described_class.download("invalid.com")
+        }.to raise_error(Ronin::Wordlists::DownloadFailed, "error message")
+      end
+    end
+  end
+
+  describe "#type" do
+    it "must return :file" do
+      expect(subject.type).to eq(:file)
+    end
+  end
+
+  describe "#filename" do
+    it "must return paths basename" do
+      expect(subject.filename).to eq("wordlists")
+    end
+  end
+
+  describe "#update" do
+    context "if #url is not nil" do
+      it "must download wordlist from url" do
+        expect(described_class).to receive(:download).with(subject.url, subject.path)
+
+        subject.update
+      end
+    end
+  end
+
+  describe "#delete" do
+    it "must delete the wordlist" do
+      expect(File).to receive(:unlink).with(subject.path)
+
+      subject.delete
+    end
+  end
+
+  describe "#to_s" do
+    it "must return path as string" do
+      expect(subject.to_s).to eq(path)
+    end
+  end
+end