mirror of
https://github.com/github/codeql.git
synced 2025-12-18 09:43:15 +01:00
162 lines
2.3 KiB
Python
162 lines
2.3 KiB
Python
import sys
|
|
import os
|
|
|
|
sys.path.append(os.path.dirname(os.path.dirname((__file__))))
|
|
from testlib import *
|
|
|
|
arg = "source"
|
|
arg1 = "source1"
|
|
arg2 = "source2"
|
|
arg3 = "source3"
|
|
|
|
|
|
def SINK(x, expected=arg):
|
|
if x == expected:
|
|
print("OK")
|
|
else:
|
|
print("Unexpected flow", x)
|
|
|
|
|
|
def SINK1(x):
|
|
SINK(x, expected=arg1)
|
|
|
|
|
|
def SINK2(x):
|
|
SINK(x, expected=arg2)
|
|
|
|
|
|
def SINK3(x):
|
|
SINK(x, expected=arg3)
|
|
|
|
|
|
def argument_passing(
|
|
a,
|
|
b,
|
|
/,
|
|
c,
|
|
d="",
|
|
*,
|
|
e="",
|
|
f,
|
|
**g,
|
|
):
|
|
SINK1(a)
|
|
SINK2(b)
|
|
SINK3(c)
|
|
SINK(f)
|
|
|
|
|
|
@expects(4)
|
|
def test_argument_passing():
|
|
argument_passing(arg1, arg2, arg3, f=arg)
|
|
|
|
|
|
def with_pos_only(a, /, b):
|
|
SINK1(a)
|
|
SINK2(b)
|
|
|
|
|
|
@expects(6)
|
|
def test_pos_only():
|
|
with_pos_only(arg1, arg2)
|
|
with_pos_only(arg1, b=arg2)
|
|
with_pos_only(arg1, *(arg2,))
|
|
|
|
|
|
def with_multiple_kw_args(a, b, c):
|
|
SINK1(a)
|
|
SINK2(b)
|
|
SINK3(c)
|
|
|
|
|
|
@expects(9)
|
|
def test_multiple_kw_args():
|
|
with_multiple_kw_args(b=arg2, c=arg3, a=arg1)
|
|
with_multiple_kw_args(arg1, *(arg2,), arg3)
|
|
with_multiple_kw_args(arg1, **{"c": arg3}, b=arg2)
|
|
|
|
|
|
def with_default_arguments(a=arg1, b=arg2, c=arg3):
|
|
SINK1(a)
|
|
SINK2(b)
|
|
SINK3(c)
|
|
|
|
|
|
@expects(12)
|
|
def test_default_arguments():
|
|
with_default_arguments()
|
|
with_default_arguments(arg1)
|
|
with_default_arguments(b=arg2)
|
|
with_default_arguments(**{"c": arg3})
|
|
|
|
|
|
# Nested constructor pattern
|
|
def grab_foo_bar_baz(foo, **kwargs):
|
|
SINK1(foo)
|
|
grab_bar_baz(**kwargs)
|
|
|
|
|
|
def grab_bar_baz(bar, **kwargs):
|
|
SINK2(bar)
|
|
grab_baz(**kwargs)
|
|
|
|
|
|
def grab_baz(baz):
|
|
SINK3(baz)
|
|
|
|
|
|
@expects(3)
|
|
def test_grab():
|
|
grab_foo_bar_baz(baz=arg3, bar=arg2, foo=arg1)
|
|
|
|
|
|
# All combinations
|
|
def test_pos_pos():
|
|
def with_pos(a):
|
|
SINK1(a)
|
|
|
|
with_pos(arg1)
|
|
|
|
|
|
def test_pos_pos_only():
|
|
def with_pos_only(a, /):
|
|
SINK1(a)
|
|
|
|
with_pos_only(arg1)
|
|
|
|
|
|
def test_pos_star():
|
|
def with_star(*a):
|
|
if len(a) > 0:
|
|
SINK1(a[0])
|
|
|
|
with_star(arg1)
|
|
|
|
|
|
def test_pos_kw():
|
|
def with_kw(a=""):
|
|
SINK1(a)
|
|
|
|
with_kw(arg1)
|
|
|
|
|
|
def test_kw_pos():
|
|
def with_pos(a):
|
|
SINK1(a)
|
|
|
|
with_pos(a=arg1)
|
|
|
|
|
|
def test_kw_kw():
|
|
def with_kw(a=""):
|
|
SINK1(a)
|
|
|
|
with_kw(a=arg1)
|
|
|
|
|
|
def test_kw_doublestar():
|
|
def with_doublestar(**a):
|
|
SINK1(a["a"])
|
|
|
|
with_doublestar(a=arg1)
|