| OLD | NEW |
| (Empty) |
| 1 # 2014 Dec 20 | |
| 2 # | |
| 3 # The author disclaims copyright to this source code. In place of | |
| 4 # a legal notice, here is a blessing: | |
| 5 # | |
| 6 # May you do good and not evil. | |
| 7 # May you find forgiveness for yourself and forgive others. | |
| 8 # May you share freely, never taking more than you give. | |
| 9 # | |
| 10 #*********************************************************************** | |
| 11 # | |
| 12 # Tests focusing on the auxiliary function APIs. | |
| 13 # | |
| 14 | |
| 15 source [file join [file dirname [info script]] fts5_common.tcl] | |
| 16 set testprefix fts5aux | |
| 17 | |
| 18 # If SQLITE_ENABLE_FTS5 is defined, omit this file. | |
| 19 ifcapable !fts5 { | |
| 20 finish_test | |
| 21 return | |
| 22 } | |
| 23 | |
| 24 proc inst {cmd i} { | |
| 25 $cmd xInst $i | |
| 26 } | |
| 27 sqlite3_fts5_create_function db inst inst | |
| 28 | |
| 29 proc colsize {cmd i} { | |
| 30 $cmd xColumnSize $i | |
| 31 } | |
| 32 sqlite3_fts5_create_function db colsize colsize | |
| 33 | |
| 34 proc totalsize {cmd i} { | |
| 35 $cmd xColumnTotalSize $i | |
| 36 } | |
| 37 sqlite3_fts5_create_function db totalsize totalsize | |
| 38 | |
| 39 do_execsql_test 1.0 { | |
| 40 CREATE VIRTUAL TABLE f1 USING fts5(a, b); | |
| 41 INSERT INTO f1 VALUES('one two', 'two one zero'); | |
| 42 INSERT INTO f1 VALUES('one one', 'one one one'); | |
| 43 } | |
| 44 | |
| 45 do_catchsql_test 1.1 { | |
| 46 SELECT inst(f1, -1) FROM f1 WHERE f1 MATCH 'two'; | |
| 47 } {1 SQLITE_RANGE} | |
| 48 do_catchsql_test 1.2 { | |
| 49 SELECT inst(f1, 0) FROM f1 WHERE f1 MATCH 'two'; | |
| 50 } {0 {{0 0 1}}} | |
| 51 do_catchsql_test 1.3 { | |
| 52 SELECT inst(f1, 1) FROM f1 WHERE f1 MATCH 'two'; | |
| 53 } {0 {{0 1 0}}} | |
| 54 do_catchsql_test 1.4 { | |
| 55 SELECT inst(f1, 2) FROM f1 WHERE f1 MATCH 'two'; | |
| 56 } {1 SQLITE_RANGE} | |
| 57 | |
| 58 do_catchsql_test 2.1 { | |
| 59 SELECT colsize(f1, 2) FROM f1 WHERE f1 MATCH 'two'; | |
| 60 } {1 SQLITE_RANGE} | |
| 61 do_execsql_test 2.2 { | |
| 62 SELECT colsize(f1, 0), colsize(f1, 1) FROM f1 WHERE f1 MATCH 'zero'; | |
| 63 } {2 3} | |
| 64 do_execsql_test 2.3 { | |
| 65 SELECT colsize(f1, -1) FROM f1 WHERE f1 MATCH 'zero'; | |
| 66 } {5} | |
| 67 | |
| 68 do_execsql_test 2.4.1 { | |
| 69 SELECT totalsize(f1, -1) FROM f1 WHERE f1 MATCH 'zero'; | |
| 70 } {10} | |
| 71 do_execsql_test 2.4.2 { | |
| 72 SELECT totalsize(f1, 0) FROM f1 WHERE f1 MATCH 'zero'; | |
| 73 } {4} | |
| 74 do_execsql_test 2.4.3 { | |
| 75 SELECT totalsize(f1, 1) FROM f1 WHERE f1 MATCH 'zero'; | |
| 76 } {6} | |
| 77 do_catchsql_test 2.4.4 { | |
| 78 SELECT totalsize(f1, 2) FROM f1 WHERE f1 MATCH 'zero'; | |
| 79 } {1 SQLITE_RANGE} | |
| 80 | |
| 81 #------------------------------------------------------------------------- | |
| 82 # Test the xSet and xGetAuxdata APIs with a NULL destructor. | |
| 83 # | |
| 84 proc prevrowid {add cmd} { | |
| 85 set res [$cmd xGetAuxdataInt 0] | |
| 86 set r [$cmd xRowid] | |
| 87 $cmd xSetAuxdataInt $r | |
| 88 return [expr $res + $add] | |
| 89 } | |
| 90 sqlite3_fts5_create_function db prevrowid [list prevrowid 0] | |
| 91 sqlite3_fts5_create_function db prevrowid1 [list prevrowid 1] | |
| 92 | |
| 93 do_execsql_test 3.0 { | |
| 94 CREATE VIRTUAL TABLE e5 USING fts5(x); | |
| 95 INSERT INTO e5 VALUES('a b c'); | |
| 96 INSERT INTO e5 VALUES('d e f'); | |
| 97 INSERT INTO e5 VALUES('a b c'); | |
| 98 INSERT INTO e5 VALUES('d e f'); | |
| 99 INSERT INTO e5 VALUES('a b c'); | |
| 100 } | |
| 101 | |
| 102 do_execsql_test 3.1 { | |
| 103 SELECT prevrowid(e5) || '+' || rowid FROM e5 WHERE e5 MATCH 'c' | |
| 104 } {0+1 1+3 3+5} | |
| 105 | |
| 106 do_execsql_test 3.2 { | |
| 107 SELECT prevrowid(e5) || '+' || prevrowid1(e5) || '+' || rowid | |
| 108 FROM e5 WHERE e5 MATCH 'e' | |
| 109 } {0+1+2 2+3+4} | |
| 110 | |
| 111 #------------------------------------------------------------------------- | |
| 112 # Test that if the xQueryPhrase callback returns other than SQLITE_OK, | |
| 113 # the query is abandoned. And that if it returns an error code other than | |
| 114 # SQLITE_DONE, the error is propagated back to the caller. | |
| 115 # | |
| 116 do_execsql_test 4.0 { | |
| 117 CREATE VIRTUAL TABLE e7 USING fts5(x); | |
| 118 INSERT INTO e7 VALUES('a x a'); | |
| 119 INSERT INTO e7 VALUES('b x b'); | |
| 120 INSERT INTO e7 VALUES('c x c'); | |
| 121 INSERT INTO e7 VALUES('d x d'); | |
| 122 INSERT INTO e7 VALUES('e x e'); | |
| 123 } | |
| 124 | |
| 125 proc xCallback {rowid code cmd} { | |
| 126 set r [$cmd xRowid] | |
| 127 lappend ::cb $r | |
| 128 if {$r==$rowid} { return $code } | |
| 129 return "" | |
| 130 } | |
| 131 | |
| 132 proc phrasequery {cmd code} { | |
| 133 set ::cb [list] | |
| 134 $cmd xQueryPhrase 1 [list xCallback [$cmd xRowid] $code] | |
| 135 set ::cb | |
| 136 } | |
| 137 | |
| 138 sqlite3_fts5_create_function db phrasequery phrasequery | |
| 139 | |
| 140 do_execsql_test 4.1 { | |
| 141 SELECT phrasequery(e7, 'SQLITE_OK') FROM e7 WHERE e7 MATCH 'c x' | |
| 142 } {{1 2 3 4 5}} | |
| 143 | |
| 144 do_execsql_test 4.2 { | |
| 145 SELECT phrasequery(e7, 'SQLITE_DONE') FROM e7 WHERE e7 MATCH 'c x' | |
| 146 } {{1 2 3}} | |
| 147 | |
| 148 do_catchsql_test 4.3 { | |
| 149 SELECT phrasequery(e7, 'SQLITE_ERROR') FROM e7 WHERE e7 MATCH 'c x' | |
| 150 } {1 SQLITE_ERROR} | |
| 151 | |
| 152 #------------------------------------------------------------------------- | |
| 153 # Auxiliary function calls with many cursors in the global cursor list. | |
| 154 # | |
| 155 do_execsql_test 5.0 { | |
| 156 CREATE VIRTUAL TABLE e9 USING fts5(y); | |
| 157 INSERT INTO e9(rowid, y) VALUES(1, 'i iii'); | |
| 158 INSERT INTO e9(rowid, y) VALUES(2, 'ii iv'); | |
| 159 INSERT INTO e9(rowid, y) VALUES(3, 'ii'); | |
| 160 INSERT INTO e9(rowid, y) VALUES(4, 'i iv'); | |
| 161 INSERT INTO e9(rowid, y) VALUES(5, 'iii'); | |
| 162 } | |
| 163 | |
| 164 proc my_rowid {cmd} { $cmd xRowid } | |
| 165 sqlite3_fts5_create_function db my_rowid my_rowid | |
| 166 | |
| 167 foreach {var q} { | |
| 168 s1 i | |
| 169 s2 ii | |
| 170 s3 iii | |
| 171 s4 iv | |
| 172 } { | |
| 173 set sql "SELECT my_rowid(e9) FROM e9 WHERE e9 MATCH '$q'" | |
| 174 set $var [sqlite3_prepare db $sql -1 dummy] | |
| 175 } | |
| 176 | |
| 177 do_test 5.1.1 { sqlite3_step $s1 ; sqlite3_column_int $s1 0 } 1 | |
| 178 do_test 5.1.2 { sqlite3_step $s2 ; sqlite3_column_int $s2 0 } 2 | |
| 179 do_test 5.1.3 { sqlite3_step $s3 ; sqlite3_column_int $s3 0 } 1 | |
| 180 do_test 5.1.4 { sqlite3_step $s4 ; sqlite3_column_int $s4 0 } 2 | |
| 181 | |
| 182 do_test 5.2.1 { sqlite3_step $s1 ; sqlite3_column_int $s1 0 } 4 | |
| 183 do_test 5.2.2 { sqlite3_step $s2 ; sqlite3_column_int $s2 0 } 3 | |
| 184 do_test 5.2.3 { sqlite3_step $s3 ; sqlite3_column_int $s3 0 } 5 | |
| 185 do_test 5.2.4 { sqlite3_step $s4 ; sqlite3_column_int $s4 0 } 4 | |
| 186 | |
| 187 sqlite3_finalize $s1 | |
| 188 sqlite3_finalize $s2 | |
| 189 sqlite3_finalize $s3 | |
| 190 sqlite3_finalize $s4 | |
| 191 | |
| 192 #------------------------------------------------------------------------- | |
| 193 # Passing an invalid first argument to an auxiliary function is detected. | |
| 194 # | |
| 195 do_execsql_test 6.0 { | |
| 196 CREATE VIRTUAL TABLE e11 USING fts5(y, z); | |
| 197 INSERT INTO e11(rowid, y, z) VALUES(1, 'a b', 45); | |
| 198 INSERT INTO e11(rowid, y, z) VALUES(2, 'b c', 46); | |
| 199 } | |
| 200 | |
| 201 do_catchsql_test 6.1 { | |
| 202 SELECT my_rowid(z) FROM e11 WHERE e11 MATCH 'b' | |
| 203 } {1 {no such cursor: 45}} | |
| 204 | |
| 205 do_catchsql_test 6.2 { | |
| 206 SELECT my_rowid(y) FROM e11 WHERE e11 MATCH 'b' | |
| 207 } {1 {no such cursor: 0}} | |
| 208 | |
| 209 #------------------------------------------------------------------------- | |
| 210 # Test passing an out-of-range phrase number to xPhraseSize (should | |
| 211 # return 0). | |
| 212 # | |
| 213 proc my_phrasesize {cmd iPhrase} { $cmd xPhraseSize $iPhrase } | |
| 214 sqlite3_fts5_create_function db my_phrasesize my_phrasesize | |
| 215 | |
| 216 do_execsql_test 7.1 { | |
| 217 CREATE VIRTUAL TABLE t1 USING fts5(a); | |
| 218 INSERT INTO t1 VALUES('a b c'); | |
| 219 } | |
| 220 do_execsql_test 7.2 { | |
| 221 SELECT | |
| 222 my_phrasesize(t1, -1), | |
| 223 my_phrasesize(t1, 0), | |
| 224 my_phrasesize(t1, 1), | |
| 225 my_phrasesize(t1, 2) | |
| 226 FROM t1 WHERE t1 MATCH 'a OR b+c' | |
| 227 } {0 1 2 0} | |
| 228 | |
| 229 #------------------------------------------------------------------------- | |
| 230 # | |
| 231 do_execsql_test 8.0 { | |
| 232 CREATE VIRTUAL TABLE x1 USING fts5(a); | |
| 233 } | |
| 234 | |
| 235 foreach {tn lRow res} { | |
| 236 4 {"a a a" "b" "a d"} {"[a] [a] [a]" "[a] d"} | |
| 237 1 {"b d" "a b"} {"[b] [d]" "[a] b"} | |
| 238 2 {"d b" "a d"} {"[d] [b]" "[a] d"} | |
| 239 3 {"a a d"} {"[a] [a] d"} | |
| 240 } { | |
| 241 execsql { DELETE FROM x1 } | |
| 242 foreach row $lRow { execsql { INSERT INTO x1 VALUES($row) } } | |
| 243 breakpoint | |
| 244 do_execsql_test 8.$tn { | |
| 245 SELECT highlight(x1, 0, '[', ']') FROM x1 WHERE x1 MATCH 'a OR (b AND d)'; | |
| 246 } $res | |
| 247 } | |
| 248 | |
| 249 finish_test | |
| 250 | |
| OLD | NEW |