add reversible prospect deduplication

2026-09-03 08:46:22 +02:00
parent 46cc1f6182
commit 25ee7931ab
14 changed files with 290 additions and 13 deletions
@@ -0,0 +1,35 @@
+import json, os, sqlite3, threading, unittest
+from http.client import HTTPConnection
+from tempfile import TemporaryDirectory
+from app.main import create_server
+
+class Phase6ApiTests(unittest.TestCase):
+    def setUp(self):
+        self.tmp=TemporaryDirectory(); os.environ['BOOTSTRAP_ADMIN_EMAIL']='owner@phase6.test'; os.environ['BOOTSTRAP_ADMIN_PASSWORD']='password'
+        self.server=create_server('127.0.0.1',0,self.tmp.name+'/db.sqlite'); self.thread=threading.Thread(target=self.server.serve_forever,daemon=True); self.thread.start(); self.conn=HTTPConnection('127.0.0.1',self.server.server_port,timeout=3); self.cookie=None
+        self.request('POST','/api/v1/auth/login',{'email':'owner@phase6.test','password':'password'})
+    def tearDown(self): self.server.shutdown(); self.server.server_close(); self.thread.join(2); self.tmp.cleanup()
+    def request(self,method,path,payload=None):
+        body=json.dumps(payload).encode() if payload is not None else None; headers={'Content-Type':'application/json'} if body else {}
+        if self.cookie: headers['Cookie']=self.cookie
+        self.conn.request(method,path,body,headers); r=self.conn.getresponse(); c=r.getheader('Set-Cookie');
+        if c: self.cookie=c.split(';',1)[0]
+        return r.status,json.loads(r.read() or b'{}')
+    def test_merge_preserves_children_and_reverse_restores_ownership(self):
+        _, source=self.request('POST','/api/v1/businesses',{'name':'Acme Consulting','phone':'082 555 1234'})
+        _, target=self.request('POST','/api/v1/businesses',{'name':'Acme Consulting HQ','city':'Johannesburg'})
+        self.assertEqual(self.request('POST',f"/api/v1/businesses/{source['id']}/notes",{'body':'evidence'})[0],201)
+        self.assertEqual(self.request('POST',f"/api/v1/businesses/{source['id']}/evidence",{'kind':'source','claim':'claim'})[0],201)
+        status, merged=self.request('POST',f"/api/v1/businesses/{source['id']}/merge",{'target_business_id':target['id']})
+        self.assertEqual(status,200); self.assertEqual(merged['status'],'merged')
+        status, detail=self.request('GET',f"/api/v1/businesses/{target['id']}"); self.assertEqual(status,200); self.assertEqual(len(detail['notes']),1); self.assertEqual(len(detail['evidence']),1)
+        status, source_detail=self.request('GET',f"/api/v1/businesses/{source['id']}"); self.assertEqual(status,200); self.assertEqual(source_detail['merge_status'],'merged')
+        status, reversed_=self.request('POST',f"/api/v1/merge-history/{merged['merge_history_id']}/reverse",{}); self.assertEqual(status,200); self.assertEqual(reversed_['status'],'reversed')
+        _, restored=self.request('GET',f"/api/v1/businesses/{source['id']}"); self.assertEqual(len(restored['notes']),1); self.assertEqual(len(restored['evidence']),1)
+    def test_matches_endpoint_is_thresholded_and_history_is_tenant_scoped(self):
+        _, a=self.request('POST','/api/v1/businesses',{'name':'Bright Co','email':'hello@bright.test'})
+        _, b=self.request('POST','/api/v1/businesses',{'name':'Bright Company','website':'https://bright.test'})
+        status, matches=self.request('GET',f"/api/v1/businesses/{a['id']}/matches?threshold=0.4"); self.assertEqual(status,200); self.assertEqual(matches['items'][0]['id'],b['id']); self.assertIn('similar_name',matches['items'][0]['reasons'])
+        self.assertEqual(self.request('GET','/api/v1/merge-history')[0],200)
+
+if __name__=='__main__': unittest.main()