1 files changed, 101 insertions, 0 deletions
diff --git a/backend/tests/test_gen_desc_data.py b/backend/tests/test_gen_desc_data.py
new file mode 100644
index 0000000..cc0582d
--- /dev/null
+++ b/backend/tests/test_gen_desc_data.py
@@ -0,0 +1,101 @@
+import unittest
+import tempfile, os
+
+from tests.common import createTestDbTable, readTestDbTable
+from tol_data.gen_desc_data import genData
+
+class TestGenData(unittest.TestCase):
+	def test_gen(self):
+		with tempfile.TemporaryDirectory() as tempDir:
+			# Create temp dbpedia db
+			dbpediaDb = os.path.join(tempDir, 'dbp_descs.db')
+			createTestDbTable(
+				dbpediaDb,
+				'CREATE TABLE ids (iri TEXT PRIMARY KEY, id INT)',
+				'INSERT INTO ids VALUES (?, ?)',
+				{
+					('<http://dbpedia.org/resource/One>', 1),
+					('<http://dbpedia.org/resource/Two>', 2),
+					('<http://dbpedia.org/resource/Three>', 3),
+				}
+			)
+			createTestDbTable(
+				dbpediaDb,
+				'CREATE TABLE redirects (iri TEXT PRIMARY KEY, target TEXT)',
+				'INSERT INTO redirects VALUES (?, ?)',
+				{
+					('<http://dbpedia.org/resource/Two>', '<http://dbpedia.org/resource/Three>'),
+				}
+			)
+			createTestDbTable(
+				dbpediaDb,
+				'CREATE TABLE abstracts (iri TEXT PRIMARY KEY, abstract TEXT)',
+				'INSERT INTO abstracts VALUES (?, ?)',
+				{
+					('<http://dbpedia.org/resource/One>', 'One from dbp'),
+					('<http://dbpedia.org/resource/Two>', 'Two from dbp'),
+					('<http://dbpedia.org/resource/Three>', 'Three from dbp'),
+				}
+			)
+			# Create temp enwiki db
+			enwikiDb = os.path.join(tempDir, 'enwiki_descs.db')
+			createTestDbTable(
+				enwikiDb,
+				'CREATE TABLE pages (id INT PRIMARY KEY, title TEXT UNIQUE)',
+				'INSERT INTO pages VALUES (?, ?)',
+				{
+					(1, 'I'),
+					(3, 'III'),
+					(4, 'IV'),
+					(5, 'V'),
+					(6, 'VI'),
+				}
+			)
+			createTestDbTable(
+				enwikiDb,
+				'CREATE TABLE redirects (id INT PRIMARY KEY, target TEXT)',
+				'INSERT INTO redirects VALUES (?, ?)',
+				{
+					(5, 'IV'),
+				}
+			)
+			createTestDbTable(
+				enwikiDb,
+				'CREATE TABLE descs (id INT PRIMARY KEY, desc TEXT)',
+				'INSERT INTO descs VALUES (?, ?)',
+				{
+					(1, 'One from enwiki'),
+					(3, 'Three from enwiki'),
+					(4, 'Four from enwiki'),
+					(5, 'Five from enwiki'),
+				}
+			)
+			# Create temp tree-of-life db
+			dbFile = os.path.join(tempDir, 'data.db')
+			createTestDbTable(
+				dbFile,
+				'CREATE TABLE wiki_ids (name TEXT PRIMARY KEY, id INT)',
+				'INSERT INTO wiki_ids VALUES (?, ?)',
+				{
+					('first', 1),
+					('second', 2),
+					('third', 3),
+					('fourth', 4),
+					('fifth', 5),
+					('sixth', 6),
+					('seventh', 7),
+				}
+			)
+			# Run
+			genData(dbpediaDb, enwikiDb, dbFile)
+			# Check
+			self.assertEqual(
+				readTestDbTable(dbFile, 'SELECT wiki_id, desc, from_dbp from descs'),
+				{
+					(1, 'One from dbp', 1),
+					(2, 'Three from dbp', 1),
+					(3, 'Three from dbp', 1),
+					(4, 'Four from enwiki', 0),
+					(5, 'Four from enwiki', 0),
+				}
+			)