From 9338076640f326769c1efe1c408d7f5e75d243c1 Mon Sep 17 00:00:00 2001
From: Rene Luria <rene.luria@infomaniak.com>
Date: Thu, 24 Jun 2021 22:46:50 +0200
Subject: [PATCH] 1.16 compat: mock certbot._internal.main.display_util.notify
 (#15)

* 1.16 compat: mock certbot._internal.main.display_util.notify
* backward compatibility for cerbot without notify in certbot.display.util
---
 certbot_dns_infomaniak/dns_infomaniak_test.py | 18 +++++++++++++++++-
 requirements.txt                              |  2 +-
 2 files changed, 18 insertions(+), 2 deletions(-)

diff --git a/certbot_dns_infomaniak/dns_infomaniak_test.py b/certbot_dns_infomaniak/dns_infomaniak_test.py
index 7b4b98a..d660607 100644
--- a/certbot_dns_infomaniak/dns_infomaniak_test.py
+++ b/certbot_dns_infomaniak/dns_infomaniak_test.py
@@ -6,6 +6,9 @@
 from unittest import mock
 import requests_mock
 
+import sys
+import io
+
 from certbot.errors import PluginError
 try:
     import certbot.compat.os as os
@@ -39,9 +42,22 @@ def setUp(self):
         self.auth = Authenticator(self.config, "infomaniak")
 
         self.mock_client = mock.MagicMock(default_propagation_seconds=15)
-        # _get_ispconfig_client | pylint: disable=protected-access
+
         self.auth._api_client = mock.MagicMock(return_value=self.mock_client)
 
+        try:
+            from certbot.display.util import notify  # noqa: F401
+            notify_patch = mock.patch('certbot._internal.main.display_util.notify')
+            self.mock_notify = notify_patch.start()
+            self.addCleanup(notify_patch.stop)
+            self.old_stdout = sys.stdout
+            sys.stdout = io.StringIO()
+        except ImportError:
+            self.old_stdout = sys.stdout
+
+    def tearDown(self):
+        sys.stdout = self.old_stdout
+
     def test_perform(self):
         """Tests the perform function to see if client method is called"""
         self.auth.perform([self.achall])
diff --git a/requirements.txt b/requirements.txt
index 854888c..7c4e489 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,4 +1,4 @@
-certbot<1.16.0
+certbot>=0.31.0
 setuptools
 requests
 requests-mock