// vtest build: started_mysqld?
import db.mysql
import orm

fn test_mysql() {
	$if !network ? {
		eprintln('> Skipping test ${@FN}, since `-d network` is not passed.')
		eprintln('> This test requires a working mysql server running on localhost.')
		return
	}
	config := mysql.Config{
		host:     '127.0.0.1'
		port:     3306
		username: 'root'
		password: '12345678'
		dbname:   'mysql'
	}

	mut db := mysql.connect(config)!
	defer {
		db.close() or {}
	}

	assert db.validate()!

	mut conn := orm.TransactionalConnection(db)
	mut tx := orm.begin(mut conn)!
	tx.transaction[int](fn (mut tx orm.Tx) !int {
		return 1
	})!
	tx.commit()!

	mut response := db.exec('drop table if exists users')!
	assert response == []mysql.Row{}

	response = db.exec('create table if not exists users (
                        id INT PRIMARY KEY AUTO_INCREMENT,
                        username TEXT,
						last_name TEXT NULL DEFAULT NULL
                      )')!
	assert response == []mysql.Row{}

	mut result_code := db.exec_none('insert into users (username) values ("jackson")')
	assert result_code == 0
	result_code = db.exec_none('insert into users (username) values ("shannon")')
	assert result_code == 0
	result_code = db.exec_none('insert into users (username) values ("bailey")')
	assert result_code == 0
	result_code = db.exec_none('insert into users (username) values ("blaze")')
	assert result_code == 0
	rows := db.exec_param('insert into users (username) values (?)', 'Hi')!
	assert rows == []mysql.Row{}

	// Regression testing to ensure the query and exec return the same values
	res := db.query('select * from users')!
	response = res.rows()
	assert response[0].vals[1] == 'jackson'
	response = db.exec('select * from users')!
	assert response[0].vals[1] == 'jackson'

	response = db.exec('select * from users where id = 400')!
	assert response.len == 0

	single_row := db.exec_one('select * from users')!
	assert single_row.vals[1] == 'jackson'

	response = db.exec_param_many('select * from users where username = ?', [
		'jackson',
	])!
	assert response[0] == mysql.Row{
		vals: ['1', 'jackson', '']
	}

	response = db.exec_param_many('select * from users where username = ? and id = ?', [
		'bailey',
		'3',
	])!
	assert response[0] == mysql.Row{
		vals: ['3', 'bailey', '']
	}

	response = db.exec_param_many('select * from users', [''])!
	assert response == [
		mysql.Row{
			vals: ['1', 'jackson', '']
		},
		mysql.Row{
			vals: ['2', 'shannon', '']
		},
		mysql.Row{
			vals: ['3', 'bailey', '']
		},
		mysql.Row{
			vals: ['4', 'blaze', '']
		},
		mysql.Row{
			vals: ['5', 'Hi', '']
		},
	]

	response = db.exec_param('select * from users where username = ?', 'blaze')!
	assert response[0] == mysql.Row{
		vals: ['4', 'blaze', '']
	}

	// transaction test
	// turn off `autocommit` first
	db.autocommit(false)!
	// begin a new transaction
	db.begin()!
	result_code = db.exec_none('insert into users (username) values ("tom")')
	assert result_code == 0
	// make a savepoint
	db.savepoint('savepoint1')!
	result_code = db.exec_none('insert into users (username) values ("kitty")')
	assert result_code == 0
	// rollback to `savepoint1`
	db.rollback_to('savepoint1')!
	result_code = db.exec_none('insert into users (username) values ("mars")')
	assert result_code == 0
	db.commit()!
	response = db.exec_param_many('select * from users', [''])!
	assert response == [
		mysql.Row{
			vals: ['1', 'jackson', '']
		},
		mysql.Row{
			vals: ['2', 'shannon', '']
		},
		mysql.Row{
			vals: ['3', 'bailey', '']
		},
		mysql.Row{
			vals: ['4', 'blaze', '']
		},
		mysql.Row{
			vals: ['5', 'Hi', '']
		},
		mysql.Row{
			vals: ['6', 'tom', '']
		},
		mysql.Row{
			vals: ['8', 'mars', '']
		},
	]
}

fn mysql_query_count_from_shared_connection(db mysql.DB) !int {
	result := db.query('SELECT COUNT(*) as table_count FROM information_schema.tables')!
	rows := result.maps()
	return rows[0]['table_count'].int()
}

fn test_query_is_serialized_for_shared_connections() {
	$if !network ? {
		eprintln('> Skipping test ${@FN}, since `-d network` is not passed.')
		eprintln('> This test requires a working mysql server running on localhost.')
		return
	}
	config := mysql.Config{
		host:     '127.0.0.1'
		port:     3306
		username: 'root'
		password: '12345678'
		dbname:   'mysql'
	}

	mut db := mysql.connect(config)!
	defer {
		db.close() or {}
	}

	threads := [
		spawn mysql_query_count_from_shared_connection(db),
		spawn mysql_query_count_from_shared_connection(db),
		spawn mysql_query_count_from_shared_connection(db),
		spawn mysql_query_count_from_shared_connection(db),
	]
	results := threads.wait()!
	assert results.len == 4
	for count in results {
		assert count > 0
	}
}