0
0
mirror of https://github.com/sqlite/sqlite.git synced 2024-11-29 00:12:23 +01:00
sqlite/test/distinctagg.test
dan bfd6f1bcd5 Fix a use-after-free error that could occur when processing "SELECT aggregate(DISTINCT <expr>)..." queries.
FossilOrigin-Name: 0e4789860b81c31d3a6d1f9f8340042ce1d08a82bf6119c783fcab85180b1b63
2021-04-08 20:29:12 +00:00

218 lines
6.4 KiB
Plaintext

# 2005 September 11
#
# The author disclaims copyright to this source code. In place of
# a legal notice, here is a blessing:
#
# May you do good and not evil.
# May you find forgiveness for yourself and forgive others.
# May you share freely, never taking more than you give.
#
#***********************************************************************
# This file implements regression tests for SQLite library. The
# focus of this script is the DISTINCT modifier on aggregate functions.
#
# $Id: distinctagg.test,v 1.3 2009/02/09 13:19:28 drh Exp $
set testdir [file dirname $argv0]
source $testdir/tester.tcl
set testprefix distinctagg
do_test distinctagg-1.1 {
execsql {
CREATE TABLE t1(a,b,c);
INSERT INTO t1 VALUES(1,2,3);
INSERT INTO t1 VALUES(1,3,4);
INSERT INTO t1 VALUES(1,3,5);
SELECT count(distinct a),
count(distinct b),
count(distinct c),
count(all a) FROM t1;
}
} {1 2 3 3}
do_test distinctagg-1.2 {
execsql {
SELECT b, count(distinct c) FROM t1 GROUP BY b
}
} {2 1 3 2}
do_test distinctagg-1.3 {
execsql {
INSERT INTO t1 SELECT a+1, b+3, c+5 FROM t1;
INSERT INTO t1 SELECT a+2, b+6, c+10 FROM t1;
INSERT INTO t1 SELECT a+4, b+12, c+20 FROM t1;
SELECT count(*), count(distinct a), count(distinct b) FROM t1
}
} {24 8 16}
do_test distinctagg-1.4 {
execsql {
SELECT a, count(distinct c) FROM t1 GROUP BY a ORDER BY a
}
} {1 3 2 3 3 3 4 3 5 3 6 3 7 3 8 3}
do_test distinctagg-2.1 {
catchsql {
SELECT count(distinct) FROM t1;
}
} {1 {DISTINCT aggregates must have exactly one argument}}
do_test distinctagg-2.2 {
catchsql {
SELECT group_concat(distinct a,b) FROM t1;
}
} {1 {DISTINCT aggregates must have exactly one argument}}
#--------------------------------------------------------------------------
reset_db
do_execsql_test 3.0 {
CREATE TABLE t1(a, b, c);
CREATE TABLE t2(d, e, f);
INSERT INTO t1 VALUES (1, 1, 1);
INSERT INTO t1 VALUES (2, 2, 2);
INSERT INTO t1 VALUES (3, 3, 3);
INSERT INTO t1 VALUES (4, 1, 4);
INSERT INTO t1 VALUES (5, 2, 1);
INSERT INTO t1 VALUES (5, 3, 2);
INSERT INTO t1 VALUES (4, 1, 3);
INSERT INTO t1 VALUES (3, 2, 4);
INSERT INTO t1 VALUES (2, 3, 1);
INSERT INTO t1 VALUES (1, 1, 2);
INSERT INTO t2 VALUES('a', 'a', 'a');
INSERT INTO t2 VALUES('b', 'b', 'b');
INSERT INTO t2 VALUES('c', 'c', 'c');
CREATE INDEX t1a ON t1(a);
CREATE INDEX t1bc ON t1(b, c);
}
foreach {tn use_eph sql res} {
1 0 "SELECT count(DISTINCT a) FROM t1" 5
2 0 "SELECT count(DISTINCT b) FROM t1" 3
3 1 "SELECT count(DISTINCT c) FROM t1" 4
4 0 "SELECT count(DISTINCT c) FROM t1 WHERE b=3" 3
5 0 "SELECT count(DISTINCT rowid) FROM t1" 10
6 0 "SELECT count(DISTINCT a) FROM t1, t2" 5
7 0 "SELECT count(DISTINCT a) FROM t2, t1" 5
8 1 "SELECT count(DISTINCT a+b) FROM t1, t2, t2, t2" 6
9 0 "SELECT count(DISTINCT c) FROM t1 WHERE c=2" 1
10 1 "SELECT count(DISTINCT t1.rowid) FROM t1, t2" 10
} {
do_test 3.$tn.1 {
set prg [db eval "EXPLAIN $sql"]
set idx [lsearch $prg OpenEphemeral]
expr {$idx>=0}
} $use_eph
do_execsql_test 3.$tn.2 $sql $res
}
do_execsql_test 3.10 {
SELECT a, count(DISTINCT b) FROM t1 GROUP BY a;
} {
1 1 2 2 3 2 4 1 5 2
}
#--------------------------------------------------------------------------
reset_db
do_execsql_test 3.0 {
CREATE TABLE t1(a, b, c);
CREATE INDEX t1a ON t1(a);
CREATE INDEX t1bc ON t1(b, c);
INSERT INTO t1 VALUES(1, 'A', 1);
INSERT INTO t1 VALUES(1, 'A', 1);
INSERT INTO t1 VALUES(2, 'A', 2);
INSERT INTO t1 VALUES(2, 'A', 2);
INSERT INTO t1 VALUES(1, 'B', 1);
INSERT INTO t1 VALUES(2, 'B', 2);
INSERT INTO t1 VALUES(3, 'B', 3);
INSERT INTO t1 VALUES(NULL, 'B', NULL);
INSERT INTO t1 VALUES(NULL, 'C', NULL);
INSERT INTO t1 VALUES('d', 'D', 'd');
CREATE TABLE t2(d, e, f);
CREATE INDEX t2def ON t2(d, e, f);
INSERT INTO t2 VALUES(1, 1, 'a');
INSERT INTO t2 VALUES(1, 1, 'a');
INSERT INTO t2 VALUES(1, 2, 'a');
INSERT INTO t2 VALUES(1, 2, 'a');
INSERT INTO t2 VALUES(1, 2, 'b');
INSERT INTO t2 VALUES(1, 3, 'b');
INSERT INTO t2 VALUES(1, 3, 'a');
INSERT INTO t2 VALUES(1, 3, 'b');
INSERT INTO t2 VALUES(2, 3, 'x');
INSERT INTO t2 VALUES(2, 3, 'y');
INSERT INTO t2 VALUES(2, 3, 'z');
CREATE TABLE t3(x, y, z);
INSERT INTO t3 VALUES(1,1,1);
INSERT INTO t3 VALUES(2,2,2);
}
foreach {tn use_eph sql res} {
1 0 "SELECT count(DISTINCT c) FROM t1 GROUP BY b" {2 3 0 1}
2 1 "SELECT count(DISTINCT a) FROM t1 GROUP BY b" {2 3 0 1}
3 1 "SELECT count(DISTINCT a) FROM t1 GROUP BY b+c" {0 1 1 1 1}
4 0 "SELECT count(DISTINCT f) FROM t2 GROUP BY d, e" {1 2 2 3}
5 1 "SELECT count(DISTINCT f) FROM t2 GROUP BY d" {2 3}
6 0 "SELECT count(DISTINCT f) FROM t2 WHERE d IS 1 GROUP BY e" {1 2 2}
} {
do_test 4.$tn.1 {
set prg [db eval "EXPLAIN $sql"]
set idx [lsearch $prg OpenEphemeral]
expr {$idx>=0}
} $use_eph
do_execsql_test 4.$tn.2 $sql $res
}
set t3root [db one {SELECT rootpage FROM sqlite_schema WHERE name='t3'}]
foreach {tn use_t3 sql res} {
1 1 "SELECT count(*) FROM t3" 2
2 0 "SELECT count(*) FROM t1" 10
2 1 "SELECT count(DISTINCT a) FROM t1, t3" 4
3 1 "SELECT count(DISTINCT a) FROM t1 LEFT JOIN t3" 4
4 1 "SELECT count(DISTINCT a) FROM t1 LEFT JOIN t3 WHERE t3.x=1" 4
5 1 "SELECT count(DISTINCT a) FROM t1 LEFT JOIN t3 WHERE t3.x=0" 0
6 1 "SELECT count(DISTINCT a) FROM t1 LEFT JOIN t3 ON (t3.x=0)" 4
7 1 "SELECT count(DISTINCT x) FROM t1 LEFT JOIN t3" 2
8 1 "SELECT count(DISTINCT x) FROM t1 LEFT JOIN t3 WHERE t3.x=1" 1
9 1 "SELECT count(DISTINCT x) FROM t1 LEFT JOIN t3 WHERE t3.x=0" 0
10 1 "SELECT count(DISTINCT x) FROM t1 LEFT JOIN t3 ON (t3.x=0)" 0
} {
do_test 5.$tn.1 {
set bUse 0
db eval "EXPLAIN $sql" a {
if {$a(opcode)=="OpenRead" && $a(p2)==$t3root} {set bUse 1}
}
set bUse
} $use_t3
do_execsql_test 5.$tn.2 $sql $res
}
#-------------------------------------------------------------------------
reset_db
do_execsql_test 6.0 {
CREATE TABLE t1(a, b);
CREATE TABLE t2(c, d);
INSERT INTO t1 VALUES(123,456);
INSERT INTO t2 VALUES(123,456);
}
do_execsql_test 6.1 {
SELECT count(DISTINCT c) FROM t1 LEFT JOIN t2;
} {1}
do_execsql_test 7.0 {
CREATE TABLE v1 ( v2 UNIQUE, v3 AS( TYPEOF ( NULL ) ) UNIQUE );
SELECT COUNT ( DISTINCT TRUE ) FROM v1 GROUP BY likelihood ( v3 , 0.100000 );
}
finish_test