text: -- Currently this tests polymorphic aggregates and indirectly does some
  -- testing of polymorphic SQL functions.  It ought to be extended.
  
  
  -- In GPDB, rows can be returned from segments in any order. Normally, we
  -- mask differences in result set order in regression tests with gpdiff.pl,
  -- but the aggregates in this test file result in arrays that have elements
  -- in random order. To fix, we force them in order with this function.
  create or replace function array_sort(x anyarray) returns anyarray as $$
    select array_agg(x) from (select unnest($1) AS x order by x) as _;
  $$ language sql;
  
  -- GPDB: the line right after Legend has been intentionally changed to not look
  -- like a psql result set header, like it does in upstream. atmsort.pl got
  -- confused by it and treated the whole comment as a result set, and re-ordered
  -- it.
  
  -- Legend:
  -- ---------
  -- A = type is ANY
  -- P = type is polymorphic
  -- N = type is non-polymorphic
  -- B = aggregate base type
  -- S = aggregate state type
  -- R = aggregate return type
  -- 1 = arg1 of a function
  -- 2 = arg2 of a function
  -- ag = aggregate
  -- tf = trans (state) function
  -- ff = final function
  -- rt = return type of a function
  -- -> = implies
  -- => = allowed
  -- !> = not allowed
  -- E  = exists
  -- NE = not-exists
  -- 
  -- Possible states:
  -- ----------------
  -- B = (A || P || N)
  --   when (B = A) -> (tf2 = NE)
  -- S = (P || N)
  -- ff = (E || NE)
  -- tf1 = (P || N)
  -- tf2 = (NE || P || N)
  -- R = (P || N)
  
  -- create functions for use as tf and ff with the needed combinations of
  -- argument polymorphism, but within the constraints of valid aggregate
  -- functions, i.e. tf arg1 and tf return type must match
  
  -- polymorphic single arg transfn
  CREATE FUNCTION stfp(anyarray) RETURNS anyarray AS
  'select $1' LANGUAGE SQL;
  -- non-polymorphic single arg transfn
  CREATE FUNCTION stfnp(int[]) RETURNS int[] AS
  'select $1' LANGUAGE SQL;
  
  -- dual polymorphic transfn
  CREATE FUNCTION tfp(anyarray,anyelement) RETURNS anyarray AS
  'select $1 || $2' LANGUAGE SQL;
  -- dual non-polymorphic transfn
  CREATE FUNCTION tfnp(int[],int) RETURNS int[] AS
  'select $1 || $2' LANGUAGE SQL;
  
  -- arg1 only polymorphic transfn
  CREATE FUNCTION tf1p(anyarray,int) RETURNS anyarray AS
  'select $1' LANGUAGE SQL;
  -- arg2 only polymorphic transfn
  CREATE FUNCTION tf2p(int[],anyelement) RETURNS int[] AS
  'select $1' LANGUAGE SQL;
  
  -- multi-arg polymorphic
  CREATE FUNCTION sum3(anyelement,anyelement,anyelement) returns anyelement AS
  'select $1+$2+$3' language sql strict;
  
  -- finalfn polymorphic
  CREATE FUNCTION ffp(anyarray) RETURNS anyarray AS
  'select $1' LANGUAGE SQL;
  -- finalfn non-polymorphic
  CREATE FUNCTION ffnp(int[]) returns int[] as
  'select $1' LANGUAGE SQL;
  
  -- Try to cover all the possible states:
  -- 
  -- Note: in Cases 1 & 2, we are trying to return P. Therefore, if the transfn
  -- is stfnp, tfnp, or tf2p, we must use ffp as finalfn, because stfnp, tfnp,
  -- and tf2p do not return P. Conversely, in Cases 3 & 4, we are trying to
  -- return N. Therefore, if the transfn is stfp, tfp, or tf1p, we must use ffnp
  -- as finalfn, because stfp, tfp, and tf1p do not return N.
  --
  --     Case1 (R = P) && (B = A)
  --     ------------------------
  --     S    tf1
  --     -------
  --     N    N
  -- should CREATE
  CREATE AGGREGATE myaggp01a(*) (SFUNC = stfnp, STYPE = int4[],
    FINALFUNC = ffp, INITCOND = '{}');
  
  --     P    N
  -- should ERROR: stfnp(anyarray) not matched by stfnp(int[])
  CREATE AGGREGATE myaggp02a(*) (SFUNC = stfnp, STYPE = anyarray,
    FINALFUNC = ffp, INITCOND = '{}');
  
  --     N    P
  -- should CREATE
  CREATE AGGREGATE myaggp03a(*) (SFUNC = stfp, STYPE = int4[],
    FINALFUNC = ffp, INITCOND = '{}');
  CREATE AGGREGATE myaggp03b(*) (SFUNC = stfp, STYPE = int4[],
    INITCOND = '{}');
  
  --     P    P
  -- should ERROR: we have no way to resolve S
  CREATE AGGREGATE myaggp04a(*) (SFUNC = stfp, STYPE = anyarray,
    FINALFUNC = ffp, INITCOND = '{}');
  CREATE AGGREGATE myaggp04b(*) (SFUNC = stfp, STYPE = anyarray,
    INITCOND = '{}');
  
  
  --    Case2 (R = P) && ((B = P) || (B = N))
  --    -------------------------------------
  --    S    tf1      B    tf2
  --    -----------------------
  --    N    N        N    N
  -- should CREATE
  CREATE AGGREGATE myaggp05a(BASETYPE = int, SFUNC = tfnp, STYPE = int[],
    FINALFUNC = ffp, INITCOND = '{}');
  
  --    N    N        N    P
  -- should CREATE
  CREATE AGGREGATE myaggp06a(BASETYPE = int, SFUNC = tf2p, STYPE = int[],
    FINALFUNC = ffp, INITCOND = '{}');
  
  --    N    N        P    N
  -- should ERROR: tfnp(int[], anyelement) not matched by tfnp(int[], int)
  CREATE AGGREGATE myaggp07a(BASETYPE = anyelement, SFUNC = tfnp, STYPE = int[],
    FINALFUNC = ffp, INITCOND = '{}');
  
  --    N    N        P    P
  -- should CREATE
  CREATE AGGREGATE myaggp08a(BASETYPE = anyelement, SFUNC = tf2p, STYPE = int[],
    FINALFUNC = ffp, INITCOND = '{}');
  
  --    N    P        N    N
  -- should CREATE
  CREATE AGGREGATE myaggp09a(BASETYPE = int, SFUNC = tf1p, STYPE = int[],
    FINALFUNC = ffp, INITCOND = '{}');
  CREATE AGGREGATE myaggp09b(BASETYPE = int, SFUNC = tf1p, STYPE = int[],
    INITCOND = '{}');
  
  --    N    P        N    P
  -- should CREATE
  CREATE AGGREGATE myaggp10a(BASETYPE = int, SFUNC = tfp, STYPE = int[],
    FINALFUNC = ffp, INITCOND = '{}');
  CREATE AGGREGATE myaggp10b(BASETYPE = int, SFUNC = tfp, STYPE = int[],
    INITCOND = '{}');
  
  --    N    P        P    N
  -- should ERROR: tf1p(int[],anyelement) not matched by tf1p(anyarray,int)
  CREATE AGGREGATE myaggp11a(BASETYPE = anyelement, SFUNC = tf1p, STYPE = int[],
    FINALFUNC = ffp, INITCOND = '{}');
  CREATE AGGREGATE myaggp11b(BASETYPE = anyelement, SFUNC = tf1p, STYPE = int[],
    INITCOND = '{}');
  
  --    N    P        P    P
  -- should ERROR: tfp(int[],anyelement) not matched by tfp(anyarray,anyelement)
  CREATE AGGREGATE myaggp12a(BASETYPE = anyelement, SFUNC = tfp, STYPE = int[],
    FINALFUNC = ffp, INITCOND = '{}');
  CREATE AGGREGATE myaggp12b(BASETYPE = anyelement, SFUNC = tfp, STYPE = int[],
    INITCOND = '{}');
  
  --    P    N        N    N
  -- should ERROR: tfnp(anyarray, int) not matched by tfnp(int[],int)
  CREATE AGGREGATE myaggp13a(BASETYPE = int, SFUNC = tfnp, STYPE = anyarray,
    FINALFUNC = ffp, INITCOND = '{}');
  
  --    P    N        N    P
  -- should ERROR: tf2p(anyarray, int) not matched by tf2p(int[],anyelement)
  CREATE AGGREGATE myaggp14a(BASETYPE = int, SFUNC = tf2p, STYPE = anyarray,
    FINALFUNC = ffp, INITCOND = '{}');
  
  --    P    N        P    N
  -- should ERROR: tfnp(anyarray, anyelement) not matched by tfnp(int[],int)
  CREATE AGGREGATE myaggp15a(BASETYPE = anyelement, SFUNC = tfnp,
    STYPE = anyarray, FINALFUNC = ffp, INITCOND = '{}');
  
  --    P    N        P    P
  -- should ERROR: tf2p(anyarray, anyelement) not matched by tf2p(int[],anyelement)
  CREATE AGGREGATE myaggp16a(BASETYPE = anyelement, SFUNC = tf2p,
    STYPE = anyarray, FINALFUNC = ffp, INITCOND = '{}');
  
  --    P    P        N    N
  -- should ERROR: we have no way to resolve S
  CREATE AGGREGATE myaggp17a(BASETYPE = int, SFUNC = tf1p, STYPE = anyarray,
    FINALFUNC = ffp, INITCOND = '{}');
  CREATE AGGREGATE myaggp17b(BASETYPE = int, SFUNC = tf1p, STYPE = anyarray,
    INITCOND = '{}');
  
  --    P    P        N    P
  -- should ERROR: tfp(anyarray, int) not matched by tfp(anyarray, anyelement)
  CREATE AGGREGATE myaggp18a(BASETYPE = int, SFUNC = tfp, STYPE = anyarray,
    FINALFUNC = ffp, INITCOND = '{}');
  CREATE AGGREGATE myaggp18b(BASETYPE = int, SFUNC = tfp, STYPE = anyarray,
    INITCOND = '{}');
  
  --    P    P        P    N
  -- should ERROR: tf1p(anyarray, anyelement) not matched by tf1p(anyarray, int)
  CREATE AGGREGATE myaggp19a(BASETYPE = anyelement, SFUNC = tf1p,
    STYPE = anyarray, FINALFUNC = ffp, INITCOND = '{}');
  CREATE AGGREGATE myaggp19b(BASETYPE = anyelement, SFUNC = tf1p,
    STYPE = anyarray, INITCOND = '{}');
  
  --    P    P        P    P
  -- should CREATE
  CREATE AGGREGATE myaggp20a(BASETYPE = anyelement, SFUNC = tfp,
    STYPE = anyarray, FINALFUNC = ffp, INITCOND = '{}');
  CREATE AGGREGATE myaggp20b(BASETYPE = anyelement, SFUNC = tfp,
    STYPE = anyarray, INITCOND = '{}');
  
  --     Case3 (R = N) && (B = A)
  --     ------------------------
  --     S    tf1
  --     -------
  --     N    N
  -- should CREATE
  CREATE AGGREGATE myaggn01a(*) (SFUNC = stfnp, STYPE = int4[],
    FINALFUNC = ffnp, INITCOND = '{}');
  CREATE AGGREGATE myaggn01b(*) (SFUNC = stfnp, STYPE = int4[],
    INITCOND = '{}');
  
  --     P    N
  -- should ERROR: stfnp(anyarray) not matched by stfnp(int[])
  CREATE AGGREGATE myaggn02a(*) (SFUNC = stfnp, STYPE = anyarray,
    FINALFUNC = ffnp, INITCOND = '{}');
  CREATE AGGREGATE myaggn02b(*) (SFUNC = stfnp, STYPE = anyarray,
    INITCOND = '{}');
  
  --     N    P
  -- should CREATE
  CREATE AGGREGATE myaggn03a(*) (SFUNC = stfp, STYPE = int4[],
    FINALFUNC = ffnp, INITCOND = '{}');
  
  --     P    P
  -- should ERROR: ffnp(anyarray) not matched by ffnp(int[])
  CREATE AGGREGATE myaggn04a(*) (SFUNC = stfp, STYPE = anyarray,
    FINALFUNC = ffnp, INITCOND = '{}');
  
  
  --    Case4 (R = N) && ((B = P) || (B = N))
  --    -------------------------------------
  --    S    tf1      B    tf2
  --    -----------------------
  --    N    N        N    N
  -- should CREATE
  CREATE AGGREGATE myaggn05a(BASETYPE = int, SFUNC = tfnp, STYPE = int[],
    FINALFUNC = ffnp, INITCOND = '{}');
  CREATE AGGREGATE myaggn05b(BASETYPE = int, SFUNC = tfnp, STYPE = int[],
    INITCOND = '{}');
  
  --    N    N        N    P
  -- should CREATE
  CREATE AGGREGATE myaggn06a(BASETYPE = int, SFUNC = tf2p, STYPE = int[],
    FINALFUNC = ffnp, INITCOND = '{}');
  CREATE AGGREGATE myaggn06b(BASETYPE = int, SFUNC = tf2p, STYPE = int[],
    INITCOND = '{}');
  
  --    N    N        P    N
  -- should ERROR: tfnp(int[], anyelement) not matched by tfnp(int[], int)
  CREATE AGGREGATE myaggn07a(BASETYPE = anyelement, SFUNC = tfnp, STYPE = int[],
    FINALFUNC = ffnp, INITCOND = '{}');
  CREATE AGGREGATE myaggn07b(BASETYPE = anyelement, SFUNC = tfnp, STYPE = int[],
    INITCOND = '{}');
  
  --    N    N        P    P
  -- should CREATE
  CREATE AGGREGATE myaggn08a(BASETYPE = anyelement, SFUNC = tf2p, STYPE = int[],
    FINALFUNC = ffnp, INITCOND = '{}');
  CREATE AGGREGATE myaggn08b(BASETYPE = anyelement, SFUNC = tf2p, STYPE = int[],
    INITCOND = '{}');
  
  --    N    P        N    N
  -- should CREATE
  CREATE AGGREGATE myaggn09a(BASETYPE = int, SFUNC = tf1p, STYPE = int[],
    FINALFUNC = ffnp, INITCOND = '{}');
  
  --    N    P        N    P
  -- should CREATE
  CREATE AGGREGATE myaggn10a(BASETYPE = int, SFUNC = tfp, STYPE = int[],
    FINALFUNC = ffnp, INITCOND = '{}');
  
  --    N    P        P    N
  -- should ERROR: tf1p(int[],anyelement) not matched by tf1p(anyarray,int)
  CREATE AGGREGATE myaggn11a(BASETYPE = anyelement, SFUNC = tf1p, STYPE = int[],
    FINALFUNC = ffnp, INITCOND = '{}');
  
  --    N    P        P    P
  -- should ERROR: tfp(int[],anyelement) not matched by tfp(anyarray,anyelement)
  CREATE AGGREGATE myaggn12a(BASETYPE = anyelement, SFUNC = tfp, STYPE = int[],
    FINALFUNC = ffnp, INITCOND = '{}');
  
  --    P    N        N    N
  -- should ERROR: tfnp(anyarray, int) not matched by tfnp(int[],int)
  CREATE AGGREGATE myaggn13a(BASETYPE = int, SFUNC = tfnp, STYPE = anyarray,
    FINALFUNC = ffnp, INITCOND = '{}');
  CREATE AGGREGATE myaggn13b(BASETYPE = int, SFUNC = tfnp, STYPE = anyarray,
    INITCOND = '{}');
  
  --    P    N        N    P
  -- should ERROR: tf2p(anyarray, int) not matched by tf2p(int[],anyelement)
  CREATE AGGREGATE myaggn14a(BASETYPE = int, SFUNC = tf2p, STYPE = anyarray,
    FINALFUNC = ffnp, INITCOND = '{}');
  CREATE AGGREGATE myaggn14b(BASETYPE = int, SFUNC = tf2p, STYPE = anyarray,
    INITCOND = '{}');
  
  --    P    N        P    N
  -- should ERROR: tfnp(anyarray, anyelement) not matched by tfnp(int[],int)
  CREATE AGGREGATE myaggn15a(BASETYPE = anyelement, SFUNC = tfnp,
    STYPE = anyarray, FINALFUNC = ffnp, INITCOND = '{}');
  CREATE AGGREGATE myaggn15b(BASETYPE = anyelement, SFUNC = tfnp,
    STYPE = anyarray, INITCOND = '{}');
  
  --    P    N        P    P
  -- should ERROR: tf2p(anyarray, anyelement) not matched by tf2p(int[],anyelement)
  CREATE AGGREGATE myaggn16a(BASETYPE = anyelement, SFUNC = tf2p,
    STYPE = anyarray, FINALFUNC = ffnp, INITCOND = '{}');
  CREATE AGGREGATE myaggn16b(BASETYPE = anyelement, SFUNC = tf2p,
    STYPE = anyarray, INITCOND = '{}');
  
  --    P    P        N    N
  -- should ERROR: ffnp(anyarray) not matched by ffnp(int[])
  CREATE AGGREGATE myaggn17a(BASETYPE = int, SFUNC = tf1p, STYPE = anyarray,
    FINALFUNC = ffnp, INITCOND = '{}');
  
  --    P    P        N    P
  -- should ERROR: tfp(anyarray, int) not matched by tfp(anyarray, anyelement)
  CREATE AGGREGATE myaggn18a(BASETYPE = int, SFUNC = tfp, STYPE = anyarray,
    FINALFUNC = ffnp, INITCOND = '{}');
  
  --    P    P        P    N
  -- should ERROR: tf1p(anyarray, anyelement) not matched by tf1p(anyarray, int)
  CREATE AGGREGATE myaggn19a(BASETYPE = anyelement, SFUNC = tf1p,
    STYPE = anyarray, FINALFUNC = ffnp, INITCOND = '{}');
  
  --    P    P        P    P
  -- should ERROR: ffnp(anyarray) not matched by ffnp(int[])
  CREATE AGGREGATE myaggn20a(BASETYPE = anyelement, SFUNC = tfp,
    STYPE = anyarray, FINALFUNC = ffnp, INITCOND = '{}');
  
  -- multi-arg polymorphic
  CREATE AGGREGATE mysum2(anyelement,anyelement) (SFUNC = sum3,
    STYPE = anyelement, INITCOND = '0');
  
  -- create test data for polymorphic aggregates
  create temp table t(f1 int, f2 int[], f3 text);
  insert into t values(1,array[1],'a');
  insert into t values(1,array[11],'b');
  insert into t values(1,array[111],'c');
  insert into t values(2,array[2],'a');
  insert into t values(2,array[22],'b');
  insert into t values(2,array[222],'c');
  insert into t values(3,array[3],'a');
  insert into t values(3,array[3],'b');
  
  -- test the successfully created polymorphic aggregates
  select f3, myaggp01a(*) from t group by f3;
  select f3, myaggp03a(*) from t group by f3;
  select f3, myaggp03b(*) from t group by f3;
  select f3, array_sort(myaggp05a(f1)) as myaggp05a from t group by f3;
  select f3, myaggp06a(f1) from t group by f3;
  select f3, myaggp08a(f1) from t group by f3;
  select f3, myaggp09a(f1) from t group by f3;
  select f3, myaggp09b(f1) from t group by f3;
  select f3, array_sort(myaggp10a(f1)) as myaggp10a from t group by f3;
  select f3, array_sort(myaggp10b(f1)) as myaggp10b from t group by f3;
  select f3, array_sort(myaggp20a(f1)) as myaggp20a from t group by f3;
  select f3, array_sort(myaggp20b(f1)) as myaggp20b from t group by f3;
  select f3, myaggn01a(*) from t group by f3;
  select f3, myaggn01b(*) from t group by f3;
  select f3, myaggn03a(*) from t group by f3;
  select f3, array_sort(myaggn05a(f1)) as myaggn05a from t group by f3;
  select f3, array_sort(myaggn05b(f1)) as myaggn05b from t group by f3;
  select f3, myaggn06a(f1) from t group by f3;
  select f3, myaggn06b(f1) from t group by f3;
  select f3, myaggn08a(f1) from t group by f3;
  select f3, myaggn08b(f1) from t group by f3;
  select f3, myaggn09a(f1) from t group by f3;
  select f3, array_sort(myaggn10a(f1)) as myaggn10a from t group by f3;
  select mysum2(f1, f1 + 1) from t;
  
  -- test inlining of polymorphic SQL functions
  create function bleat(int) returns int as $$
  begin
    raise notice 'bleat %', $1;
    return $1;
  end$$ language plpgsql;
  
  create function sql_if(bool, anyelement, anyelement) returns anyelement as $$
  select case when $1 then $2 else $3 end $$ language sql;
  
  -- create a table with two columns and insert all the rows into the same segment
  -- by having the same value for the distributed column for multiple rows.
  -- We need this to ensure that the NOTICE raised by bleat function gets returned
  -- in the same order.
  create table int4_tbl_new(f0 int, f1 int) distributed by(f0);
  insert into int4_tbl_new values(1, 0), (1, 123456), (1, -123456), (1, 2147483647), (1, -2147483647);
  
  -- Note this would fail with integer overflow, never mind wrong bleat() output,
  -- if the CASE expression were not successfully inlined
  select f1, sql_if(f1 > 0, bleat(f1), bleat(f1 + 1)) from int4_tbl_new;
  
  select q2, sql_if(q2 > 0, q2, q2 + 1) from int8_tbl;
  
  -- another kind of polymorphic aggregate
  
  create function add_group(grp anyarray, ad anyelement, size integer)
    returns anyarray
    as $$
  begin
    if grp is null then
      return array[ad];
    end if;
    if array_upper(grp, 1) < size then
      return grp || ad;
    end if;
    return grp;
  end;
  $$
    language plpgsql immutable;
  
  create aggregate build_group(anyelement, integer) (
    SFUNC = add_group,
    STYPE = anyarray
  );
  
  select build_group(q1,3) from (select q1 from int8_tbl order by q1) as t;
  
  -- this should fail because stype isn't compatible with arg
  create aggregate build_group(int8, integer) (
    SFUNC = add_group,
    STYPE = int2[]
  );
  
  -- but we can make a non-poly agg from a poly sfunc if types are OK
  create aggregate build_group(int8, integer) (
    SFUNC = add_group,
    STYPE = int8[]
  );
  