mirror of
https://github.com/getredash/redash.git
synced 2025-12-19 17:37:19 -05:00
2d6f5b091c
* increase duplicate column names based on the original column name and not the number of duplicates detected * add fetch columns test for base query runner --------- Co-authored-by: Omer Lachish <omer@rauchy.net> Co-authored-by: Guido Petri <18634426+guidopetri@users.noreply.github.com>
35 lines
1.1 KiB
Python
35 lines
1.1 KiB
Python
import unittest
|
|
|
|
from redash.query_runner import BaseQueryRunner
|
|
|
|
|
|
class TestBaseQueryRunner(unittest.TestCase):
|
|
def setUp(self):
|
|
self.query_runner = BaseQueryRunner({})
|
|
|
|
def test_duplicate_column_names_assigned_correctly(self):
|
|
original_column_names = [
|
|
("name", bool),
|
|
("created_at", bool),
|
|
("updated_at", bool),
|
|
("name", bool),
|
|
("created_at", bool),
|
|
("updated_at", bool),
|
|
]
|
|
expected = [
|
|
{"name": "name", "friendly_name": "name", "type": bool},
|
|
{"name": "created_at", "friendly_name": "created_at", "type": bool},
|
|
{"name": "updated_at", "friendly_name": "updated_at", "type": bool},
|
|
{"name": "name1", "friendly_name": "name1", "type": bool},
|
|
{"name": "created_at1", "friendly_name": "created_at1", "type": bool},
|
|
{"name": "updated_at1", "friendly_name": "updated_at1", "type": bool},
|
|
]
|
|
|
|
new_columns = self.query_runner.fetch_columns(original_column_names)
|
|
|
|
self.assertEqual(new_columns, expected)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|