v/vlib/orm/orm_test.v

284 lines
6.0 KiB
V
Raw Normal View History

2020-12-26 13:40:22 +01:00
// import os
// import pg
// import term
2020-06-16 12:14:22 +02:00
import sqlite
2020-06-07 15:44:33 +02:00
2020-06-17 04:05:13 +02:00
struct Module {
2020-12-26 13:40:22 +01:00
id int
name string
2020-06-28 21:10:49 +02:00
nr_downloads int
}
struct User {
2021-04-07 15:27:02 +02:00
id int [primary]
2020-12-26 13:40:22 +01:00
age int
name string
is_customer bool
2020-06-28 21:10:49 +02:00
skipped_string string [skip]
}
struct Foo {
age int
2019-08-02 06:57:35 +02:00
}
2020-06-16 12:14:22 +02:00
fn test_orm_sqlite() {
db := sqlite.connect(':memory:') or { panic(err) }
2020-12-26 13:40:22 +01:00
db.exec('drop table if exists User')
2021-04-07 15:27:02 +02:00
sql db {
create table User
}
name := 'Peter'
2020-06-17 04:05:13 +02:00
db.exec("insert into User (name, age) values ('Sam', 29)")
2020-06-17 14:18:32 +02:00
db.exec("insert into User (name, age) values ('Peter', 31)")
2020-06-23 13:59:00 +02:00
db.exec("insert into User (name, age, is_customer) values ('Kate', 30, 1)")
2021-02-01 21:44:09 +01:00
2021-04-08 02:13:25 +02:00
c := sql db {
select count from User where id != 1
}
assert c == 2
2020-06-17 00:59:33 +02:00
nr_all_users := sql db {
select count from User
2020-06-16 12:14:22 +02:00
}
2020-06-17 00:59:33 +02:00
assert nr_all_users == 3
println('nr_all_users=$nr_all_users')
//
nr_users1 := sql db {
select count from User where id == 1
}
assert nr_users1 == 1
println('nr_users1=$nr_users1')
//
nr_peters := sql db {
select count from User where id == 2 && name == 'Peter'
}
assert nr_peters == 1
println('nr_peters=$nr_peters')
//
nr_peters2 := sql db {
select count from User where id == 2 && name == name
2020-06-17 00:59:33 +02:00
}
assert nr_peters2 == 1
nr_peters3 := sql db {
select count from User where name == name
}
assert nr_peters3 == 1
peters := sql db {
2020-06-22 16:58:43 +02:00
select from User where name == name
}
assert peters.len == 1
assert peters[0].name == 'Peter'
2020-06-22 16:58:43 +02:00
one_peter := sql db {
select from User where name == name limit 1
}
assert one_peter.name == 'Peter'
assert one_peter.id == 2
2020-06-17 04:05:13 +02:00
//
user := sql db {
select from User where id == 1
}
println(user)
assert user.name == 'Sam'
assert user.id == 1
assert user.age == 29
2020-06-17 14:18:32 +02:00
//
users := sql db {
select from User where id > 0
}
println(users)
assert users.len == 3
assert users[0].name == 'Sam'
assert users[1].name == 'Peter'
assert users[1].age == 31
2020-06-17 14:29:44 +02:00
//
users2 := sql db {
select from User where id < 0
}
println(users2)
assert users2.len == 0
//
users3 := sql db {
select from User where age == 29 || age == 31
}
println(users3)
assert users3.len == 2
assert users3[0].age == 29
assert users3[1].age == 31
2020-06-19 16:43:32 +02:00
//
2020-12-26 13:40:22 +01:00
new_user := User{
name: 'New user'
age: 30
}
2020-06-19 16:43:32 +02:00
sql db {
insert new_user into User
}
2020-12-26 13:40:22 +01:00
// db.insert<User>(user2)
2020-06-19 16:43:32 +02:00
x := sql db {
select from User where id == 4
}
println(x)
assert x.age == 30
assert x.id == 4
assert x.name == 'New user'
2020-06-23 13:59:00 +02:00
//
kate := sql db {
select from User where id == 3
}
assert kate.is_customer == true
//
customer := sql db {
select from User where is_customer == true limit 1
}
assert customer.is_customer == true
assert customer.name == 'Kate'
2020-06-24 14:32:14 +02:00
//
sql db {
update User set age = 31 where name == 'Kate'
2020-06-24 14:32:14 +02:00
}
2020-06-25 17:12:32 +02:00
kate2 := sql db {
select from User where id == 3
}
assert kate2.age == 31
assert kate2.name == 'Kate'
2020-06-25 17:18:44 +02:00
//
sql db {
update User set age = 32, name = 'Kate N' where name == 'Kate'
}
mut kate3 := sql db {
select from User where id == 3
}
assert kate3.age == 32
assert kate3.name == 'Kate N'
//
/*
sql db {
update User set age = age + 1, name = 'Kate N' where name == 'Kate'
}
kate3 = sql db {
2020-06-25 17:18:44 +02:00
select from User where id == 3
}
println(kate3)
assert kate3.age == 32
assert kate3.name == 'Kate N'
*/
new_age := 33
sql db {
update User set age = new_age, name = 'Kate N' where id == 3
}
kate3 = sql db {
select from User where id == 3
}
assert kate3.age == 33
assert kate3.name == 'Kate N'
//
foo := Foo{34}
sql db {
update User set age = foo.age, name = 'Kate N' where id == 3
}
kate3 = sql db {
select from User where id == 3
}
assert kate3.age == 34
assert kate3.name == 'Kate N'
2020-06-26 20:38:17 +02:00
//
no_user := sql db {
select from User where id == 30
}
assert no_user.name == '' // TODO optional
assert no_user.age == 0
2020-06-27 16:19:12 +02:00
//
two_users := sql db {
select from User limit 2
}
assert two_users.len == 2
2020-06-27 16:25:25 +02:00
assert two_users[0].id == 1
//
y := sql db {
select from User limit 2 offset 1
}
assert y.len == 2
assert y[0].id == 2
2020-06-27 16:41:29 +02:00
//
offset_const := 2
z := sql db {
select from User limit 2 offset offset_const
}
assert z.len == 2
assert z[0].id == 3
2020-07-02 19:29:22 +02:00
oldest := sql db {
select from User order by age desc limit 1
2020-07-02 19:29:22 +02:00
}
assert oldest.age == 34
offs := 1
second_oldest := sql db {
select from User order by age desc limit 1 offset offs
}
assert second_oldest.age == 31
2020-07-07 16:55:32 +02:00
sql db {
delete from User where age == 34
}
updated_oldest := sql db {
select from User order by age desc limit 1
}
assert updated_oldest.age == 31
2021-02-01 21:44:09 +01:00
db.exec('insert into User (name, age) values (NULL, 31)')
null_user := sql db {
select from User where id == 5
}
assert null_user.name == ''
2020-06-16 12:14:22 +02:00
}
fn test_orm_pg() {
2020-12-26 13:40:22 +01:00
/*
dbname := os.getenv('VDB_NAME')
dbuser := os.getenv('VDB_USER')
if dbname == '' || dbuser == '' {
eprintln(term.red('NB: this test requires VDB_NAME and VDB_USER env variables to be set'))
return
}
db := pg.connect(dbname: dbname, user: dbuser) or { panic(err) }
2020-06-16 12:14:22 +02:00
_ = db
2020-06-17 00:59:33 +02:00
nr_modules := db.select count from modules
2020-06-07 15:44:33 +02:00
//nr_modules := db.select count from Modules where id == 1
nr_modules := db.select count from Modules where
name == 'Bob' && id == 1
println(nr_modules)
mod := db.select from Modules where id = 1 limit 1
println(mod)
mods := db.select from Modules limit 10
for mod in mods {
println(mod)
}
2020-12-26 13:40:22 +01:00
*/
/*
2020-06-07 15:44:33 +02:00
mod := db.retrieve<Module>(1)
2020-06-17 00:59:33 +02:00
mod := db.select from Module where id = 1
2020-06-07 15:44:33 +02:00
mod := db.update Module set name = name + '!' where id > 10
nr_modules := db.select count from Modules
where id > 1 && name == ''
println(nr_modules)
nr_modules := db.select count from modules
nr_modules := db.select from modules
nr_modules := db[:modules].select
2020-12-26 13:40:22 +01:00
*/
/*
2020-06-07 15:44:33 +02:00
mod := select from db.modules where id = 1 limit 1
println(mod.name)
top_mods := select from db.modules where nr_downloads > 1000 order by nr_downloads desc limit 10
top_mods := db.select from modules where nr_downloads > 1000 order by nr_downloads desc limit 10
top_mods := db.select<Module>(m => m.nr_downloads > 1000).order_by(m => m.nr_downloads).desc().limit(10)
names := select name from db.modules // []string
n := db.q_int('select count(*) from modules')
println(n)
2020-12-26 13:40:22 +01:00
*/
2020-06-07 15:44:33 +02:00
}