165 lines
		
	
	
		
			4.2 KiB
		
	
	
	
		
			Factor
		
	
	
			
		
		
	
	
			165 lines
		
	
	
		
			4.2 KiB
		
	
	
	
		
			Factor
		
	
	
| USING: io io.files io.files.temp io.directories io.launcher
 | |
| kernel namespaces prettyprint tools.test db.sqlite db sequences
 | |
| continuations db.types db.tuples unicode.case accessors arrays
 | |
| sorting ;
 | |
| IN: db.sqlite.tests
 | |
| 
 | |
| : db-path ( -- path ) "test.db" temp-file ;
 | |
| : test.db ( -- sqlite-db ) db-path <sqlite-db> ;
 | |
| 
 | |
| [ ] [ [ db-path delete-file ] ignore-errors ] unit-test
 | |
| 
 | |
| [ ] [
 | |
|     test.db [
 | |
|         "create table person (name varchar(30), country varchar(30))" sql-command
 | |
|         "insert into person values('John', 'America')" sql-command
 | |
|         "insert into person values('Jane', 'New Zealand')" sql-command
 | |
|     ] with-db
 | |
| ] unit-test
 | |
| 
 | |
| 
 | |
| [ { { "John" "America" } { "Jane" "New Zealand" } } ] [
 | |
|     test.db [
 | |
|         "select * from person" sql-query
 | |
|     ] with-db
 | |
| ] unit-test
 | |
| 
 | |
| [ { { "1" "John" "America" } { "2" "Jane" "New Zealand" } } ]
 | |
| [ test.db [ "select rowid, * from person" sql-query ] with-db ] unit-test
 | |
| 
 | |
| [ ] [
 | |
|     test.db [
 | |
|         "insert into person(name, country) values('Jimmy', 'Canada')"
 | |
|         sql-command
 | |
|     ] with-db
 | |
| ] unit-test
 | |
| 
 | |
| [
 | |
|     {
 | |
|         { "1" "John" "America" }
 | |
|         { "2" "Jane" "New Zealand" }
 | |
|         { "3" "Jimmy" "Canada" }
 | |
|     }
 | |
| ] [ test.db [ "select rowid, * from person" sql-query ] with-db ] unit-test
 | |
| 
 | |
| [
 | |
|     test.db [
 | |
|         [
 | |
|             "insert into person(name, country) values('Jose', 'Mexico')" sql-command
 | |
|             "insert into person(name, country) values('Jose', 'Mexico')" sql-command
 | |
|             "oops" throw
 | |
|         ] with-transaction
 | |
|     ] with-db
 | |
| ] must-fail
 | |
| 
 | |
| [ 3 ] [
 | |
|     test.db [
 | |
|         "select * from person" sql-query length
 | |
|     ] with-db
 | |
| ] unit-test
 | |
| 
 | |
| [ ] [
 | |
|     test.db [
 | |
|         [
 | |
|             "insert into person(name, country) values('Jose', 'Mexico')"
 | |
|             sql-command
 | |
|             "insert into person(name, country) values('Jose', 'Mexico')"
 | |
|             sql-command
 | |
|         ] with-transaction
 | |
|     ] with-db
 | |
| ] unit-test
 | |
| 
 | |
| [ 5 ] [
 | |
|     test.db [
 | |
|         "select * from person" sql-query length
 | |
|     ] with-db
 | |
| ] unit-test
 | |
| 
 | |
| [ \ swap ensure-table ] must-fail
 | |
| 
 | |
| ! You don't need a primary key
 | |
| TUPLE: things one two ;
 | |
| 
 | |
| things "THINGS" {
 | |
|     { "one" "ONE" INTEGER +not-null+ }
 | |
|     { "two" "TWO" INTEGER +not-null+ }
 | |
| } define-persistent
 | |
| 
 | |
| [ { { 0 0 } { 0 1 } { 1 0 } { 1 1 } } ] [
 | |
|     test.db [
 | |
|        things create-table
 | |
|         0 0 things boa insert-tuple
 | |
|         0 1 things boa insert-tuple
 | |
|         1 1 things boa insert-tuple
 | |
|         1 0 things boa insert-tuple
 | |
|         f f things boa select-tuples
 | |
|         [ [ one>> ] [ two>> ] bi 2array ] map natural-sort
 | |
|        things drop-table
 | |
|     ] with-db
 | |
| ] unit-test
 | |
| 
 | |
| ! Tables can have different names than the name of the tuple
 | |
| TUPLE: foo slot ;
 | |
| C: <foo> foo
 | |
| foo "BAR" { { "slot" "SOMETHING" INTEGER +not-null+ } } define-persistent
 | |
| 
 | |
| TUPLE: hi bye try ;
 | |
| C: <hi> hi
 | |
| hi "HELLO" {
 | |
|     { "bye" "BUHBYE" INTEGER { +foreign-id+ foo "SOMETHING" } }
 | |
|     { "try" "RETHROW" INTEGER { +foreign-id+ foo "SOMETHING" } }
 | |
| } define-persistent
 | |
| 
 | |
| [ T{ foo { slot 1 } } T{ hi { bye 1 } { try 1 } } ] [
 | |
|     test.db [
 | |
|         foo create-table
 | |
|         hi create-table
 | |
|         1 <foo> insert-tuple
 | |
|         f <foo> select-tuple
 | |
|         1 1 <hi> insert-tuple
 | |
|         f f <hi> select-tuple
 | |
|         hi drop-table
 | |
|         foo drop-table
 | |
|     ] with-db
 | |
| ] unit-test
 | |
| 
 | |
| 
 | |
| ! Test SQLite triggers
 | |
| 
 | |
| TUPLE: show id ;
 | |
| TUPLE: user username data ;
 | |
| TUPLE: watch show user ;
 | |
| 
 | |
| user "USER" {
 | |
|     { "username" "USERNAME" TEXT +not-null+ +user-assigned-id+ }
 | |
|     { "data" "DATA" TEXT }
 | |
| } define-persistent
 | |
| 
 | |
| show "SHOW" {
 | |
|     { "id" "ID" +db-assigned-id+ }
 | |
| } define-persistent
 | |
| 
 | |
| watch "WATCH" {
 | |
|     { "user" "USER" TEXT +not-null+ +user-assigned-id+
 | |
|         { +foreign-id+ user "USERNAME" } }
 | |
|     { "show" "SHOW" BIG-INTEGER +not-null+ +user-assigned-id+
 | |
|         { +foreign-id+ show "ID" } }
 | |
| } define-persistent
 | |
|     
 | |
| [ T{ user { username "littledan" } { data "foo" } } ] [
 | |
|     test.db [
 | |
|         user create-table
 | |
|         show create-table
 | |
|         watch create-table
 | |
|         "littledan" "foo" user boa insert-tuple
 | |
|         "mark" "bar" user boa insert-tuple
 | |
|         show new insert-tuple
 | |
|         show new select-tuple
 | |
|         "littledan" f user boa select-tuple
 | |
|         [ id>> ] [ username>> ] bi*
 | |
|         watch boa insert-tuple
 | |
|         watch new select-tuple
 | |
|         user>> f user boa select-tuple
 | |
|     ] with-db
 | |
| ] unit-test
 |