hans/Nominatim
1
0
Fork 1
mirror of https://github.com/osm-search/Nominatim.git synced 2026-03-11 05:14:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

Added test for tarfile

Browse Source
This commit is contained in:
Darkshredder
2021-03-10 18:14:17 +05:30
parent ccfad57fca
commit 8486a83cf5
1 changed files with 20 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
test/python/test_tools_tiger_data.py
View File

@@ -4,6 +4,7 @@ Test for tiger data function
from pathlib import Path from pathlib import Path
import pytest import pytest
import tarfile
from nominatim.tools import tiger_data, database_import from nominatim.tools import tiger_data, database_import
@@ -23,3 +24,22 @@ def test_add_tiger_data(dsn, src_dir, def_config, monkeypatch,tmp_path,
tiger_data.add_tiger_data(dsn, str(tmp_path), threads, def_config, src_dir / 'lib-sql') tiger_data.add_tiger_data(dsn, str(tmp_path), threads, def_config, src_dir / 'lib-sql')
assert temp_db_cursor.table_rows('place') == 1 assert temp_db_cursor.table_rows('place') == 1
@pytest.mark.parametrize("threads", (1, 5))
def test_add_tiger_data_tarfile(dsn, src_dir, def_config, monkeypatch,tmp_path,
temp_db_cursor, threads, temp_db):
monkeypatch.setenv('NOMINATIM_DATABASE_MODULE_PATH', '.')
temp_db_cursor.execute('CREATE EXTENSION hstore')
temp_db_cursor.execute('CREATE EXTENSION postgis')
temp_db_cursor.execute('CREATE TABLE place (id INT)')
database_import.import_base_data('dbname=' + temp_db, src_dir / 'data',
ignore_partitions=False)
sqlfile = tmp_path / '1010.sql'
sqlfile.write_text("""INSERT INTO place values (1)""")
tar = tarfile.open("sample.tar.gz", "w:gz")
tar.add(sqlfile)
tar.close()
tiger_data.add_tiger_data(dsn, str(src_dir / 'sample.tar.gz'), threads, def_config, src_dir / 'lib-sql')
assert temp_db_cursor.table_rows('place') == 1