QL tests for Python queries and libraries.

python/ql/test/query-tests/Statements/asserts/assert.py

@@ -0,0 +1,106 @@
+import sys
+import six
+
+def _f():
+    assert (yield 3)
+    x = [ 1 ]
+    assert len(x)   #Call without side-effects
+    assert sys.exit(1) #Call with side-effects
+    expected_types = (Response, six.text_type, six.binary_type)
+    assert isinstance(obj, expected_types), \
+        "obj must be %s, not %s" % (
+            " or ".join(t.__name__ for t in expected_types),
+            type(obj).__name__)
+
+def assert_tuple(x, y):
+    assert ()
+    assert (x, y)
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+def error_assert_true(x):
+    if x:
+        assert True, "Bad"
+    else:
+        assert True
+
+def error_assert_false(x):
+    if x:
+        assert False, "Bad"
+
+def error_assert_zero(x):
+    if x:
+        assert 0, "Bad"
+
+def error_assert_one(x):
+    if x:
+        assert 1, "Bad"
+
+def error_assert_empty_string(x):
+    if x:
+        assert "", "Bad"
+
+def error_assert_nonempty_string(x):
+    if x:
+        assert "X", "Bad"
+    else:
+        assert "X"
+
+def ok_assert_false(x):
+    if x:
+        assert 0==1, "Ok"
+
+class TestCase:
+    pass
+
+class MyTest(TestCase):
+    def test_ok_assert_in_test(self, x):
+        if x:
+            assert False, "Ok"
+
+def ok_assert_in_final_branch3(x):
+    if foo(x):
+        pass
+    elif bar(x):
+        pass
+    elif quux(x):
+        pass
+    else:
+        assert False, "Ok"
+
+def ok_assert_in_final_branch2(x):
+    if foo(x):
+        pass
+    elif bar(x):
+        pass
+    else:
+        assert False, "Ok"
+
+def error_assert_in_final_branch1(x):
+    if foo(x):
+        pass
+    else:
+        assert False, "Error"
+
+def error_assert_in_intermediate_branch(x):
+    if foo(x):
+        pass
+    elif bar(x):
+        pass
+    elif quux(x):
+        assert False, "Error"
+    elif yks(x):
+        pass
+    else:
+        pass