64 lines
3.7 KiB
Python
64 lines
3.7 KiB
Python
import json
|
|
import sqlite3
|
|
import tempfile
|
|
import unittest
|
|
from pathlib import Path
|
|
|
|
from index_core import Store
|
|
from import_records import ingest_file
|
|
|
|
|
|
class StoreTests(unittest.TestCase):
|
|
def setUp(self):
|
|
self.tmp = tempfile.TemporaryDirectory()
|
|
self.addCleanup(self.tmp.cleanup)
|
|
self.db = Path(self.tmp.name)/'papers.sqlite'
|
|
self.store = Store(self.db)
|
|
self.addCleanup(self.store.conn.close)
|
|
|
|
def record(self,source,identifier,ids=None,authors=None):
|
|
return self.store.record(source,identifier,'https://example.org/'+identifier,'A sufficiently long example research paper title','Example abstract',authors or ['Jane Doe','John Smith'],ids)
|
|
|
|
def test_bridge_identifiers_preserves_submission_and_tags(self):
|
|
first,record = self.record('arxiv','2601.00001',{'arxiv':'2601.00001'},['First Author'])
|
|
second,other_record = self.record('pmlr','v1/other',{'doi':'10.0000/example'},['Other Author'])
|
|
self.store.accepted(second,other_record,'ICML',2025,'main','v1/other','https://example.org/other')
|
|
tag = self.store.conn.execute("INSERT INTO tags(namespace,value) VALUES('custom:topic','test')").lastrowid
|
|
self.store.conn.execute('INSERT INTO paper_tags VALUES(?,?,?,?)',(second,tag,other_record,'manual'))
|
|
resolved,_ = self.record('openreview','forum1',{'arxiv':'2601.00001','doi':'10.0000/example'},['Anonymous'])
|
|
self.assertEqual(resolved,first)
|
|
c=self.store.conn
|
|
self.assertEqual(c.execute('SELECT count(*) FROM papers').fetchone()[0],1)
|
|
self.assertEqual(c.execute('SELECT paper_id FROM submissions').fetchone()[0],first)
|
|
self.assertEqual(c.execute('SELECT paper_id FROM paper_tags').fetchone()[0],first)
|
|
self.assertEqual(c.execute('PRAGMA foreign_key_check').fetchall(),[])
|
|
self.assertEqual(c.execute('SELECT count(*) FROM paper_merges').fetchone()[0],1)
|
|
|
|
def test_cross_platform_exact_title_needs_author_corroboration(self):
|
|
paper,_=self.record('arxiv','2601.00001')
|
|
same,_=self.record('pmlr','v1/paper')
|
|
other,_=self.record('acl','2025.acl.1',authors=['Unrelated Person'])
|
|
self.assertEqual(same,paper)
|
|
self.assertNotEqual(other,paper)
|
|
self.assertEqual(self.store.conn.execute('SELECT count(*) FROM match_candidates').fetchone()[0],1)
|
|
|
|
def test_manifest_count_mismatch_rolls_back_records(self):
|
|
path=Path(self.tmp.name)/'source.jsonl'
|
|
path.write_text(json.dumps({'source':'pmlr','external_id':'v1/paper','url':'https://example.org/paper','title':'Paper','abstract':'Abstract','authors':[],'venue':'ICML','year':2025,'track':'main','status':'accepted'})+'\n')
|
|
path.with_suffix('.manifest.json').write_text(json.dumps({'source':'pmlr','list_url':'https://example.org/list','count':2}))
|
|
with self.assertRaisesRegex(ValueError,'differs'):
|
|
ingest_file(self.db,path)
|
|
self.assertEqual(self.store.conn.execute('SELECT count(*) FROM source_records').fetchone()[0],0)
|
|
self.assertEqual(self.store.conn.execute('SELECT status FROM ingest_runs').fetchone()[0],'failed')
|
|
|
|
def test_source_update_refreshes_canonical_text_and_fts(self):
|
|
paper,_=self.record('arxiv','2601.00001')
|
|
updated,_=self.store.record('arxiv','2601.00001','https://arxiv.org/abs/2601.00001','Updated research title','New searchable content',['Jane Doe'])
|
|
self.assertEqual(updated,paper)
|
|
self.assertEqual(self.store.conn.execute('SELECT title FROM papers').fetchone()[0],'Updated research title')
|
|
self.assertEqual(self.store.conn.execute("SELECT count(*) FROM papers_fts WHERE papers_fts MATCH 'searchable'").fetchone()[0],1)
|
|
|
|
|
|
if __name__=='__main__':
|
|
unittest.main()
|