1 files changed, 116 insertions, 0 deletions
diff --git a/python/tests/merge_containers.py b/python/tests/merge_containers.py
new file mode 100644
index 00000000..a657522e
--- /dev/null
+++ b/python/tests/merge_containers.py
@@ -0,0 +1,116 @@
+from fatcat_openapi_client import ContainerEntity
+from fixtures import api
+
+from fatcat_tools.mergers.containers import ContainerMerger
+
+
+def test_choose_primary_container(api) -> None:
+
+    release_counts = dict()
+    redirects = dict()
+    em = ContainerMerger(api=api)
+
+    ce_stub = ContainerEntity(
+        ident="pppppp5apzfhbbxxc7rgu2yw6m",
+        name="dummy journal",
+    )
+    release_counts[ce_stub.ident] = 0
+    redirects[ce_stub.ident] = []
+
+    ce_partial = ContainerEntity(
+        ident="eeeeeeeapzfhbbxxc7rgu2yw6m",
+        name="dummy complete journal",
+        publisher="some publisher",
+        issnl="1234-5678",
+        publication_status="active",
+        extra=dict(asdf=123, ia=dict(asdf=True)),
+    )
+    release_counts[ce_partial.ident] = 0
+    redirects[ce_partial.ident] = []
+
+    ce_partial_redirects = ContainerEntity(
+        ident="rrrrrrrrrrfhbbxxc7rgu2yw6m",
+        name="dummy complete journal",
+        publisher="some publisher",
+        issnl="1234-5678",
+        publication_status="active",
+        extra=dict(asdf=123, ia=dict(asdf=True)),
+    )
+    release_counts[ce_partial_redirects.ident] = 0
+    redirects[ce_partial_redirects.ident] = [
+        "zzzzzzzzrrfhbbxxc7rgu2yw6m",
+    ]
+
+    ce_complete_zero = ContainerEntity(
+        ident="oooooooapzfhbbxxc7rgu2yw6m",
+        name="dummy complete journal",
+        publisher="some publisher",
+        issnl="1234-5678",
+        publication_status="active",
+        extra=dict(asdf=123, ia=dict(asdf=True)),
+    )
+    release_counts[ce_complete_zero.ident] = 0
+    redirects[ce_complete_zero.ident] = []
+
+    ce_complete_small = ContainerEntity(
+        ident="cccccccapzfhbbxxc7rgu2yw6m",
+        name="dummy complete journal",
+        publisher="some publisher",
+        issnl="1234-5678",
+        publication_status="active",
+        extra=dict(asdf=123, ia=dict(asdf=True)),
+    )
+    release_counts[ce_complete_small.ident] = 10
+    redirects[ce_complete_small.ident] = []
+
+    ce_complete_big = ContainerEntity(
+        ident="ddddddddpzfhbbxxc7rgu2yw6m",
+        name="dummy complete journal",
+        publisher="some publisher",
+        issnl="1234-5678",
+        publication_status="active",
+        extra=dict(asdf=123, ia=dict(asdf=True)),
+    )
+    release_counts[ce_complete_big.ident] = 9999999
+    redirects[ce_complete_big.ident] = []
+
+    assert (
+        em.choose_primary_container([ce_stub, ce_partial], redirects, release_counts)
+        == ce_partial.ident
+    )
+    assert (
+        em.choose_primary_container(
+            [ce_stub, ce_complete_zero, ce_partial], redirects, release_counts
+        )
+        == ce_complete_zero.ident
+    )
+    assert (
+        em.choose_primary_container(
+            [ce_stub, ce_partial_redirects, ce_complete_zero, ce_partial],
+            redirects,
+            release_counts,
+        )
+        == ce_partial_redirects.ident
+    )
+    assert (
+        em.choose_primary_container(
+            [ce_stub, ce_complete_zero, ce_complete_small, ce_partial],
+            redirects,
+            release_counts,
+        )
+        == ce_complete_small.ident
+    )
+    assert (
+        em.choose_primary_container(
+            [ce_stub, ce_complete_big, ce_complete_zero, ce_complete_small, ce_partial],
+            redirects,
+            release_counts,
+        )
+        == ce_complete_big.ident
+    )
+    assert (
+        em.choose_primary_container(
+            [ce_complete_small, ce_complete_big], redirects, release_counts
+        )
+        == ce_complete_big.ident
+    )