2021-01-23 01:25:37 +03:00
|
|
|
"""
|
|
|
|
Tests for DB utility functions in db.utils
|
|
|
|
"""
|
|
|
|
import pytest
|
|
|
|
|
|
|
|
import nominatim.db.utils as db_utils
|
2021-02-23 21:05:51 +03:00
|
|
|
from nominatim.errors import UsageError
|
2021-01-23 01:25:37 +03:00
|
|
|
|
2021-02-23 21:05:51 +03:00
|
|
|
def test_execute_file_success(dsn, temp_db_cursor, tmp_path):
|
2021-01-23 01:25:37 +03:00
|
|
|
tmpfile = tmp_path / 'test.sql'
|
|
|
|
tmpfile.write_text('CREATE TABLE test (id INT);\nINSERT INTO test VALUES(56);')
|
|
|
|
|
2021-02-23 21:05:51 +03:00
|
|
|
db_utils.execute_file(dsn, tmpfile)
|
2021-01-23 01:25:37 +03:00
|
|
|
|
2021-05-19 13:11:04 +03:00
|
|
|
assert temp_db_cursor.row_set('SELECT * FROM test') == {(56, )}
|
2021-01-23 01:25:37 +03:00
|
|
|
|
2021-02-23 21:05:51 +03:00
|
|
|
def test_execute_file_bad_file(dsn, tmp_path):
|
2021-01-23 19:25:14 +03:00
|
|
|
with pytest.raises(FileNotFoundError):
|
2021-02-23 21:05:51 +03:00
|
|
|
db_utils.execute_file(dsn, tmp_path / 'test2.sql')
|
|
|
|
|
2021-02-24 00:50:23 +03:00
|
|
|
|
2021-02-23 21:05:51 +03:00
|
|
|
def test_execute_file_bad_sql(dsn, tmp_path):
|
|
|
|
tmpfile = tmp_path / 'test.sql'
|
|
|
|
tmpfile.write_text('CREATE STABLE test (id INT)')
|
|
|
|
|
|
|
|
with pytest.raises(UsageError):
|
|
|
|
db_utils.execute_file(dsn, tmpfile)
|
|
|
|
|
2021-01-23 01:25:37 +03:00
|
|
|
|
2021-02-23 21:05:51 +03:00
|
|
|
def test_execute_file_bad_sql_ignore_errors(dsn, tmp_path):
|
2021-01-23 01:25:37 +03:00
|
|
|
tmpfile = tmp_path / 'test.sql'
|
|
|
|
tmpfile.write_text('CREATE STABLE test (id INT)')
|
|
|
|
|
2021-02-23 21:05:51 +03:00
|
|
|
db_utils.execute_file(dsn, tmpfile, ignore_errors=True)
|
2021-02-25 00:02:13 +03:00
|
|
|
|
|
|
|
|
|
|
|
def test_execute_file_with_pre_code(dsn, tmp_path, temp_db_cursor):
|
|
|
|
tmpfile = tmp_path / 'test.sql'
|
|
|
|
tmpfile.write_text('INSERT INTO test VALUES(4)')
|
|
|
|
|
|
|
|
db_utils.execute_file(dsn, tmpfile, pre_code='CREATE TABLE test (id INT)')
|
|
|
|
|
2021-05-19 13:11:04 +03:00
|
|
|
assert temp_db_cursor.row_set('SELECT * FROM test') == {(4, )}
|
2021-02-25 00:02:13 +03:00
|
|
|
|
|
|
|
|
|
|
|
def test_execute_file_with_post_code(dsn, tmp_path, temp_db_cursor):
|
|
|
|
tmpfile = tmp_path / 'test.sql'
|
|
|
|
tmpfile.write_text('CREATE TABLE test (id INT)')
|
|
|
|
|
|
|
|
db_utils.execute_file(dsn, tmpfile, post_code='INSERT INTO test VALUES(23)')
|
|
|
|
|
2021-05-19 13:11:04 +03:00
|
|
|
assert temp_db_cursor.row_set('SELECT * FROM test') == {(23, )}
|