summaryrefslogtreecommitdiff
path: root/tests/unittests/sources
diff options
context:
space:
mode:
Diffstat (limited to 'tests/unittests/sources')
-rw-r--r--tests/unittests/sources/azure/test_imds.py1
-rw-r--r--tests/unittests/sources/test_azure.py4
2 files changed, 2 insertions, 3 deletions
diff --git a/tests/unittests/sources/azure/test_imds.py b/tests/unittests/sources/azure/test_imds.py
index 03f66502..9a8aad88 100644
--- a/tests/unittests/sources/azure/test_imds.py
+++ b/tests/unittests/sources/azure/test_imds.py
@@ -549,7 +549,6 @@ class TestFetchReprovisionData:
"terminal_error",
[
requests.ConnectionError("Fake connection error"),
- requests.Timeout("Fake connection timeout"),
],
)
def test_retry_until_failure(
diff --git a/tests/unittests/sources/test_azure.py b/tests/unittests/sources/test_azure.py
index 9815c913..6c3e9281 100644
--- a/tests/unittests/sources/test_azure.py
+++ b/tests/unittests/sources/test_azure.py
@@ -2921,8 +2921,8 @@ class TestPreprovisioningHotAttachNics(CiTestCase):
# Re-run tests to verify max connection error retries.
m_request.reset_mock()
m_request.side_effect = [
- requests.Timeout("Fake connection timeout")
- ] * 9 + [requests.ConnectionError("Fake Network Unreachable")] * 9
+ requests.ConnectionError("Fake Network Unreachable")
+ ] * 15
dsa = dsaz.DataSourceAzure({}, distro=distro, paths=self.paths)