From 55c58e3c7a34db5490e549242d462aeba4ab06c1 Mon Sep 17 00:00:00 2001
From: holger krekel <holger@merlinux.eu>
Date: Mon, 16 Oct 2023 19:10:59 +0200
Subject: [PATCH] add support for using a second chatmail server

---
 online-tests/conftest.py     | 43 ++++++++++++++++++++++++++++++++----
 online-tests/test_0_basic.py | 10 +++++++++
 2 files changed, 49 insertions(+), 4 deletions(-)

diff --git a/online-tests/conftest.py b/online-tests/conftest.py
index cd2a2dbb..e1167663 100644
--- a/online-tests/conftest.py
+++ b/online-tests/conftest.py
@@ -1,6 +1,7 @@
 import os
 import io
 import random
+import contextlib
 import subprocess
 import imaplib
 import smtplib
@@ -34,6 +35,19 @@ def sshdomain(maildomain):
     return os.environ.get("CHATMAIL_SSH", maildomain)
 
 
+@pytest.fixture
+def maildomain2():
+    domain = os.environ.get("CHATMAIL_DOMAIN2")
+    if not domain:
+        pytest.skip("set CHATMAIL_DOMAIN2 to a ssh-reachable chatmail instance")
+    return domain
+
+
+@pytest.fixture
+def sshdomain2(maildomain2):
+    return os.environ.get("CHATMAIL_SSH2", maildomain2)
+
+
 def pytest_report_header():
     domain = os.environ.get("CHATMAIL_DOMAIN")
     if domain:
@@ -93,7 +107,9 @@ def gencreds(maildomain):
     count = itertools.count()
     next(count)
 
-    def gen():
+    def gen(domain=None):
+        if domain is not None:
+            maildomain = domain
         while 1:
             num = next(count)
             alphanumeric = "abcdefghijklmnopqrstuvwxyz1234567890"
@@ -102,7 +118,7 @@ def gencreds(maildomain):
             password = "".join(random.choices(alphanumeric, k=10))
             yield f"{user}@{maildomain}", f"{password}"
 
-    return lambda: next(gen())
+    return lambda domain=None: next(gen(domain))
 
 
 #
@@ -117,12 +133,13 @@ class ChatmailTestProcess:
     def __init__(self, pytestconfig, maildomain, gencreds):
         self.pytestconfig = pytestconfig
         self.maildomain = maildomain
+        assert "." in self.maildomain, maildomain
         self.gencreds = gencreds
         self._addr2files = {}
 
     def get_liveconfig_producer(self):
         while 1:
-            user, password = self.gencreds()
+            user, password = self.gencreds(self.maildomain)
             config = {
                 "addr": user,
                 "mail_pw": password,
@@ -140,7 +157,25 @@ class ChatmailTestProcess:
 
 
 @pytest.fixture
-def cmfactory(request, maildomain, gencreds, tmpdir, data):
+def switch_maildomain():
+    """return a function that allows to switch an account factory temporarily
+    to another maildomain.
+    """
+
+    # nb. a bit hacky
+    # would probably be better if deltachat's test machinery grows native support
+    @contextlib.contextmanager
+    def switch(acfactory, maildomain2):
+        old_domain = acfactory.testprocess.maildomain
+        acfactory.testprocess.maildomain = maildomain2
+        yield
+        acfactory.testprocess.maildomain = old_domain
+
+    return switch
+
+
+@pytest.fixture
+def cmfactory(request, gencreds, tmpdir, data, maildomain):
     # cloned from deltachat.testplugin.amfactory
     pytest.importorskip("deltachat")
     from deltachat.testplugin import ACFactory
diff --git a/online-tests/test_0_basic.py b/online-tests/test_0_basic.py
index 5d3a9648..f4d090b9 100644
--- a/online-tests/test_0_basic.py
+++ b/online-tests/test_0_basic.py
@@ -2,3 +2,13 @@ def test_remotelog(remotelog, imap_or_smtp):
     lineproducer = remotelog.iter(imap_or_smtp.logunit)
     imap_or_smtp.connect()
     assert imap_or_smtp.logunit in next(lineproducer)
+
+
+def test_use_two_chatmailservers(cmfactory, maildomain2, switch_maildomain):
+    (ac1,) = cmfactory.get_online_accounts(1)
+    with switch_maildomain(cmfactory, maildomain2):
+        (ac2,) = cmfactory.get_online_accounts(1)
+    cmfactory.get_accepted_chat(ac1, ac2)
+    domain1 = ac1.get_config("addr").split("@")[1]
+    domain2 = ac2.get_config("addr").split("@")[1]
+    assert domain1 != domain2