v/vlib/orm/orm_sql_or_blocks_test.v

import os
import db.sqlite

struct User {
	id   i64    [primary; sql: serial]
	name string [unique]
}

const db_folder = os.join_path(os.vtmp_dir(), 'v', 'orm_sql')

const db_path = os.join_path(db_folder, 'sql_statement_or_blocks.db')

fn testsuite_begin() {
	os.mkdir_all(db_folder) or {}
}

fn testsuite_end() {
	os.rmdir_all(db_folder) or {}
}

fn test_ensure_db_exists_and_user_table_is_ok() {
	mut db := sqlite.connect(db_path)!
	assert true

	eprintln('> drop pre-existing User table...')
	db.exec('drop table if exists User')

	eprintln('> creating User table...')
	sql db {
		create table User
	} or { panic(err) }
	assert true
	db.close()!
}

fn test_sql_or_block_for_insert() {
	mut db := sqlite.connect(db_path)!
	user := User{1, 'bilbo'}

	eprintln('> inserting user 1 (first try)...')
	mut is_user_inserted := true

	sql db {
		insert user into User
	} or {
		println('user should have been inserted, but could not, err: ${err}')
		is_user_inserted = false
	}

	assert is_user_inserted

	eprintln('> inserting user 1 (second try)...')
	sql db {
		insert user into User
	} or {
		println('user could not be inserted, err: ${err}')
		is_user_inserted = false
	}

	assert !is_user_inserted
	eprintln('LINE: ${@LINE}')
	db.close()!
}

fn test_sql_or_block_for_select() {
	mut db := sqlite.connect(db_path)!

	eprintln('> selecting user with id 1...')
	mut users := sql db {
		select from User where id == 1
	} or {
		eprintln('could not select user, err: ${err}')
		[]User{}
	}
	eprintln('LINE: ${@LINE}')

	single := users.first()
	assert single.id == 1

	users = sql db {
		select from User where id == 0
	} or {
		eprintln('could not select user, err: ${err}')
		[]User{}
	}
	eprintln('LINE: ${@LINE}')

	assert users.len == 0
	eprintln('LINE: ${@LINE}')

	eprintln('> selecting users...')
	multiple := sql db {
		select from User
	} or {
		eprintln('could not users, err: ${err}')
		[]User{}
	}
	eprintln('LINE: ${@LINE}')

	assert multiple.len == 1
	eprintln('LINE: ${@LINE}')
	db.close()!
}

fn test_finish() {
	eprintln('done')
	assert true
}