diff --git a/ate/utils.py b/ate/utils.py index 97a3e7a5..bdc921b5 100644 --- a/ate/utils.py +++ b/ate/utils.py @@ -196,6 +196,16 @@ def match_expected(value, expected, comparator="eq"): assert value != expected elif comparator in ["len_eq", "length_equal", "count_eq"]: assert len(value) == expected + elif comparator in ["len_gt", "count_gt", "length_greater_than", "count_greater_than"]: + assert len(value) > expected + elif comparator in ["len_ge", "count_ge", "length_greater_than_or_equals", \ + "count_greater_than_or_equals"]: + assert len(value) >= expected + elif comparator in ["len_lt", "count_lt", "length_less_than", "count_less_than"]: + assert len(value) < expected + elif comparator in ["len_le", "count_le", "length_less_than_or_equals", \ + "count_less_than_or_equals"]: + assert len(value) <= expected elif comparator in ["lt", "less_than"]: assert value < expected elif comparator in ["le", "less_than_or_equals"]: diff --git a/test/test_utils.py b/test/test_utils.py index 8e4cbe28..36e93097 100644 --- a/test/test_utils.py +++ b/test/test_utils.py @@ -188,6 +188,11 @@ class TestUtils(ApiServerUnittest): self.assertTrue(utils.match_expected(123, "123", "str_eq")) self.assertTrue(utils.match_expected(123, "123", "ne")) + self.assertTrue(utils.match_expected("123", 2, "len_gt")) + self.assertTrue(utils.match_expected("123", 3, "len_ge")) + self.assertTrue(utils.match_expected("123", 4, "len_lt")) + self.assertTrue(utils.match_expected("123", 3, "len_le")) + self.assertTrue(utils.match_expected(1, 2, "lt")) self.assertTrue(utils.match_expected(1, 1, "le")) self.assertTrue(utils.match_expected(2, 1, "gt"))