123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271 |
- # 2010 August 4
- #
- # 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.
- #
- #***********************************************************************
- #
- # The focus of this file is testing the CLI shell tool.
- # These tests are specific to the .import command.
- #
- # $Id: shell5.test,v 1.7 2009/07/17 16:54:48 shaneh Exp $
- #
- # Test plan:
- #
- # shell5-1.*: Basic tests specific to the ".import" command.
- #
- set testdir [file dirname $argv0]
- source $testdir/tester.tcl
- if {$tcl_platform(platform)=="windows"} {
- set CLI "sqlite3.exe"
- } else {
- set CLI "./sqlite3"
- }
- if {![file executable $CLI]} {
- finish_test
- return
- }
- db close
- forcedelete test.db test.db-journal test.db-wal
- #----------------------------------------------------------------------------
- # Test cases shell5-1.*: Basic handling of the .import and .separator commands.
- #
- # .import FILE TABLE Import data from FILE into TABLE
- do_test shell5-1.1.1 {
- catchcmd "test.db" ".import"
- } {1 {Error: unknown command or invalid arguments: "import". Enter ".help" for help}}
- do_test shell5-1.1.2 {
- catchcmd "test.db" ".import FOO"
- } {1 {Error: unknown command or invalid arguments: "import". Enter ".help" for help}}
- #do_test shell5-1.1.2 {
- # catchcmd "test.db" ".import FOO BAR"
- #} {1 {Error: no such table: BAR}}
- do_test shell5-1.1.3 {
- # too many arguments
- catchcmd "test.db" ".import FOO BAR BAD"
- } {1 {Error: unknown command or invalid arguments: "import". Enter ".help" for help}}
- # .separator STRING Change separator used by output mode and .import
- do_test shell1-1.2.1 {
- catchcmd "test.db" ".separator"
- } {1 {Error: unknown command or invalid arguments: "separator". Enter ".help" for help}}
- do_test shell1-1.2.2 {
- catchcmd "test.db" ".separator FOO"
- } {0 {}}
- do_test shell1-1.2.3 {
- # too many arguments
- catchcmd "test.db" ".separator FOO BAD"
- } {1 {Error: unknown command or invalid arguments: "separator". Enter ".help" for help}}
- # separator should default to "|"
- do_test shell5-1.3.1 {
- set res [catchcmd "test.db" ".show"]
- list [regexp {separator: \"\|\"} $res]
- } {1}
- # set separator to different value.
- # check that .show reports new value
- do_test shell5-1.3.2 {
- set res [catchcmd "test.db" {.separator ,
- .show}]
- list [regexp {separator: \",\"} $res]
- } {1}
- # import file doesn't exist
- do_test shell5-1.4.1 {
- forcedelete FOO
- set res [catchcmd "test.db" {CREATE TABLE t1(a, b);
- .import FOO t1}]
- } {1 {Error: cannot open "FOO"}}
- # empty import file
- do_test shell5-1.4.2 {
- forcedelete shell5.csv
- set in [open shell5.csv w]
- close $in
- set res [catchcmd "test.db" {.import shell5.csv t1
- SELECT COUNT(*) FROM t1;}]
- } {0 0}
- # import file with 1 row, 1 column (expecting 2 cols)
- do_test shell5-1.4.3 {
- set in [open shell5.csv w]
- puts $in "1"
- close $in
- set res [catchcmd "test.db" {.import shell5.csv t1}]
- } {1 {shell5.csv:1: expected 2 columns but found 1 - filling the rest with NULL}}
- # import file with 1 row, 3 columns (expecting 2 cols)
- do_test shell5-1.4.4 {
- set in [open shell5.csv w]
- puts $in "1|2|3"
- close $in
- set res [catchcmd "test.db" {.import shell5.csv t1}]
- } {1 {shell5.csv:1: expected 2 columns but found 3 - extras ignored}}
- # import file with 1 row, 2 columns
- do_test shell5-1.4.5 {
- set in [open shell5.csv w]
- puts $in "1|2"
- close $in
- set res [catchcmd "test.db" {DELETE FROM t1;
- .import shell5.csv t1
- SELECT COUNT(*) FROM t1;}]
- } {0 1}
- # import file with 2 rows, 2 columns
- # note we end up with 3 rows because of the 1 row
- # imported above.
- do_test shell5-1.4.6 {
- set in [open shell5.csv w]
- puts $in "2|3"
- puts $in "3|4"
- close $in
- set res [catchcmd "test.db" {.import shell5.csv t1
- SELECT COUNT(*) FROM t1;}]
- } {0 3}
- # import file with 1 row, 2 columns, using a comma
- do_test shell5-1.4.7 {
- set in [open shell5.csv w]
- puts $in "4,5"
- close $in
- set res [catchcmd "test.db" {.separator ,
- .import shell5.csv t1
- SELECT COUNT(*) FROM t1;}]
- } {0 4}
- # import file with 1 row, 2 columns, text data
- do_test shell5-1.4.8.1 {
- set in [open shell5.csv w]
- puts $in "5|Now is the time for all good men to come to the aid of their country."
- close $in
- set res [catchcmd "test.db" {.import shell5.csv t1
- SELECT COUNT(*) FROM t1;}]
- } {0 5}
- do_test shell5-1.4.8.2 {
- catchcmd "test.db" {SELECT b FROM t1 WHERE a='5';}
- } {0 {Now is the time for all good men to come to the aid of their country.}}
- # import file with 1 row, 2 columns, quoted text data
- # note that currently sqlite doesn't support quoted fields, and
- # imports the entire field, quotes and all.
- do_test shell5-1.4.9.1 {
- set in [open shell5.csv w]
- puts $in "6|'Now is the time for all good men to come to the aid of their country.'"
- close $in
- set res [catchcmd "test.db" {.import shell5.csv t1
- SELECT COUNT(*) FROM t1;}]
- } {0 6}
- do_test shell5-1.4.9.2 {
- catchcmd "test.db" {SELECT b FROM t1 WHERE a='6';}
- } {0 {'Now is the time for all good men to come to the aid of their country.'}}
- # import file with 1 row, 2 columns, quoted text data
- do_test shell5-1.4.10.1 {
- set in [open shell5.csv w]
- puts $in "7|\"Now is the time for all good men to come to the aid of their country.\""
- close $in
- set res [catchcmd "test.db" {.import shell5.csv t1
- SELECT COUNT(*) FROM t1;}]
- } {0 7}
- do_test shell5-1.4.10.2 {
- catchcmd "test.db" {SELECT b FROM t1 WHERE a='7';}
- } {0 {Now is the time for all good men to come to the aid of their country.}}
- # check importing very long field
- do_test shell5-1.5.1 {
- set str [string repeat X 999]
- set in [open shell5.csv w]
- puts $in "8|$str"
- close $in
- set res [catchcmd "test.db" {.import shell5.csv t1
- SELECT length(b) FROM t1 WHERE a='8';}]
- } {0 999}
- # try importing into a table with a large number of columns.
- # This is limited by SQLITE_MAX_VARIABLE_NUMBER, which defaults to 999.
- set cols 999
- do_test shell5-1.6.1 {
- set data {}
- for {set i 1} {$i<$cols} {incr i} {
- append data "c$i|"
- }
- append data "c$cols\n";
- for {set i 1} {$i<$cols} {incr i} {
- append data "$i|"
- }
- append data "$cols"
- set in [open shell5.csv w]
- puts $in $data
- close $in
- set res [catchcmd "test.db" {.import shell5.csv t2
- SELECT COUNT(*) FROM t2;}]
- } {0 1}
- # try importing a large number of rows
- set rows 9999
- do_test shell5-1.7.1 {
- set in [open shell5.csv w]
- puts $in a
- for {set i 1} {$i<=$rows} {incr i} {
- puts $in $i
- }
- close $in
- set res [catchcmd "test.db" {.mode csv
- .import shell5.csv t3
- SELECT COUNT(*) FROM t3;}]
- } [list 0 $rows]
- # Inport from a pipe. (Unix only, as it requires "awk")
- if {$tcl_platform(platform)=="unix"} {
- do_test shell5-1.8 {
- forcedelete test.db
- catchcmd test.db {.mode csv
- .import "|awk 'END{print \"x,y\";for(i=1;i<=5;i++){print i \",this is \" i}}'" t1
- SELECT * FROM t1;}
- } {0 {1,"this is 1"
- 2,"this is 2"
- 3,"this is 3"
- 4,"this is 4"
- 5,"this is 5"}}
- }
- # Import columns containing quoted strings
- do_test shell5-1.9 {
- set out [open shell5.csv w]
- fconfigure $out -translation lf
- puts $out {1,"",11}
- puts $out {2,"x",22}
- puts $out {3,"""",33}
- puts $out {4,"hello",44}
- puts $out "5,55,\"\"\r"
- puts $out {6,66,"x"}
- puts $out {7,77,""""}
- puts $out {8,88,"hello"}
- puts $out {"",9,99}
- puts $out {"x",10,110}
- puts $out {"""",11,121}
- puts $out {"hello",12,132}
- close $out
- forcedelete test.db
- catchcmd test.db {.mode csv
- CREATE TABLE t1(a,b,c);
- .import shell5.csv t1
- }
- sqlite3 db test.db
- db eval {SELECT *, '|' FROM t1 ORDER BY rowid}
- } {1 {} 11 | 2 x 22 | 3 {"} 33 | 4 hello 44 | 5 55 {} | 6 66 x | 7 77 {"} | 8 88 hello | {} 9 99 | x 10 110 | {"} 11 121 | hello 12 132 |}
- db close
- finish_test
|