Collapse linkify mentions (#123)

1 files changed, 40 insertions, 0 deletions

diff --git a/tests/conftest.py b/tests/conftest.py
index 80622f0..c67717c 100644
--- a/tests/conftest.py
+++ b/tests/conftest.py
@@ -80,6 +80,12 @@ def domain() -> Domain:
 
 @pytest.fixture
 @pytest.mark.django_db
+def domain2() -> Domain:
+    return Domain.objects.create(domain="example2.com", local=True, public=True)
+
+
+@pytest.fixture
+@pytest.mark.django_db
 def identity(user, domain) -> Identity:
     """
     Creates a basic test identity with a user and domain.
@@ -96,6 +102,23 @@ def identity(user, domain) -> Identity:
 
 
 @pytest.fixture
+@pytest.mark.django_db
+def identity2(user, domain2) -> Identity:
+    """
+    Creates a basic test identity with a user and domain.
+    """
+    identity = Identity.objects.create(
+        actor_uri="https://example2.com/@test@example2.com/",
+        username="test",
+        domain=domain2,
+        name="Test User Domain2",
+        local=True,
+    )
+    identity.users.set([user])
+    return identity
+
+
+@pytest.fixture
 def other_identity(user, domain) -> Identity:
     """
     Creates a different basic test identity with a user and domain.
@@ -129,6 +152,23 @@ def remote_identity() -> Identity:
 
 
 @pytest.fixture
+@pytest.mark.django_db
+def remote_identity2() -> Identity:
+    """
+    Creates a basic remote test identity with a domain.
+    """
+    domain = Domain.objects.create(domain="remote2.test", local=False)
+    return Identity.objects.create(
+        actor_uri="https://remote2.test/test-actor/",
+        profile_uri="https://remote2.test/@test/",
+        username="test",
+        domain=domain,
+        name="Test2 Remote User",
+        local=False,
+    )
+
+
+@pytest.fixture
 def stator(config_system) -> StatorRunner:
     """
     Return an initialized StatorRunner for tests that need state transitioning