Jelajahi Sumber

Add two missing test cases

Benjamin Sago 4 tahun lalu
induk
melakukan
42ebb61d40
1 mengubah file dengan 12 tambahan dan 0 penghapusan
  1. 12 0
      src/options.rs

+ 12 - 0
src/options.rs

@@ -766,12 +766,24 @@ mod test {
                    OptionsResult::InvalidOptions(OptionsError::InvalidQueryClass("tubes".into())));
     }
 
+    #[test]
+    fn invalid_named_class_too_big() {
+        assert_eq!(Options::getopts(&[ "lookup.dog", "--class", "999999" ]),
+                   OptionsResult::InvalidOptions(OptionsError::InvalidQueryClass("999999".into())));
+    }
+
     #[test]
     fn invalid_named_type() {
         assert_eq!(Options::getopts(&[ "lookup.dog", "--type", "tubes" ]),
                    OptionsResult::InvalidOptions(OptionsError::InvalidQueryType("tubes".into())));
     }
 
+    #[test]
+    fn invalid_named_type_too_big() {
+        assert_eq!(Options::getopts(&[ "lookup.dog", "--type", "999999" ]),
+                   OptionsResult::InvalidOptions(OptionsError::InvalidQueryType("999999".into())));
+    }
+
     #[test]
     fn invalid_capsword() {
         assert_eq!(Options::getopts(&[ "SMH", "lookup.dog" ]),