Expression Fuzzer¶
Velox allows users to define UDFs (user-defined functions) and provides an Expression Fuzzer tool to test the engine and UDFs thoroughly. This tool is being used to test builtin Presto and Spark functions and have discovered numerous bugs caused by corner cases that are difficult to cover in unit tests.
The Expression Fuzzer tests the expression evaluation engine and UDFs by generating random expressions and evaluating these on random input vectors. Each generated expression may contain multiple sub-expressions and each input vector can have random and potentially nested encodings.
To ensure that evaluation engine and UDFs handle vector encodings correctly, the expression fuzzer evaluates each expression twice and asserts the results to be the same: using regular evaluation path and using simplified evaluation that flattens all input vectors before evaluating an expression.
How to integrate¶
To integrate with the Expression Fuzzer, create a test, register all scalar
functions supported by the engine, and call FuzzerRunner::run()
defined in
FuzzerRunner.h. See ExpressionFuzzerTest.cpp.
Functions with known bugs can be excluded from testing using a skip-list.
Custom Input Generators¶
Custom input generators for expression fuzzer are needed when:
A function requires input values to satisfy specific constraints (e.g., valid JSON or URL).
The default random input generation does not provide sufficient coverage of edge cases.
A custom input generator can be defined with the following steps.
Step 1: Define the generator class¶
The developer can implement a custom input generator class by
inheriting from ArgValuesGenerator
and overriding the generate()
method. An example is the JsonParseArgValuesGenerator class.
class JsonParseArgValuesGenerator : public ArgValuesGenerator {
public:
~JsonParseArgValuesGenerator() override = default;
std::vector<core::TypedExprPtr> generate(
const CallableSignature& signature,
const VectorFuzzer::Options& options,
FuzzerGenerator& rng,
ExpressionFuzzerState& state) override;
};
Step 2: Implement the generate() method¶
When implementing the generate()
method, the developer first call
populateInputTypesAndNames(signature, state)
to append argument types
of signature
and the corresponding input column names to state
.
Next, for every argument of signature
in order, if it requires a custom
input generator, the developer emplace back an std::shared_ptr of a subclass
of AbstractInputGenerator
into state.customInputGenerators_
. The
developer also emplace back a FieldAccessTypedExprPtr
created with this
argument type and the corresponding input column name into inputExpressions
.
Every subclass of AbstractInputGenerator
allows generating input values
satisfying certain constraint. E.g., RangeConstrainedGenerator
allows
generating values in a specified range; JsonInputGenerator
allows generating
valid JSON strings that represent a specified data type.
If the argument doesn’t require a custom input generator, the developer emplace
a nullptr back into both state.customInputGenerators_
and inputExpressions
.
Finally, this generate()
method returns inputExpressions
.
std::vector<core::TypedExprPtr> JsonParseArgValuesGenerator::generate(
const CallableSignature& signature,
const VectorFuzzer::Options& options,
FuzzerGenerator& rng,
ExpressionFuzzerState& state) {
VELOX_CHECK_EQ(signature.args.size(), 1);
populateInputTypesAndNames(signature, state);
const auto representedType = facebook::velox::randType(rng, 3);
const auto seed = rand<uint32_t>(rng);
const auto nullRatio = options.nullRatio;
state.customInputGenerators_.emplace_back(
std::make_shared<fuzzer::JsonInputGenerator>(
seed,
signature.args[0],
nullRatio,
fuzzer::getRandomInputGenerator(seed, representedType, nullRatio),
true));
std::vector<core::TypedExprPtr> inputExpressions{
signature.args.size(), nullptr};
inputExpressions[0] = std::make_shared<core::FieldAccessTypedExpr>(
signature.args[0], state.inputRowNames_.back());
return inputExpressions;
};
Step 3: Register the custom input generator¶
Once the custom input generator is implemented, the developer register it in
the main
function in ExpressionFuzzerTest.cpp as follows.
std::unordered_map<std::string, std::shared_ptr<ArgValuesGenerator>>
argValuesGenerators = {
{"json_parse", std::make_shared<JsonParseArgValuesGenerator>()},
{"function_name", std::make_shared<CustomInputGeneratorClassName>()},
...};
How to run¶
Expression Fuzzer supports a number of powerful command line arguments.
–-steps
: How many iterations to run. Each iteration generates and evaluates one expression or aggregation. Default is 10.–-duration_sec
: For how long to run in seconds. If both-–steps
and-–duration_sec
are specified, –duration_sec takes precedence.–-seed
: The seed to generate random expressions and input vectors with.–-v=1
: Verbose logging (from Google Logging Library).–-only
: A comma-separated list of functions to use in generated expressions.–-batch_size
: The size of input vectors to generate. Default is 100.--null_ratio
: Chance of adding a null constant to the plan, or null value in a vector (expressed as double from 0 to 1). Default is 0.1.--max_num_varargs
: The maximum number of variadic arguments fuzzer will generate for functions that accept variadic arguments. Fuzzer will generate up to max_num_varargs arguments for the variadic list in addition to the required arguments by the function. Default is 10.--retry_with_try
: Retry failed expressions by wrapping it using a try() statement. Default is false.--enable_variadic_signatures
: Enable testing of function signatures with variadic arguments. Default is false.--special_forms
: Enable testing of specified special forms, including and, or, cast, coalesce, if, and switch. Every fuzzer test specifies the enabled special forms of its own. velox_expression_fuzzer_test has all the aforementioned special forms enabled by default.--enable_dereference
: Enable testing of the field-reference from structs and row_constructor functions. Default is false.--velox_fuzzer_enable_complex_types
: Enable testing of function signatures with complex argument or return types. Default is false.--velox_fuzzer_enable_decimal_type
: Enable testing of function signatures with decimal argument or return type. Default is false.--lazy_vector_generation_ratio
: Specifies the probability with which columns in the input row vector will be selected to be wrapped in lazy encoding (expressed as double from 0 to 1). Default is 0.0.--velox_fuzzer_enable_column_reuse
: Enable generation of expressions where one input column can be used by multiple subexpressions. Default is false.--velox_fuzzer_enable_expression_reuse
: Enable generation of expressions that re-uses already generated subexpressions. Default is false.--assign_function_tickets
: Comma separated list of function names and their tickets in the format <function_name>=<tickets>. Every ticket represents an opportunity for a function to be chosen from a pool of candidates. By default, every function has one ticket, and the likelihood of a function being picked can be increased by allotting it more tickets. Note that in practice, increasing the number of tickets does not proportionally increase the likelihood of selection, as the selection process involves filtering the pool of candidates by a required return type so not all functions may compete against the same number of functions at every instance. Number of tickets must be a positive integer. Example: eq=3,floor=5.--max_expression_trees_per_step
: This sets an upper limit on the number of expression trees to generate per step. These trees would be executed in the same ExprSet and can re-use already generated columns and subexpressions (if re-use is enabled). Default is 1.--velox_fuzzer_max_level_of_nesting
: Max levels of expression nesting. Default is 10 and minimum is 1.
If you would like to run Expression Fuzzer with Presto as the source of truth, two command line arguments can be used to specify the url and timeout of Presto:
--presto_url
: Presto coordinator URI along with port.--req_timeout_ms
: Timeout in milliseconds for HTTP requests made to the reference DB, such as Presto.
If running from CLion IDE, add --logtostderr=1
to see the full output.
An example set of arguments to run the expression fuzzer with all features enabled is as follows:
--duration_sec 60
--enable_variadic_signatures
--lazy_vector_generation_ratio 0.2
--velox_fuzzer_enable_complex_types
--velox_fuzzer_enable_expression_reuse
--velox_fuzzer_enable_column_reuse
--retry_with_try
--enable_dereference
--special_forms="and,or,cast,coalesce,if,switch"
--max_expression_trees_per_step=2
--repro_persist_path=<a_valid_local_path>
--logtostderr=1
Expression fuzzer with Presto as the source of truth currently only supports a subset of features:
--duration_sec 60
--presto_url=http://127.0.0.1:8080
--req_timeout_ms 10000
--enable_variadic_signatures
--velox_fuzzer_enable_complex_types
--special_forms="cast,coalesce,if,switch"
--lazy_vector_generation_ratio 0.2
--velox_fuzzer_enable_column_reuse
--velox_fuzzer_enable_expression_reuse
--max_expression_trees_per_step 2
--logtostderr=1
How to reproduce failures¶
When Fuzzer test fails, a seed number and the evaluated expression are
printed to the log. An example is given below. Developers can use --seed
with this seed number to rerun the exact same expression with the same inputs,
and use a debugger to investigate the issue. For the example below, the command
to reproduce the error would be velox/expression/fuzzer/velox_expression_fuzzer_test --seed 1188545576
.
I0819 18:37:52.249965 1954756 ExpressionFuzzer.cpp:685] ==============================> Started iteration 38
(seed: 1188545576)
I0819 18:37:52.250263 1954756 ExpressionFuzzer.cpp:578]
Executing expression: in("c0",10 elements starting at 0 {120, 19, -71, null, 27, ...})
I0819 18:37:52.250350 1954756 ExpressionFuzzer.cpp:581] 1 vectors as input:
I0819 18:37:52.250401 1954756 ExpressionFuzzer.cpp:583] [FLAT TINYINT: 100 elements, 6 nulls]
E0819 18:37:52.252044 1954756 Exceptions.h:68] Line: velox/expression/tests/ExpressionFuzzer.cpp:153, Function:compareVectors, Expression: vec1->equalValueAt(vec2.get(), i, i)Different results at idx '78': 'null' vs. '1', Source: RUNTIME, ErrorCode: INVALID_STATE
terminate called after throwing an instance of 'facebook::velox::VeloxRuntimeError'
...
Note that changes to the set of all UDFs to test with invalidates this
reproduction, which can be affected by the skip function list, the --only
argument, or the base commit, etc. This is because the chosen UDFs in the
expression are determined by both the seed and the pool of all UDFs to choose
from. So make sure you use the same configuration when reproducing a failure.
Accurate on-disk reproduction¶
Sometimes developers may want to capture an issue and investigate later,
possibly by someone else using a different machine. Using --seed
is not
sufficient to accurately reproduce the failure in this scenario. This could be
cased by different behaviors of random generator on different platforms,
additions/removals of UDFs from the list, and etc. To have an accurate
reproduction of a fuzzer failure regardless of environments you can record the
input vector and expression to files and replay these later.
Run Fuzzer using
--seed
and--repro_persist_path
flags to save the input vector and expression to files in the specified directory. Add “–persist_and_run_once” if the issue is not an exception failure but a crash failure.Run Expression Runner using generated files.
--repro_persist_path <path/to/directory>
flag tells the Fuzzer to save the
input vector, initial result vector, expression SQL, and other relevant data to files in a new directory saved within
the specified directory. It also prints out the exact paths for these. Fuzzer uses VectorSaver
for storing vectors on disk while preserving encodings.
If an iteration crashes the process before data can be persisted, run the fuzzer with the seed used for that iteration and use the following flag:
--persist_and_run_once
Persist repro info before evaluation and only run one iteration.
This is to rerun with the seed number and persist repro info upon a crash failure.
Only effective if repro_persist_path is set.
ExpressionRunner needs at the very least a path to input vector and path to expression SQL to run. However, you might need more files to reproduce the issue. All of which will be present in the directory that the fuzzer test generated. You can directly point the ExpressionRunner to that directory using –fuzzer_repro_path where it will pick up all the files automatically or you can specify each explicitly using other startup flags. ExpressionRunner supports the following flags:
--fuzzer_repro_path
directory path where all input files (required to reproduce a failure) that are generated by the Fuzzer are expected to reside. ExpressionRunner will automatically pick up all the files from this folder unless they are explicitly specified via their respective startup flag.--input_path
path to input vector that was created by the Fuzzer--sql_path
path to expression SQL that was created by the Fuzzer--registry
function registry to use for evaluating expression. One of “presto” (default) or “spark”.--complex_constant_path
optional path to complex constants that aren’t accurately expressable in SQL (Array, Map, Structs, …). This is used with SQL file to reproduce the exact expression, not needed when the expression doesn’t contain complex constants.--input_row_metadata_path
optional path for the file stored on-disk which contains a struct containing input row metadata. This includes columns in the input row vector to be wrapped in a lazy vector and/or dictionary encoded. It may also contain a dictionary peel for columns requiring dictionary encoding. This is used when the failing test included input columns that were lazy vectors and/or had columns wrapped with a common dictionary wrap.--result_path
optional path to result vector that was created by the Fuzzer. Result vector is used to reproduce cases where Fuzzer passes dirty vectors to expression evaluation as a result buffer. This ensures that functions are implemented correctly, taking into consideration dirty result buffer.--mode
run mode. One of “verify”, “common” (default), “simplified”.verify
evaluates the expression using common and simplified paths and compares the results. This is identical to a fuzzer run.common
evaluates the expression using common path and prints the results to stdout.simplified
evaluates the expression using simplified path and prints the results to stdout.query
evaluate SQL query specified in –sql or –sql_path and print out results. If –input_path is specified, the query may reference it as table ‘t’.
--num_rows
optional number of rows to process in common and simplified modes. Default: 10. 0 means all rows. This flag is ignored in ‘verify’ mode.--store_result_path
optional directory path for storing the results of evaluating SQL expression or query in ‘common’, ‘simplified’ or ‘query’ modes.--findMinimalSubExpression
optional Whether to find minimum failing subexpression on result mismatch. Set to false by default.--useSeperatePoolForInput
optional If true (default), expression evaluator and input vectors use different memory pools. This helps trigger code-paths that can depend on vectors having different pools. For eg, when copying a flat string vector copies of the strings stored in the string buffers need to be created. If however, the pools were the same between the vectors then the buffers can simply be shared between them instead.
Example command:
velox/expression/tests:velox_expression_runner_test --input_path "/path/to/input" --sql_path "/path/to/sql" --result_path "/path/to/result"
To assist debugging workload, ExpressionRunner supports --sql
to specify
SQL expression on the command line. --sql
option can be used standalone to
evaluate constant expression or together with --input_path
to evaluate
expression on a vector. --sql
and --sql_path
flags are mutually
exclusive. If both are specified, --sql
is used while --sql_path
is
ignored. --sql
option allow to specify multiple comma-separated SQL
expressions.
$ velox/expression/tests:velox_expression_runner_test --sql "pow(2, 3), ceil(1.3)"
I1101 11:32:51.955689 2306506 ExpressionRunner.cpp:127] Evaluating SQL expression(s): pow(2, 3), ceil(1.3)
Result: ROW<_col0:DOUBLE,_col1:DOUBLE>
8 | 2
$ velox/expression/tests:velox_expression_runner_test --sql "pow(2, 3)"
Evaluating SQL expression(s): pow(2, 3)
Result: ROW<_col0:DOUBLE>
8
$ velox/expression/tests:velox_expression_runner_test --sql "array_sort(array[3,6,1,null,2])"
Building: finished in 0.3 sec (100%) 817/3213 jobs, 0/3213 updated
Evaluating SQL expression(s): array_sort(array[3,6,1,null,2])
Result: ROW<_col0:ARRAY<INTEGER>>
[1,2,3,6,null]
$ velox/expression/tests:velox_expression_runner_test --sql "array_sort(array[3,6,1,null,2]), filter(array[1, 2, 3, 4], x -> (x % 2 == 0))"
Evaluating SQL expression(s): array_sort(array[3,6,1,null,2]), filter(array[1, 2, 3, 4], x -> (x % 2 == 0))
Result: ROW<_col0:ARRAY<INTEGER>,_col1:ARRAY<INTEGER>>
[1,2,3,6,null] | [2,4]