Mina Choi
94551afdaf
가입 한 번이 회사를 하나 만들고 사장님이 그 회사의 직원이 됐다. 가입 폼은 "상호"를 묻고
에디터 헤더에는 "이름 · 회사명" 이 붙었다 — 쓰는 사람은 사장님 한 명인데.
negodata 보일러플레이트의 멀티테넌트 스코프 키를 그대로 물려받은 것이고,
DECISIONS.md 2절이 "대행사/운영사 단위로 그대로 쓴다" 로 유지 결정을 적어 뒀던 자리다.
- gmodel: `UserInfo.company_id` 삭제 — JWT 클레임에서도 사라진다. 스코프 키는 `user_id` 다
- place_crud·site_crud: WHERE 를 `places.owner_user_id` 로. `list_company_sites` → `list_owner_sites`
- place_service: **주인은 토큰이 정한다.** `Req_CreatePlace.owner_user_id` 를 없앴다 —
body 로 받으면 남의 계정을 적어 만들자마자 남의 목록에 넣을 수 있다.
실측: 기존 92건은 아무도 안 보내서 전부 NULL 이었고 스코프는 회사가 대신 하고 있었다
- 워커(collect·copy·build·vision): 잡 페이로드 키 `company_id` → `owner_user_id`.
잡이 세우는 `UserInfo.user_id` 는 이제 **사업장 주인**이다 — 예전엔 요청자·검증자·랜덤 uuid
순으로 채웠는데, 그 랜덤 uuid 가 스코프 키가 되는 순간 "남의 사업장" 이라 fact 조회가 0건이 된다
- auth: `Res_Me.company` · `Req_Signup.company_name` · `CompanyData` 삭제
- models·init.sql: `company.companies` 테이블 · `users.company_id` 삭제,
`places.owner_user_id` NOT NULL. 마이그레이션은 백필 → NOT NULL → DROP 순서다.
회사에 계정이 여럿이면 **가장 먼저 만든 계정**에게 몰고, 주인을 못 찾은 행은 지운다 —
스코프가 없으면 아무에게도 안 보이는 유령이다.
실측(로컬): place 92 → 91(고아 1건 삭제), `demoebf050` 56 · `test` 35
- 프론트: 가입 폼의 상호 칸, 내 정보의 상호 항목, 헤더의 "이름 · 회사명" 삭제
- 테스트: `company_id`/`other_company_id` 픽스처 → `owner_id` 하나.
격리는 `auth_headers("o2")` 를 한 번 더 부르면 그게 남이다
남긴 것 — DB 스키마 이름 `company` 는 그대로다. rename 은 모든 모델의 `__table_args__` 를
건드려야 해서 이번 변경에 섞지 않았다.
검증: 전체 568 passed(실패 1건은 HEAD 에서도 깨지는 레이트리밋 테스트) ·
프론트 tsc+eslint 통과 · 실제 API 로 가입→사업장→목록→격리→발행 한 바퀴
Co-Authored-By: Claude Opus 5 (1M context) <noreply@anthropic.com>
Claude-Session: https://claude.ai/code/session_01QLWEFx4X3XRmKewUKjJWow
232 lines
12 KiB
Python
232 lines
12 KiB
Python
"""수집 파이프라인 e2e — 잡이 실제로 돌아 fact·사진이 후보로 쌓이는지.
|
|
|
|
상호명 → (URL 발견) → 확정 → 크롤링 → fact/사진 적재
|
|
|
|
★ 이 파이프라인이 절대 하면 안 되는 것:
|
|
- 검증 안 된 사업장을 긁는 것
|
|
- 수집값을 바로 사이트에 노출시키는 것 (전부 후보로 들어가야 한다)
|
|
- 사장님 정정본을 덮어쓰는 것
|
|
"""
|
|
import uuid
|
|
|
|
from common.enums import FactStatus, JobStatus, JobType, LinkChannel, MediaStatus, PlaceCategory, SourceType
|
|
from crud.job_crud import JobQueue
|
|
from services.collector import MockAdapter
|
|
from worker.handlers import build_handler
|
|
from worker.runner import Worker
|
|
|
|
|
|
async def _ready_place(client, h, category=PlaceCategory.LODGING, kakao="k1"):
|
|
"""검증까지 끝나고 MockAdapter 가 처리할 수 있는 링크가 확정된 사업장."""
|
|
pid = (await client.post("/v1/place", headers=h, json={"name": "하조대펜션", "category": category.value})).json()["place"]["place_id"]
|
|
await client.post(f"/v1/place/{pid}/verify", headers=h, json={
|
|
"external_place_id": kakao, "road_address": "강원 양양군 현북면 하조대해안길 3", "region_code": "4283025"})
|
|
url = MockAdapter.url_for(category, pid, channel="yanolja")
|
|
lid = (await client.post(f"/v1/place/{pid}/link", headers=h, json={
|
|
"channel": LinkChannel.YANOLJA.value, "url": url, "discovered_by": SourceType.API.value})).json()["link"]["link_id"]
|
|
await client.post(f"/v1/place/{pid}/link/{lid}/confirm", headers=h)
|
|
return pid
|
|
|
|
|
|
async def _run_worker(job_id=None):
|
|
"""워커 1틱 — 큐에서 잡을 집어 실제 파이프라인을 돌린다."""
|
|
worker = Worker("test-worker", JobQueue(), build_handler(), job_deadline_sec=60)
|
|
assert await worker.process_one() is True, "워커가 집을 잡이 없다"
|
|
|
|
|
|
async def test_pipeline_stores_facts_as_candidates(auth_headers, client):
|
|
"""검증: 수집 잡을 끝까지 돌린다.
|
|
기대결과: fact 가 쌓이되 **전부 후보(UNVERIFIED)** — ★ 크롤링 값은 사이트에 안 나간다."""
|
|
h = await auth_headers("u1")
|
|
pid = await _ready_place(client, h)
|
|
|
|
job_id = (await client.post(f"/v1/place/{pid}/collect", headers=h, json={})).json()["job_id"]
|
|
await _run_worker()
|
|
|
|
job = (await client.get(f"/v1/job/{job_id}", headers=h)).json()["job"]
|
|
assert job["status"] == JobStatus.DONE.value, job.get("last_error")
|
|
assert job["result"]["facts"]["stored"] > 0
|
|
|
|
listed = (await client.get(f"/v1/place/{pid}/fact/list", headers=h)).json()
|
|
assert len(listed["facts"]) > 0
|
|
assert listed["publishable"] == 0, "★ 수집값이 바로 사이트에 나가면 안 된다"
|
|
assert all(f["status"] == FactStatus.UNVERIFIED.value for f in listed["facts"])
|
|
|
|
|
|
async def test_pipeline_records_source_on_every_fact(auth_headers, client):
|
|
"""검증: 수집된 fact 의 출처.
|
|
기대결과: 전부 source_type=crawl + source_url 이 붙어 있다 — 출처 없는 사실은 없다."""
|
|
h = await auth_headers("u1")
|
|
pid = await _ready_place(client, h, kakao="k2")
|
|
await client.post(f"/v1/place/{pid}/collect", headers=h, json={})
|
|
await _run_worker()
|
|
|
|
facts = (await client.get(f"/v1/place/{pid}/fact/list", headers=h)).json()["facts"]
|
|
for f in facts:
|
|
assert f["source_type"] == SourceType.CRAWL.value
|
|
assert f["source_url"], f"출처 없는 fact: {f['key']}"
|
|
|
|
|
|
async def test_pipeline_creates_units_and_unit_scoped_facts(auth_headers, client):
|
|
"""검증: 숙박 수집 결과의 객실 단위 fact.
|
|
기대결과: units 가 생기고 객실별 fact 가 각자 붙는다(A동·B동이 각자 기준인원을 갖는다)."""
|
|
h = await auth_headers("u1")
|
|
pid = await _ready_place(client, h, kakao="k3")
|
|
await client.post(f"/v1/place/{pid}/collect", headers=h, json={})
|
|
await _run_worker()
|
|
|
|
units = (await client.get(f"/v1/place/{pid}/unit/list", headers=h)).json()["units"]
|
|
assert len(units) >= 2
|
|
|
|
facts = (await client.get(f"/v1/place/{pid}/fact/list", headers=h)).json()["facts"]
|
|
unit_facts = [f for f in facts if f.get("unit_id")]
|
|
assert unit_facts, "객실 단위 fact 가 하나도 없다"
|
|
capacities = [f for f in unit_facts if f["key"] == "standard_capacity"]
|
|
assert len({f["unit_id"] for f in capacities}) >= 2, "객실별로 따로 붙어야 한다"
|
|
|
|
|
|
async def test_pipeline_stores_media_with_origin_and_pending_review(auth_headers, client):
|
|
"""검증: 수집된 사진.
|
|
기대결과: origin_url·source_type=crawl 이 남고 PENDING_REVIEW 다 —
|
|
★ 재게시 권리 결론에 따라 통째로 걸러낼 수 있어야 하고, Vision 전이라 사람 확인 큐다."""
|
|
from sqlalchemy import text
|
|
|
|
h = await auth_headers("u1")
|
|
pid = await _ready_place(client, h, kakao="k4")
|
|
job_id = (await client.post(f"/v1/place/{pid}/collect", headers=h, json={})).json()["job_id"]
|
|
await _run_worker()
|
|
|
|
job = (await client.get(f"/v1/job/{job_id}", headers=h)).json()["job"]
|
|
assert job["result"]["media"]["stored"] > 0
|
|
|
|
|
|
async def test_recollect_skips_crawl_when_already_enough(auth_headers, client):
|
|
"""검증: 필수 항목이 이미 다 찬 사업장에 다시 수집을 건다.
|
|
기대결과: ★ 크롤링을 아예 하지 않는다 — 사이트를 만들 정보가 충분하면 여분의 크롤링은 낭비다."""
|
|
h = await auth_headers("u1")
|
|
pid = await _ready_place(client, h, kakao="k5")
|
|
|
|
await client.post(f"/v1/place/{pid}/collect", headers=h, json={})
|
|
await _run_worker()
|
|
first = (await client.get(f"/v1/place/{pid}/fact/list", headers=h)).json()
|
|
|
|
job_id = (await client.post(f"/v1/place/{pid}/collect", headers=h, json={})).json()["job_id"]
|
|
await _run_worker()
|
|
job = (await client.get(f"/v1/job/{job_id}", headers=h)).json()["job"]
|
|
|
|
assert job["result"]["coverage"]["enough"] is True
|
|
assert "크롤링 생략" in job["result"].get("note", "")
|
|
assert "fetch" not in job["result"], "충분한데 크롤링을 시도했다"
|
|
|
|
second = (await client.get(f"/v1/place/{pid}/fact/list", headers=h)).json()
|
|
assert len(second["facts"]) == len(first["facts"]), "재수집이 fact 를 중복 생성했다"
|
|
|
|
|
|
async def test_forced_recollect_is_idempotent(auth_headers, client):
|
|
"""검증: force=true 로 강제 재수집한다(항목이 이미 차 있어도).
|
|
기대결과: 다시 긁되 fact 는 REFRESHED, 사진은 중복 스킵 — 데이터가 부풀지 않는다."""
|
|
h = await auth_headers("u1")
|
|
pid = await _ready_place(client, h, kakao="k5f")
|
|
|
|
await client.post(f"/v1/place/{pid}/collect", headers=h, json={})
|
|
await _run_worker()
|
|
first = (await client.get(f"/v1/place/{pid}/fact/list", headers=h)).json()
|
|
|
|
job_id = (await client.post(f"/v1/place/{pid}/collect", headers=h, json={"force": True})).json()["job_id"]
|
|
await _run_worker()
|
|
job = (await client.get(f"/v1/job/{job_id}", headers=h)).json()["job"]
|
|
|
|
assert job["result"]["fetch"]["fetched"] == 1, "force 인데 크롤링을 안 했다"
|
|
assert job["result"]["media"]["stored"] == 0
|
|
assert job["result"]["media"]["skipped_duplicate"] > 0
|
|
|
|
second = (await client.get(f"/v1/place/{pid}/fact/list", headers=h)).json()
|
|
assert len(second["facts"]) == len(first["facts"]), "재수집이 fact 를 중복 생성했다"
|
|
|
|
|
|
async def test_coverage_reports_missing_required_fields(auth_headers, client):
|
|
"""검증: 수집 후 필수 항목 충족도.
|
|
기대결과: coverage 에 required/covered/missing 이 담긴다 — UI 가 '뭐가 비었나'를 보여줄 수 있다."""
|
|
h = await auth_headers("u1")
|
|
pid = await _ready_place(client, h, kakao="k5c")
|
|
|
|
job_id = (await client.post(f"/v1/place/{pid}/collect", headers=h, json={})).json()["job_id"]
|
|
await _run_worker()
|
|
cov = (await client.get(f"/v1/job/{job_id}", headers=h)).json()["job"]["result"]["coverage"]
|
|
|
|
assert cov["total"] > 0
|
|
assert cov["covered"] == cov["total"]
|
|
assert cov["missing"] == []
|
|
assert "check_in_time" in cov["required"]
|
|
|
|
|
|
async def test_recollect_does_not_touch_verified_value(auth_headers, client):
|
|
"""검증: 수집값을 사람이 승인한 뒤 다시 수집한다.
|
|
기대결과: 값이 같으므로 REFRESHED — ★ 사이트에 나가던 사실이 사라지지 않는다."""
|
|
h = await auth_headers("u1")
|
|
pid = await _ready_place(client, h, kakao="k6")
|
|
await client.post(f"/v1/place/{pid}/collect", headers=h, json={})
|
|
await _run_worker()
|
|
|
|
facts = (await client.get(f"/v1/place/{pid}/fact/list", headers=h)).json()["facts"]
|
|
target = next(f for f in facts if f["key"] == "check_in_time")
|
|
await client.post(f"/v1/place/{pid}/fact/{target['fact_id']}/transition", headers=h,
|
|
json={"status": FactStatus.VERIFIED.value})
|
|
assert (await client.get(f"/v1/place/{pid}/fact/list", headers=h)).json()["publishable"] == 1
|
|
|
|
await client.post(f"/v1/place/{pid}/collect", headers=h, json={})
|
|
await _run_worker()
|
|
|
|
after = (await client.get(f"/v1/place/{pid}/fact/list", headers=h)).json()
|
|
assert after["publishable"] == 1, "★ 재수집이 확인된 값을 밀어냈다"
|
|
check_in = [f for f in after["facts"] if f["key"] == "check_in_time"]
|
|
assert len(check_in) == 1 and check_in[0]["status"] == FactStatus.VERIFIED.value
|
|
|
|
|
|
async def test_recollect_cannot_overwrite_corrected_value(auth_headers, client):
|
|
"""검증: 사장님이 정정한 값에 재수집이 다른 값을 들고 온다.
|
|
기대결과: 노출값은 정정본 그대로, 크롤링 값은 후보로만 남는다 — ★ 절대규칙 6."""
|
|
h = await auth_headers("u1")
|
|
pid = await _ready_place(client, h, kakao="k7")
|
|
await client.post(f"/v1/place/{pid}/collect", headers=h, json={})
|
|
await _run_worker()
|
|
|
|
facts = (await client.get(f"/v1/place/{pid}/fact/list", headers=h)).json()["facts"]
|
|
target = next(f for f in facts if f["key"] == "check_in_time")
|
|
fid = target["fact_id"]
|
|
await client.post(f"/v1/place/{pid}/fact/{fid}/transition", headers=h, json={"status": FactStatus.VERIFIED.value})
|
|
await client.post(f"/v1/place/{pid}/fact/{fid}/transition", headers=h,
|
|
json={"status": FactStatus.CORRECTED.value, "value": "16:30"})
|
|
|
|
await client.post(f"/v1/place/{pid}/collect", headers=h, json={})
|
|
await _run_worker()
|
|
|
|
published = (await client.get(f"/v1/place/{pid}/fact/list", headers=h, params={"publishable_only": True})).json()["facts"]
|
|
check_in = [f for f in published if f["key"] == "check_in_time"]
|
|
assert len(check_in) == 1
|
|
assert check_in[0]["value"] == "16:30", "★ 자동 수집이 사장님 정정본을 덮어썼다"
|
|
assert check_in[0]["status"] == FactStatus.CORRECTED.value
|
|
|
|
|
|
async def test_pipeline_refuses_unverified_place(db_engine, owner_id):
|
|
"""검증: 검증 안 된 사업장의 수집 잡이 큐에 직접 들어간 경우(잡 적재 후 검증이 취소된 상황).
|
|
기대결과: 잡이 실패한다 — ★ 잡 실행 시점에도 게이트를 다시 확인한다."""
|
|
from sqlalchemy import text
|
|
|
|
pid = uuid.uuid4()
|
|
async with db_engine.begin() as conn:
|
|
await conn.execute(
|
|
text("INSERT INTO places (place_id, owner_user_id, name, category, status) "
|
|
"VALUES (:pid, :cid, :n, 1, 1)"),
|
|
{"pid": pid, "cid": uuid.UUID(owner_id), "n": "미검증펜션"},
|
|
)
|
|
|
|
q = JobQueue()
|
|
job_id = await q.enqueue(JobType.COLLECT.value, {"place_id": str(pid), "owner_user_id": owner_id}, max_attempts=1)
|
|
worker = Worker("test-worker", q, build_handler(), backoff_fn=lambda _a: 0)
|
|
await worker.process_one()
|
|
|
|
row = await q.get(job_id)
|
|
assert row["status"] == JobStatus.DEAD.value
|
|
assert "동일 업소 검증" in row["last_error"]
|