Merge pull request #13 from wchen-r7/pr6455

Add rspec for lib/net/dns/names/names.rb

Author: Vex Woo

spec/lib/net/dns/names/names_spec.rb

@@ -0,0 +1,129 @@
+require 'msf/core'
+
+RSpec.describe Net::DNS::Names do
+  subject do
+    obj = Object.new
+    obj.extend(described_class)
+  end
+
+  describe '#dn_expand' do
+    context 'when offset is great than packet length' do
+      let(:packet) do
+        'AAAAA'
+      end
+
+      let(:offset) do
+        10
+      end
+
+      it 'raises an ExpandError exception' do
+        expect { subject.dn_expand(packet, offset) }.to raise_exception(ExpandError)
+      end
+    end
+
+    context 'when packet length is less than offset + INT16SZ' do
+      let(:packet) do
+        "\xc0"
+      end
+
+      let(:offset) do
+        0
+      end
+
+      it 'raises an ExpandError exception' do
+        expect { subject.dn_expand(packet, offset) }.to raise_exception(ExpandError)
+      end
+    end
+
+    context 'when packet length is less than offset + packet length' do
+      let(:packet) do
+        'AAAAA'
+      end
+
+      let(:offset) do
+        4
+      end
+
+      it 'raises an ExpandError exception' do
+        expect { subject.dn_expand(packet, offset) }.to raise_exception(ExpandError)
+      end
+    end
+  end
+
+  describe '#pack_name' do
+    context 'when name data size is larger than 255 bytes' do
+      let(:name) do
+        'A' * (255+1)
+      end
+
+      it 'raises an ArgumentError exception' do
+        expect { subject.pack_name(name) }.to raise_exception(ArgumentError)
+      end
+    end
+
+    context 'when label data is larger than 63 bytes' do
+      let(:name) do
+        'A' * (63+1) + '.'
+      end
+
+      it 'raises an ArgumentError exception' do
+        expect { subject.pack_name(name) }.to raise_exception(ArgumentError)
+      end 
+    end
+  end
+
+  describe '#names_array' do
+    let(:name) do
+      "AAA.AAA"
+    end
+
+    it 'returns an Array' do
+      expect(subject.names_array(name)).to be_kind_of(Array)
+    end
+  end
+
+  describe '#dn_comp' do
+    let(:name) do
+      'AAAA'
+    end
+
+    let(:offset) do
+      0
+    end
+
+    let(:compnames) do
+      {}
+    end
+
+    it 'returns 3 values' do
+      v = subject.dn_comp(name, offset, compnames)
+      expect(v.length).to eq(3)
+      expect(v[0]).to be_kind_of(String)
+      expect(v[1]).to be_kind_of(Fixnum)
+      expect(v[2]).to be_kind_of(Hash)
+    end
+  end
+
+  describe '#valid?' do
+    context 'when FQDN is valid' do
+      let(:fqdn) do
+        'example.com'
+      end
+
+      it 'returns the FQDN' do
+        expect(subject.valid?(fqdn)).to eq(fqdn)
+      end
+
+    end
+
+    context 'when FQDN is not valid' do
+      let(:fqdn) do
+        'INVALID'
+      end
+
+      it 'raises ArgumentError exception' do
+        expect { subject.valid?(fqdn) }.to raise_exception(ArgumentError)
+      end
+    end
+  end
+end