Added specs for --output and --output-format.

ronin-rb · Jul 6, 2024 · 1d8232d · 1d8232d
1 parent 4b203d8
commit 1d8232d
Show file tree

Hide file tree

Showing 2 changed files with 208 additions and 0 deletions.
diff --git a/spec/cli/commands/dns_spec.rb b/spec/cli/commands/dns_spec.rb
@@ -4,4 +4,108 @@
 
 describe Ronin::Listener::CLI::Commands::Dns do
   include_examples "man_page"
+
+  describe "options" do
+    before { subject.option_parser.parse(argv) }
+
+    describe "when given the '--output FILE' option" do
+      let(:argv) { ['--output', file] }
+
+      describe "and the file ends with '.txt'" do
+        let(:file) { 'log.txt' }
+
+        it "must set options[:output] to the given file" do
+          expect(subject.options[:output]).to eq(file)
+        end
+
+        it "must set options[:output_format] to Ronin::Listener::OutputFormats::TXT" do
+          expect(subject.options[:output_format]).to be(Ronin::Listener::OutputFormats::TXT)
+        end
+      end
+
+      describe "and the file ends with '.csv'" do
+        let(:file) { 'log.csv' }
+
+        it "must set options[:output] to the given file" do
+          expect(subject.options[:output]).to eq(file)
+        end
+
+        it "must set options[:output_format] to Ronin::Listener::OutputFormats::CSV" do
+          expect(subject.options[:output_format]).to be(Ronin::Listener::OutputFormats::CSV)
+        end
+      end
+
+      describe "and the file ends with '.json'" do
+        let(:file) { 'log.json' }
+
+        it "must set options[:output] to the given file" do
+          expect(subject.options[:output]).to eq(file)
+        end
+
+        it "must set options[:output_format] to Ronin::Listener::OutputFormats::JSON" do
+          expect(subject.options[:output_format]).to be(Ronin::Listener::OutputFormats::JSON)
+        end
+      end
+
+      describe "and the file ends with '.ndjson'" do
+        let(:file) { 'log.ndjson' }
+
+        it "must set options[:output] to the given file" do
+          expect(subject.options[:output]).to eq(file)
+        end
+
+        it "must set options[:output_format] to Ronin::Listener::OutputFormats::NDJSON" do
+          expect(subject.options[:output_format]).to be(Ronin::Listener::OutputFormats::NDJSON)
+        end
+      end
+
+      describe "but the file's extension is not recognized" do
+        let(:file) { 'log.foo' }
+
+        it "must set options[:output] to the given file" do
+          expect(subject.options[:output]).to eq(file)
+        end
+
+        it "must set options[:output_format] to Ronin::Listener::OutputFormats::TXT" do
+          expect(subject.options[:output_format]).to be(Ronin::Listener::OutputFormats::TXT)
+        end
+      end
+    end
+
+    describe "when given the '--output-format FORMAT' option" do
+      let(:argv) { ['--output-format', format] }
+
+      describe "and when FORMAT is 'txt'" do
+        let(:format) { 'txt' }
+
+        it "must set options[:output_format] to Ronin::Listener::OutputFormats::TXT" do
+          expect(subject.options[:output_format]).to be(Ronin::Listener::OutputFormats::TXT)
+        end
+      end
+
+      describe "and when FORMAT is 'csv'" do
+        let(:format) { 'csv' }
+
+        it "must set options[:output_format] to Ronin::Listener::OutputFormats::CSV" do
+          expect(subject.options[:output_format]).to be(Ronin::Listener::OutputFormats::CSV)
+        end
+      end
+
+      describe "and when FORMAT is 'json'" do
+        let(:format) { 'json' }
+
+        it "must set options[:output_format] to Ronin::Listener::OutputFormats::JSON" do
+          expect(subject.options[:output_format]).to be(Ronin::Listener::OutputFormats::JSON)
+        end
+      end
+
+      describe "and when FORMAT is 'ndjson'" do
+        let(:format) { 'ndjson' }
+
+        it "must set options[:output_format] to Ronin::Listener::OutputFormats::NDJSON" do
+          expect(subject.options[:output_format]).to be(Ronin::Listener::OutputFormats::NDJSON)
+        end
+      end
+    end
+  end
 end
diff --git a/spec/cli/commands/http_spec.rb b/spec/cli/commands/http_spec.rb
@@ -4,4 +4,108 @@
 
 describe Ronin::Listener::CLI::Commands::Http do
   include_examples "man_page"
+
+  describe "options" do
+    before { subject.option_parser.parse(argv) }
+
+    describe "when given the '--output FILE' option" do
+      let(:argv) { ['--output', file] }
+
+      describe "and the file ends with '.txt'" do
+        let(:file) { 'log.txt' }
+
+        it "must set options[:output] to the given file" do
+          expect(subject.options[:output]).to eq(file)
+        end
+
+        it "must set options[:output_format] to Ronin::Listener::OutputFormats::TXT" do
+          expect(subject.options[:output_format]).to be(Ronin::Listener::OutputFormats::TXT)
+        end
+      end
+
+      describe "and the file ends with '.csv'" do
+        let(:file) { 'log.csv' }
+
+        it "must set options[:output] to the given file" do
+          expect(subject.options[:output]).to eq(file)
+        end
+
+        it "must set options[:output_format] to Ronin::Listener::OutputFormats::CSV" do
+          expect(subject.options[:output_format]).to be(Ronin::Listener::OutputFormats::CSV)
+        end
+      end
+
+      describe "and the file ends with '.json'" do
+        let(:file) { 'log.json' }
+
+        it "must set options[:output] to the given file" do
+          expect(subject.options[:output]).to eq(file)
+        end
+
+        it "must set options[:output_format] to Ronin::Listener::OutputFormats::JSON" do
+          expect(subject.options[:output_format]).to be(Ronin::Listener::OutputFormats::JSON)
+        end
+      end
+
+      describe "and the file ends with '.ndjson'" do
+        let(:file) { 'log.ndjson' }
+
+        it "must set options[:output] to the given file" do
+          expect(subject.options[:output]).to eq(file)
+        end
+
+        it "must set options[:output_format] to Ronin::Listener::OutputFormats::NDJSON" do
+          expect(subject.options[:output_format]).to be(Ronin::Listener::OutputFormats::NDJSON)
+        end
+      end
+
+      describe "but the file's extension is not recognized" do
+        let(:file) { 'log.foo' }
+
+        it "must set options[:output] to the given file" do
+          expect(subject.options[:output]).to eq(file)
+        end
+
+        it "must set options[:output_format] to Ronin::Listener::OutputFormats::TXT" do
+          expect(subject.options[:output_format]).to be(Ronin::Listener::OutputFormats::TXT)
+        end
+      end
+    end
+
+    describe "when given the '--output-format FORMAT' option" do
+      let(:argv) { ['--output-format', format] }
+
+      describe "and when FORMAT is 'txt'" do
+        let(:format) { 'txt' }
+
+        it "must set options[:output_format] to Ronin::Listener::OutputFormats::TXT" do
+          expect(subject.options[:output_format]).to be(Ronin::Listener::OutputFormats::TXT)
+        end
+      end
+
+      describe "and when FORMAT is 'csv'" do
+        let(:format) { 'csv' }
+
+        it "must set options[:output_format] to Ronin::Listener::OutputFormats::CSV" do
+          expect(subject.options[:output_format]).to be(Ronin::Listener::OutputFormats::CSV)
+        end
+      end
+
+      describe "and when FORMAT is 'json'" do
+        let(:format) { 'json' }
+
+        it "must set options[:output_format] to Ronin::Listener::OutputFormats::JSON" do
+          expect(subject.options[:output_format]).to be(Ronin::Listener::OutputFormats::JSON)
+        end
+      end
+
+      describe "and when FORMAT is 'ndjson'" do
+        let(:format) { 'ndjson' }
+
+        it "must set options[:output_format] to Ronin::Listener::OutputFormats::NDJSON" do
+          expect(subject.options[:output_format]).to be(Ronin::Listener::OutputFormats::NDJSON)
+        end
+      end
+    end
+  end
 end