Skip to content

Commit

Permalink
Extract AggregationFuzzerBase from AggregationFuzzer (facebookincubat…
Browse files Browse the repository at this point in the history
…or#7916)

Summary:
Pull Request resolved: facebookincubator#7916

Extract AggregationFuzzerBase from AggregationFuzzer. This is needed for building WindowFuzzer
that reuses common logic in AggregaitonFuzzerBase.

This is the first piece of facebookincubator#7754.

Differential Revision: https://internalfb.com/D51692940

fbshipit-source-id: 950e85c713cb1b1af9ca503a95545ae1fb81fc70
  • Loading branch information
kagamiori authored and facebook-github-bot committed Dec 13, 2023
1 parent eb8e863 commit d37874c
Show file tree
Hide file tree
Showing 27 changed files with 2,174 additions and 1,710 deletions.

Large diffs are not rendered by default.

46 changes: 46 additions & 0 deletions velox/exec/fuzzer/AggregationFuzzer.h
Original file line number Diff line number Diff line change
@@ -0,0 +1,46 @@
/*
* Copyright (c) Facebook, Inc. and its affiliates.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#pragma once

#include "velox/exec/Aggregate.h"
#include "velox/exec/fuzzer/AggregationFuzzerBase.h"
#include "velox/exec/fuzzer/ReferenceQueryRunner.h"
#include "velox/vector/fuzzer/VectorFuzzer.h"

namespace facebook::velox::exec::test {

/// Runs the aggregation fuzzer.
/// @param signatureMap Map of all aggregate function signatures.
/// @param seed Random seed - Pass the same seed for reproducibility.
/// @param orderDependentFunctions Map of functions that depend on order of
/// input.
/// @param planPath Path to persisted plan information. If this is
/// supplied, fuzzer will only verify the plans.
/// @param referenceQueryRunner Reference query runner for results
/// verification.
void aggregateFuzzer(
AggregateFunctionSignatureMap signatureMap,
size_t seed,
const std::unordered_map<std::string, std::shared_ptr<ResultVerifier>>&
orderDependentFunctions,
const std::unordered_map<std::string, std::shared_ptr<InputGenerator>>&
customInputGenerators,
VectorFuzzer::Options::TimestampPrecision timestampPrecision,
const std::unordered_map<std::string, std::string>& queryConfigs,
const std::optional<std::string>& planPath,
std::unique_ptr<ReferenceQueryRunner> referenceQueryRunner);

} // namespace facebook::velox::exec::test
Loading

0 comments on commit d37874c

Please sign in to comment.