Raise an error if length is missing

rkh · May 15, 2011 · 7fdae39 · 7fdae39
1 parent 79b7a07
commit 7fdae39
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/lib/otnetstring.rb b/lib/otnetstring.rb
@@ -14,6 +14,8 @@ def self.parse(io)
 
     if length.size > 9
       raise Error, "#{length} is longer than 9 digits"
+    elsif length !~ /\d+/
+      raise Error, "Expected '#{byte}' to be a digit"
     end
     length = Integer(length)
 

diff --git a/spec/otnetstring_spec.rb b/spec/otnetstring_spec.rb
@@ -60,6 +60,12 @@
       OTNetstring.parse('4!true!').should == true
     end
 
+    it "raises an error if length is missing" do
+      lambda {
+        OTNetstring.parse('#123')
+      }.should raise_error(OTNetstring::Error, "Expected '#' to be a digit")
+    end
+
     it "raises an error if length is longer than 9 digits" do
       lambda {
         OTNetstring.parse('9' * 10 + ',')