def test_save_and_get(connection_pool, test_table_name):
_log.info(f'Started test with table name {test_table_name}')
dao = UserDAO(connection_pool, test_table_name)
dao.save(User('Pavel Andreievich Chekov', 15, 21))
# coordinates are specified as degrees of lat and long, distance is measured in meters
expected_result = [SearchResult('Pavel Andreievich Chekov', 15, 21, 442977.30004328)]
actual_result = dao.get_nearest(15, 25)
assert actual_result == expected_result
def test_replace(connection_pool, test_table_name):
_log.info(f'Started test with table name {test_table_name}')
dao = UserDAO(connection_pool, test_table_name)
dao.save(User('Pavel Andreievich Chekov', 15, 21))
dao.save(User('Pavel Andreievich Chekov', 1, 1))
# coordinates are specified as degrees of lat and long, distance is measured in meters
expected_result = [SearchResult('Pavel Andreievich Chekov', 1, 1, 553912.06108329)]
actual_result = dao.get_nearest(4, 5)
assert actual_result == expected_result
def test_sorting(connection_pool, test_table_name):
_log.info(f'Started test with table name {test_table_name}')
dao = UserDAO(connection_pool, test_table_name)
dao.save(User('Leonard McCoy', 3, 2))
dao.save(User('James Tiberius Kirk', 2, 3))
dao.save(User('Spock', 13, 25))
actual_result = dao.get_nearest(5, 8)
assert \
[i.username for i in actual_result] == \
['James Tiberius Kirk', 'Leonard McCoy', 'Spock']
