From a46ea8d3e56f5789f4b33a97691b06856bd420ab Mon Sep 17 00:00:00 2001
From: Iain Lane <iain.lane@canonical.com>
Date: Tue, 31 Mar 2020 09:30:50 +0100
Subject: [PATCH] autopkgtest: Add some retry logic

---
 britney2/policies/autopkgtest.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/britney2/policies/autopkgtest.py b/britney2/policies/autopkgtest.py
index 15ecad5..1a93691 100644
--- a/britney2/policies/autopkgtest.py
+++ b/britney2/policies/autopkgtest.py
@@ -25,6 +25,8 @@ import re
 import sys
 import urllib.parse
 import requests
+from requests.adapters import HTTPAdapter
+from requests.packages.urllib3.util.retry import Retry
 
 import apt_pkg
 import amqplib.client_0_8 as amqp
@@ -85,6 +87,10 @@ class AutopkgtestPolicy(BasePolicy):
             self.results_cache_file = os.path.join(self.test_state_dir, 'results.cache')
 
         self.session = requests.Session()
+        retry = Retry(total=3, read=3, connect=3)
+        adapter = HTTPAdapter(max_retries=retry)
+        self.session.mount('http://', adapter)
+        self.session.mount('https://', adapter)
 
         try:
             self.options.adt_ppas = self.options.adt_ppas.strip().split()