Function TEST_F

bolt/exec/tests/AggregateCompanionAdapterTest.cpp:108–157

};

TEST_F(AggregateCompanionRegistryTest, basic) {
  std::vector<std::shared_ptr<AggregateFunctionSignature>> signatures{
      AggregateFunctionSignatureBuilder()
          .returnType("double")
          .intermediateType("array(double)")
          .argumentType("double")
          .build(),
      AggregateFunctionSignatureBuilder()
          .returnType("bigint")
          .intermediateType("array(bigint)")
          .argumentType("bigint")
          .build()};
  registerDummyAggregateFunction("aggregateFunc1", signatures);

  checkAggregateSignaturesCount("aggregateFunc1_partial", 2);
  checkAggregateTypeResolution(
      "aggregateFunc1_partial", {DOUBLE()}, ARRAY(DOUBLE()), ARRAY(DOUBLE()));
  checkAggregateTypeResolution(
      "aggregateFunc1_partial", {BIGINT()}, ARRAY(BIGINT()), ARRAY(BIGINT()));

  checkAggregateSignaturesCount("aggregateFunc1_merge", 2);
  checkAggregateTypeResolution(
      "aggregateFunc1_merge",
      {ARRAY(DOUBLE())},
      ARRAY(DOUBLE()),
      ARRAY(DOUBLE()));
  checkAggregateTypeResolution(
      "aggregateFunc1_merge",
      {ARRAY(BIGINT())},
      ARRAY(BIGINT()),
      ARRAY(BIGINT()));

  checkAggregateSignaturesCount("aggregateFunc1_merge_extract", 2);
  checkAggregateTypeResolution(
      "aggregateFunc1_merge_extract",
      {ARRAY(DOUBLE())},
      ARRAY(DOUBLE()),
      DOUBLE());
  checkAggregateTypeResolution(
      "aggregateFunc1_merge_extract",
      {ARRAY(BIGINT())},
      ARRAY(BIGINT()),
      BIGINT());

  checkScalarSignaturesCount("aggregateFunc1_extract", 2);
  checkScalarTypeResolution(
      "aggregateFunc1_extract", {ARRAY(DOUBLE())}, DOUBLE());
  checkScalarTypeResolution(
      "aggregateFunc1_extract", {ARRAY(BIGINT())}, BIGINT());
}

TEST_F(AggregateCompanionRegistryTest, extractFunctionNameWithSuffix) {
  std::vector<std::shared_ptr<AggregateFunctionSignature>> signatures{