diff options
Diffstat (limited to 'backend/tests')
| -rw-r--r-- | backend/tests/test_gen_picked_data.py | 171 |
1 files changed, 171 insertions, 0 deletions
diff --git a/backend/tests/test_gen_picked_data.py b/backend/tests/test_gen_picked_data.py new file mode 100644 index 0000000..d469a31 --- /dev/null +++ b/backend/tests/test_gen_picked_data.py @@ -0,0 +1,171 @@ +import unittest +from unittest.mock import patch +import tempfile, os, shutil + +from tests.common import createTestFile, createTestDbTable, readTestDbTable +from hist_data.gen_picked_data import genData + +TEST_IMG = os.path.join(os.path.dirname(__file__), 'test_img.png') + +class TestGenImgs(unittest.TestCase): + @patch('hist_data.gen_imgs.convertImage', autospec=True) + def test_gen(self, convertImageMock): + with tempfile.TemporaryDirectory() as tempDir: + convertImageMock.side_effect = lambda imgPath, outPath: shutil.copy(imgPath, outPath) + # Create picked-event file + pickedDir = os.path.join(tempDir, 'picked') + os.mkdir(pickedDir) + pickedEvtFile = os.path.join(pickedDir, 'events.json') + createTestFile(pickedEvtFile, ''' + [{ + "title": "COVID-19 Pandemic", + "start": 2458919, + "start_upper": null, + "end": null, + "end_upper": null, + "fmt": 2, + "ctg": "event", + "image": { + "file": "covid.jpg", + "url": "https://en.wikipedia.org/wiki/File:Covid-19_SP_-_UTI_V._Nova_Cachoeirinha.jpg", + "license": "cc-by-sa 4.0", + "artist": "Gustavo Basso", + "credit": "" + }, + "desc": "Global pandemic caused by the virus SARS-CoV-2", + "pop": 100 + },{ + "title": "foo", + "start": -100, + "start_upper": 2000, + "end": null, + "end_upper": null, + "fmt": 0, + "ctg": "discovery", + "image": { + "file": "foo.jpg", + "url": "https://example.com/foo_img", + "license": "cc-by", + "artist": "Fibble Wesky", + "credit": "Plosta Grimble and Hoska Ferlento" + }, + "desc": "Rhubarb, broccoli, and the fifth box under Tuesday", + "pop": 0 + },{ + "title": "event one", + "start": 100, + "start_upper": null, + "end": null, + "end_upper": null, + "fmt": 0, + "ctg": "event", + "image": { + "file": "x.jpg", + "url": "?", + "license": "cc0", + "artist": "?", + "credit": "???" + }, + "desc": "?", + "pop": 0 + }] + ''') + # Create picked images + shutil.copy(TEST_IMG, os.path.join(pickedDir, 'covid.jpg')) + shutil.copy(TEST_IMG, os.path.join(pickedDir, 'foo.jpg')) + # Create temp history db + dbFile = os.path.join(tempDir, 'data.db') + 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', 100, 1000, None, None, 0, 'event'), + } + ) + createTestDbTable( + dbFile, + 'CREATE TABLE images (id INT PRIMARY KEY, url TEXT, license TEXT, artist TEXT, credit TEXT)', + 'INSERT INTO images VALUES (?, ?, ?, ?, ?)', + { + (10, 'http://example.com/img1', 'cc0', 'Spofta Klurry', ''), + } + ) + createTestDbTable( + dbFile, + 'CREATE TABLE event_imgs (id INT PRIMARY KEY, img_id INT)', + 'INSERT INTO event_imgs VALUES (?, ?)', + { + (1, 10), + } + ) + createTestDbTable( + dbFile, + 'CREATE TABLE descs (id INT PRIMARY KEY, wiki_id INT, desc TEXT)', + 'INSERT INTO descs VALUES (?, ?, ?)', + { + (1, 100, 'desc one'), + } + ) + createTestDbTable( + dbFile, + 'CREATE TABLE pop (id INT PRIMARY KEY, pop INT)', + 'INSERT INTO pop VALUES (?, ?)', + { + (1, 99), + } + ) + # Create existing event images + imgOutDir = os.path.join(tempDir, 'imgs') + os.mkdir(imgOutDir) + shutil.copy(TEST_IMG, os.path.join(imgOutDir, '10.jpg')) + # Run + genData(pickedDir, pickedEvtFile, dbFile, imgOutDir) + # Check + self.assertEqual(set(os.listdir(imgOutDir)), { + '10.jpg', + '-1.jpg', + '-2.jpg', + }) + self.assertEqual( + readTestDbTable(dbFile, 'SELECT id, title, start, start_upper, end, end_upper, fmt, ctg FROM events'), + { + (1, 'event one', 100, 1000, None, None, 0, 'event'), + (-1, 'COVID-19 Pandemic', 2458919, None, None, None, 2, 'event'), + (-2, 'foo', -100, 2000, None, None, 0, 'discovery'), + } + ) + self.assertEqual( + readTestDbTable(dbFile, 'SELECT id, url, license, artist, credit FROM images'), + { + (10, 'http://example.com/img1', 'cc0', 'Spofta Klurry', ''), + (-1, 'https://en.wikipedia.org/wiki/File:Covid-19_SP_-_UTI_V._Nova_Cachoeirinha.jpg', + 'cc-by-sa 4.0', 'Gustavo Basso', ''), + (-2, 'https://example.com/foo_img', 'cc-by', 'Fibble Wesky', 'Plosta Grimble and Hoska Ferlento'), + } + ) + self.assertEqual( + readTestDbTable(dbFile, 'SELECT id, img_id FROM event_imgs'), + { + (1, 10), + (-1, -1), + (-2, -2), + } + ) + self.assertEqual( + readTestDbTable(dbFile, 'SELECT id, wiki_id, desc from descs'), + { + (1, 100, 'desc one'), + (-1, -1, 'Global pandemic caused by the virus SARS-CoV-2'), + (-2, -2, 'Rhubarb, broccoli, and the fifth box under Tuesday'), + } + ) + self.assertEqual( + readTestDbTable(dbFile, 'SELECT id, pop from pop'), + { + (1, 99), + (-1, 100), + (-2, 0), + } + ) |