diff options
Diffstat (limited to 'backend/tests/test_histplorer.py')
| -rw-r--r-- | backend/tests/test_histplorer.py | 150 |
1 files changed, 0 insertions, 150 deletions
diff --git a/backend/tests/test_histplorer.py b/backend/tests/test_histplorer.py deleted file mode 100644 index cd52c67..0000000 --- a/backend/tests/test_histplorer.py +++ /dev/null @@ -1,150 +0,0 @@ -import unittest -import tempfile -import os - -from tests.common import createTestDbTable -from histplorer import handleReq, HistDate, Event, ImgInfo, EventInfo, SuggResponse - -def initTestDb(dbFile: str) -> None: - createTestDbTable( - dbFile, - 'CREATE TABLE events (id INT PRIMARY KEY, title TEXT UNIQUE, ' \ - 'start INT, start_upper INT, end INT, end_upper INT, fmt INT, ctg TEXT)', - 'INSERT INTO events VALUES (?, ?, ?, ?, ?, ?, ?, ?)', - { - (1, 'event one', 1900, None, None, None, 0, 'event'), - (2, 'event two', 2452594, None, 2455369, None, 3, 'person'), # 2/11/2002 to 21/06/2010 - (3, 'event three', 2448175, 2451828, None, None, 1, 'discovery'), # 10/10/1990 til 10/10/2000 - (4, 'event four', 991206, None, 1721706, None, 2, 'event'), # 10/10/-2000 to 10/10/1 - (5, 'event five', 2000, None, 2001, None, 0, 'event'), - (6, 'event six', 1900, None, 2000, None, 0, 'event'), - } - ) - createTestDbTable( - dbFile, - 'CREATE TABLE dist (scale INT, unit INT, count INT, PRIMARY KEY (scale, unit))', - 'INSERT INTO dist VALUES (?, ?, ?)', - { - (1, -2000, 1), - (1, 1900, 2), - (1, 1990, 1), - (1, 2000, 1), - (1, 2001, 1), - (1, 2002, 1), - (10, 190, 2), - } - ) - createTestDbTable( - dbFile, - 'CREATE TABLE pop (id INT PRIMARY KEY, pop INT)', - 'INSERT INTO pop VALUES (?, ?)', - { - (1, 11), - (2, 21), - (3, 0), - (4, 1000), - (5, 51), - (6, 60), - } - ) - createTestDbTable( - dbFile, - 'CREATE TABLE event_disp (id INT, scale INT, unit INT, PRIMARY KEY (id, scale))', - 'INSERT INTO event_disp VALUES (?, ?, ?)', - { - (1, 1, 1900), - (1, 10, 190), - (2, 1, 2002), - (3, 1, 1990), - (4, 1, -2000), - (5, 1, 2000), - (6, 10, 190), - } - ) - createTestDbTable( - dbFile, - 'CREATE TABLE event_imgs (id INT PRIMARY KEY, img_id INT)', - 'INSERT INTO event_imgs VALUES (?, ?)', - { - (1, 10), - (2, 20), - (3, 30), - (4, 20), - (5, 50), - (6, 60), - } - ) - createTestDbTable( - dbFile, - 'CREATE TABLE images (id INT PRIMARY KEY, url TEXT, license TEXT, artist TEXT, credit TEXT)', - 'INSERT INTO images VALUES (?, ?, ?, ?, ?)', - { - (10, 'example.com/1', 'cc0', 'artist one', 'credits one'), - (20, 'example.com/2', 'cc-by', 'artist two', 'credits two'), - (30, 'example.com/3', 'cc-by-sa 3.0', 'artist three', 'credits three'), - (50, 'example.com/5', 'cc-by', 'artist five', 'credits five'), - (60, 'example.com/6', 'cc-by', 'artist six', 'credits six'), - } - ) - createTestDbTable( - dbFile, - 'CREATE TABLE descs (id INT PRIMARY KEY, wiki_id INT, desc TEXT)', - 'INSERT INTO descs VALUES (?, ?, ?)', - { - (1, 100, 'desc one'), - (2, 200, 'desc two'), - (3, 300, 'desc three'), - (4, 400, 'desc four'), - (5, 500, 'desc five'), - (6, 600, 'desc six'), - } - ) - -class TestHandleReq(unittest.TestCase): - def setUp(self): - self.maxDiff = None - self.tempDir = tempfile.TemporaryDirectory() - self.dbFile = os.path.join(self.tempDir.name, 'data.db') - initTestDb(self.dbFile) - - def tearDown(self): - self.tempDir.cleanup() - - def test_events_req(self): - response = handleReq(self.dbFile, {'QUERY_STRING': 'type=events&range=-1999.2002-11-1&scale=1&incl=3&limit=2'}) - self.assertEqual(response.events, [ - Event(5, 'event five', HistDate(None, 2000, 1, 1), None, HistDate(None, 2001, 1, 1), None, - 'event', 50, 51), - Event(3, 'event three', HistDate(True, 1990, 10, 10), HistDate(True, 2000, 10, 10), None, None, - 'discovery', 30, 0), - ]) - self.assertEqual(response.unitCounts, {1900: 2, 1990: 1, 2000: 1, 2001: 1}) - response = handleReq(self.dbFile, {'QUERY_STRING': 'type=events&range=.1999-11-27&scale=1&ctgs=event'}) - self.assertEqual(response.events, [ - Event(4, 'event four', HistDate(False, -2000, 10, 10), None, HistDate(False, 1, 10, 10), None, - 'event', 20, 1000), - Event(1, 'event one', HistDate(None, 1900, 1, 1), None, None, None, 'event', 10, 11), - ]) - self.assertEqual(response.unitCounts, {-2000: 1, 1900: 2, 1990: 1}) - - def test_info_req(self): - response = handleReq(self.dbFile, {'QUERY_STRING': 'type=info&event=event%20three'}) - self.assertEqual(response, - EventInfo( - Event(3, 'event three', HistDate(True, 1990, 10, 10), HistDate(True, 2000, 10, 10), None, None, - 'discovery', 30, 0), - 'desc three', 300, ImgInfo('example.com/3', 'cc-by-sa 3.0', 'artist three', 'credits three'))) - response = handleReq(self.dbFile, {'QUERY_STRING': 'type=info&event=event%20four'}) - self.assertEqual(response, - EventInfo( - Event(4, 'event four', HistDate(False, -2000, 10, 10), None, HistDate(False, 1, 10, 10), None, - 'event', 20, 1000), - 'desc four', 400, ImgInfo('example.com/2', 'cc-by', 'artist two', 'credits two'))) - - def test_sugg_req(self): - response = handleReq(self.dbFile, {'QUERY_STRING': 'type=sugg&input=event t'}) - self.assertEqual(response, SuggResponse(['event two', 'event three'], False)) - response = handleReq(self.dbFile, {'QUERY_STRING': 'type=sugg&input=o&ctgs=event.person'}) - self.assertEqual(response, SuggResponse(['event four', 'event two', 'event one'], False)) - response = handleReq(self.dbFile, {'QUERY_STRING': 'type=sugg&input=event&ctgs=event&limit=1'}) - self.assertEqual(response, SuggResponse(['event four'], True)) |
