basic TLV method tests

spec/lib/rex/post/meterpreter/packet_spec.rb

@@ -39,4 +39,55 @@ describe Rex::Post::Meterpreter::Tlv do
     subject.should respond_to :from_r
   end
 
+  context "A String TLV" do
+    it "should return the correct TLV type" do
+      subject.type.should == Rex::Post::Meterpreter::TLV_TYPE_STRING
+    end
+
+    it "should return the correct value" do
+      subject.value.should == "test"
+    end
+
+    context "the type? method" do
+      it "should return true for STRING" do
+        subject.type?(Rex::Post::Meterpreter::TLV_TYPE_STRING).should == true
+      end
+
+      it "should return false for UINT" do
+        subject.type?(Rex::Post::Meterpreter::TLV_TYPE_UINT).should == false
+      end
+    end
+
+    context "the value? method" do
+      it "should return true for the correct value" do
+        subject.value?("test").should == true
+      end
+
+      it "should return false for an incorrect value" do
+        subject.value?("fake").should == false
+      end
+    end
+
+    context "the inspect method" do
+      it "should return a string representation of the TLV" do
+        tlv_to_s = "#<Rex::Post::Meterpreter::Tlv type=STRING          meta=STRING     value=\"test\">"
+        subject.inspect.should == tlv_to_s
+      end
+    end
+
+    context "the to_r method" do
+      it "should return the raw bytes of the TLV to send over the wire" do
+        tlv_bytes = "\x00\x00\x00\r\x00\x01\x00\ntest\x00"
+        subject.to_r.should == tlv_bytes
+      end
+    end
+
+    context "the from_r method" do
+      it "should adjust the tlv attributes from the given raw bytes" do
+        subject.from_r("\x00\x00\x00\r\x00\x01\x00\ntes2\x00")
+        subject.value.should == "tes2"
+      end
+    end
+  end
+
 end