Merge df40335 into 6bddbe3

bddrest/helpers.py

@@ -1,3 +1,4 @@
+import re
 import threading
 from urllib.parse import parse_qs
 
@@ -100,3 +101,40 @@ def normalize_query_string(query):
         k: v[0] if len(v) == 1 else v for k, v in parse_qs(query).items()
     } if isinstance(query, str) else query
 
+
+
+class Status:
+
+    def __init__(self, code):
+        self.code = int(code.split(' ', 1)[0])
+        self.text = code.lower()
+
+    def __eq__(self, other):
+        if isinstance(other, int):
+            return self.code == other
+        return self.text == other.lower()
+
+    def raise_value_error(self):
+        raise ValueError('Cannot compare with string, Use integer instead for\
+                          all comparisons except equality')
+
+    def __gt__(self, other):
+        if isinstance(other, int):
+            return self.code > other
+        self.raise_value_error
+
+    def __ge__(self, other):
+        if isinstance(other, int):
+            return self.code >= other
+        self.raise_value_error
+
+    def __lt__(self, other):
+        if isinstance(other, int):
+            return self.code < other
+        self.raise_value_error
+
+    def __le__(self, other):
+        if isinstance(other, int):
+            return self.code <= other
+        self.raise_value_error
+

bddrest/practice/status_practice.py

@@ -0,0 +1,56 @@
+import re
+
+
+class Status:
+
+    def __init__(self, code):
+        self.code = int(code.split(' ', 1)[0])
+        self.text = code.lower()
+
+    def __eq__(self, other):
+        if isinstance(other, int):
+            return self.code == other
+        return self.text == other.lower()
+
+    def raise_value_error(self):
+        raise ValueError('Cannot compare with string, Use integer instead for\
+                          all comparisons except equality')
+
+    def __gt__(self, other):
+        if isinstance(other, int):
+            return self.code > other
+        self.raise_value_error
+
+    def __ge__(self, other):
+        if isinstance(other, int):
+            return self.code >= other
+        self.raise_value_error
+
+    def __lt__(self, other):
+        if isinstance(other, int):
+            return self.code < other
+         self.raise_value_error
+
+    def __le__(self, other):
+        if isinstance(other, int):
+            return self.code <= other
+         self.raise_value_error
+
+
+if __name__ == '__main__':
+    s = Status('200 OK')
+    assert s == '200 ok'
+    assert s == 200
+    assert s != 201
+    assert s != '200 OKOK'
+    assert s >= 100
+    assert s >= '100 Continue'
+    assert s > 100
+    assert s > '100 Continue'
+    assert s <= 300
+    assert s <= '300 Continue'
+    assert s < 300
+    assert s < '300 Continue'
+    assert s >= 200
+    assert s <= 200
+