mirror of https://github.com/crytic/slither
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
43 lines
1.8 KiB
43 lines
1.8 KiB
from pathlib import Path
|
|
from solc_select import solc_select
|
|
from slither import Slither
|
|
from slither.core.cfg.node import NodeType
|
|
from slither.slithir.operations import Assignment
|
|
from slither.core.expressions import AssignmentOperation, TupleExpression
|
|
|
|
|
|
TEST_DATA_DIR = Path(__file__).resolve().parent / "test_data"
|
|
# pylint: disable=too-many-nested-blocks
|
|
def test_ternary_conversions() -> None:
|
|
"""This tests that true and false sons define the same number of variables that the father node declares"""
|
|
solc_select.switch_global_version("0.8.0", always_install=True)
|
|
slither = Slither(Path(TEST_DATA_DIR, "ternary_expressions.sol").as_posix())
|
|
for contract in slither.contracts:
|
|
for function in contract.functions:
|
|
vars_declared = 0
|
|
vars_assigned = 0
|
|
for node in function.nodes:
|
|
if node.type in [NodeType.IF, NodeType.IFLOOP]:
|
|
|
|
# Iterate over true and false son
|
|
for inner_node in node.sons:
|
|
# Count all variables declared
|
|
expression = inner_node.expression
|
|
if isinstance(expression, AssignmentOperation):
|
|
var_expr = expression.expression_left
|
|
# Only tuples declare more than one var
|
|
if isinstance(var_expr, TupleExpression):
|
|
vars_declared += len(var_expr.expressions)
|
|
else:
|
|
vars_declared += 1
|
|
|
|
for ir in inner_node.irs:
|
|
# Count all variables defined
|
|
if isinstance(ir, Assignment):
|
|
vars_assigned += 1
|
|
|
|
assert vars_declared == vars_assigned
|
|
|
|
|
|
if __name__ == "__main__":
|
|
test_ternary_conversions()
|
|
|