Skip to content

Commit fd14a15

Browse files
committed
Updated test_identical_url_in_both in test_migrate_dump
1 parent df188ff commit fd14a15

File tree

1 file changed

+31
-4
lines changed

1 file changed

+31
-4
lines changed

sde_collections/tests/test_migrate_dump.py

Lines changed: 31 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33

44
import pytest
55

6-
from sde_collections.models.collection_choice_fields import DocumentTypes
6+
from sde_collections.models.collection_choice_fields import Divisions, DocumentTypes
77
from sde_collections.models.delta_patterns import (
88
DeltaDocumentTypePattern,
99
DeltaExcludePattern,
@@ -16,7 +16,7 @@
1616
DumpUrlFactory,
1717
)
1818

19-
DELTA_COMPARISON_FIELDS = ["scraped_title"] # Assuming a central definition
19+
DELTA_COMPARISON_FIELDS = ["scraped_title", "tdamm_tag", "division"] # Assuming a central definition
2020

2121

2222
@pytest.mark.django_db
@@ -80,9 +80,36 @@ def test_url_in_curated_only(self):
8080
assert delta.scraped_title == curated_url.scraped_title
8181

8282
def test_identical_url_in_both(self):
83+
"""When DumpUrl and CuratedUrl have identical values, no DeltaUrl should be created."""
8384
collection = CollectionFactory()
84-
dump_url = DumpUrlFactory(collection=collection, scraped_title="Same Title")
85-
CuratedUrlFactory(collection=collection, url=dump_url.url, scraped_title="Same Title")
85+
86+
# Create DumpUrl with specific values
87+
dump_url = DumpUrlFactory(collection=collection, scraped_title="Same Title", division=Divisions.ASTROPHYSICS)
88+
89+
# Ensure tdamm_tag is explicitly set to match
90+
dump_url.tdamm_tag_manual = []
91+
dump_url.tdamm_tag_ml = []
92+
dump_url.save()
93+
94+
# Create CuratedUrl with identical values
95+
curated_url = CuratedUrlFactory(
96+
collection=collection,
97+
url=dump_url.url, # Use the same URL
98+
scraped_title="Same Title",
99+
division=Divisions.ASTROPHYSICS,
100+
)
101+
102+
# Set identical tdamm_tag values
103+
curated_url.tdamm_tag_manual = []
104+
curated_url.tdamm_tag_ml = []
105+
curated_url.save()
106+
107+
# Verify fields are identical before migration
108+
assert dump_url.scraped_title == curated_url.scraped_title
109+
assert dump_url.division == curated_url.division
110+
assert dump_url.tdamm_tag == curated_url.tdamm_tag
111+
112+
# Migrate and assert
86113
collection.migrate_dump_to_delta()
87114
assert not DeltaUrl.objects.filter(url=dump_url.url).exists()
88115

0 commit comments

Comments
 (0)